go-calc-client-mock.go

func TestRemoteHandlerWithClientMock(t *testing.T) {
	mockRes := &http.Response{
		StatusCode: http.StatusOK,
		Body:       ioutil.NopCloser(strings.NewReader(`{"foo": "bar"}`)),
	}
	clientMock := mocks.HTTPClient{}
	clientMock.On("Do", mock.AnythingOfType("*http.Request")).Return(mockRes, nil)

	req := httptest.NewRequest("GET", "/remote", nil)
	rr := httptest.NewRecorder()

	remoteHandler(&clientMock)(rr, req)
	res := rr.Result()
	resBody, _ := ioutil.ReadAll(res.Body)

	assert.Equal(t, http.StatusOK, res.StatusCode)
	assert.JSONEq(t, `{"foo": "bar"}`, string(resBody))
}