fix tests

annabarnes1138 · May 26, 2021 · e599f1d · e599f1d
1 parent 75a468d
commit e599f1d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/releaser/releaser_test.go b/pkg/releaser/releaser_test.go
@@ -53,6 +53,16 @@ func (m *MockClient) Get(url string) (*http.Response, error) {
 	}
 }
 
+func (m *MockClient) GetWithToken(url string, token string) (*http.Response, error) {
+	if m.statusCode == http.StatusOK {
+		file, _ := os.Open(m.file)
+		reader := bufio.NewReader(file)
+		return &http.Response{StatusCode: http.StatusOK, Body: ioutil.NopCloser(reader)}, nil
+	} else {
+		return &http.Response{StatusCode: http.StatusNotFound, Body: ioutil.NopCloser(nil)}, nil
+	}
+}
+
 func (f *FakeGitHub) CreateRelease(ctx context.Context, input *github.Release) error {
 	f.Called(ctx, input)
 	f.release = input