console logger HTTP status bug fixed and the corresponding unit test added (#3453)

This commit is contained in:
Alireza (Pure)
2023-01-02 07:08:53 +03:30
committed by GitHub
parent 82e1c53cc0
commit 41f2669ebc
2 changed files with 19 additions and 0 deletions
+18
View File
@@ -132,3 +132,21 @@ func TestResponseWriterFlush(t *testing.T) {
assert.NoError(t, err)
assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
}
func TestResponseWriterStatusCode(t *testing.T) {
testWriter := httptest.NewRecorder()
writer := &responseWriter{}
writer.reset(testWriter)
w := ResponseWriter(writer)
w.WriteHeader(http.StatusOK)
w.WriteHeaderNow()
assert.Equal(t, http.StatusOK, w.Status())
assert.True(t, w.Written())
w.WriteHeader(http.StatusUnauthorized)
// status must be 200 although we tried to change it
assert.Equal(t, http.StatusOK, w.Status())
}