Function TestJSONToHTTP

server/middleware_test.go:73–121

(t *testing.T)

}

func TestJSONToHTTP(t *testing.T) {
	tests := []struct {
		given     JSONEndpoint
		givenBody io.Reader

		wantCode int
		wantBody string
	}{
		{
			JSONEndpoint(func(r *http.Request) (int, interface{}, error) {
				bod, err := ioutil.ReadAll(r.Body)
				if err != nil {
					t.Error("unable to read given request body: ", err)
				}
				if string(bod) != "yup" {
					t.Errorf("expected 'yup', got %+v", string(bod))
				}
				return http.StatusOK, struct{ Howdy string }{"Hi"}, nil
			}),
			bytes.NewBufferString("yup"),
			http.StatusOK,
			"{\"Howdy\":\"Hi\"}\n",
		},
		{
			JSONEndpoint(func(r *http.Request) (int, interface{}, error) {
				return http.StatusServiceUnavailable, nil, &testJSONError{"nope"}
			}),
			nil,
			http.StatusServiceUnavailable,
			"{\"error\":\"nope\"}\n",
		},
	}

	for _, test := range tests {
		r, _ := http.NewRequest("GET", "", test.givenBody)
		w := httptest.NewRecorder()
		JSONToHTTP(test.given).ServeHTTP(w, r)

		if w.Code != test.wantCode {
			t.Errorf("expected status code %d, got %d", test.wantCode, w.Code)
		}
		if gotHdr := w.Header().Get("Content-Type"); gotHdr != jsonContentType {
			t.Errorf("expected Content-Type header of '%#v', got '%#v'", jsonContentType, gotHdr)
		}
		if got := w.Body.String(); got != test.wantBody {
			t.Errorf("expected body of '%#v', got '%#v'", test.wantBody, got)
		}
	}
}

type testJSONError struct {
	Err string `json:"error"`