Function TestBadResponseLoggingWriter

middleware/response_test.go:16–49

(t *testing.T)

)

func TestBadResponseLoggingWriter(t *testing.T) {
	for _, tc := range []struct {
		statusCode int
		data       string
		expected   string
	}{
		{http.StatusOK, "", ""},
		{http.StatusOK, "some data", ""},
		{http.StatusUnprocessableEntity, "unprocessable", ""},
		{http.StatusInternalServerError, "", ""},
		{http.StatusInternalServerError, "bad juju", "bad juju\n"},
	} {
		w := httptest.NewRecorder()
		var buf bytes.Buffer
		wrapped := newBadResponseLoggingWriter(w, &buf)
		switch {
		case tc.data == "":
			wrapped.WriteHeader(tc.statusCode)
		case tc.statusCode < 300 && tc.data != "":
			wrapped.WriteHeader(tc.statusCode)
			_, err := wrapped.Write([]byte(tc.data))
			require.NoError(t, err)
		default:
			http.Error(wrapped, tc.data, tc.statusCode)
		}
		if wrapped.getStatusCode() != tc.statusCode {
			t.Errorf("Wrong status code: have %d want %d", wrapped.getStatusCode(), tc.statusCode)
		}
		data := buf.String()
		if data != tc.expected {
			t.Errorf("Wrong data: have %q want %q", data, tc.expected)
		}
	}
}

// nonFlushingResponseWriter implements http.ResponseWriter but does not implement http.Flusher
type nonFlushingResponseWriter struct{}