test(response_writer): add tests for ResponseWriter.Unwrap method

This commit is contained in:
Raju Ahmed 2026-03-27 07:32:52 +06:00
parent 1fde3243c2
commit 675838ec99

View File

@ -33,12 +33,23 @@ func init() {
SetMode(TestMode) SetMode(TestMode)
} }
// test for ResponseWriter.Unwrap
func TestResponseWriterUnwrap(t *testing.T) { func TestResponseWriterUnwrap(t *testing.T) {
testWriter := httptest.NewRecorder() testWriter := httptest.NewRecorder()
writer := &responseWriter{ResponseWriter: testWriter} writer := &responseWriter{ResponseWriter: testWriter}
assert.Same(t, testWriter, writer.Unwrap()) assert.Same(t, testWriter, writer.Unwrap())
} }
func TestResponseWriterUnwrapViaInterface(t *testing.T) {
testWriter := httptest.NewRecorder()
writer := &responseWriter{}
writer.reset(testWriter)
var w ResponseWriter = writer
unwrapped := w.Unwrap()
assert.Same(t, testWriter, unwrapped)
}
func TestResponseWriterReset(t *testing.T) { func TestResponseWriterReset(t *testing.T) {
testWriter := httptest.NewRecorder() testWriter := httptest.NewRecorder()
writer := &responseWriter{} writer := &responseWriter{}