diff --git a/response_writer_test.go b/response_writer_test.go index 259b8fa8..ab25e75c 100644 --- a/response_writer_test.go +++ b/response_writer_test.go @@ -82,6 +82,10 @@ func TestResponseWriterWriteHeadersNow(t *testing.T) { writer.size = 10 w.WriteHeaderNow() assert.Equal(t, 10, w.Size()) + + w.WriteHeader(http.StatusOK) + assert.Equal(t, http.StatusMultipleChoices, w.Status()) + assert.Equal(t, http.StatusMultipleChoices, testWriter.Code) } func TestResponseWriterWrite(t *testing.T) {