go-calc-stub-server.go

func TestRemoteHandlerWithStubServer(t *testing.T) {
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusServiceUnavailable)
	}))
	defer ts.Close()

	url := fmt.Sprintf("/remote?url=%s", ts.URL)
	req := httptest.NewRequest("GET", url, nil)
	rr := httptest.NewRecorder()

	remoteHandler(&http.Client{})(rr, req)
	res := rr.Result()
	resBody, _ := ioutil.ReadAll(res.Body)

	assert.Equal(t, http.StatusInternalServerError, res.StatusCode)
	assert.JSONEq(t, `{"message": "remote call failed, returned status code 503"}`, string(resBody))
}