mirror of
https://github.com/zeromicro/go-zero.git
synced 2026-05-10 08:29:58 +08:00
chore: add unit test for WithCodeResponseWriter (#5028)
Signed-off-by: kevin <wanjunfeng@gmail.com>
This commit is contained in:
@@ -46,3 +46,15 @@ func TestWithCodeResponseWriter_Hijack(t *testing.T) {
|
||||
writer.Hijack()
|
||||
})
|
||||
}
|
||||
|
||||
func TestWithCodeResponseWriter_Unwrap(t *testing.T) {
|
||||
resp := httptest.NewRecorder()
|
||||
writer := NewWithCodeResponseWriter(resp)
|
||||
unwrapped := writer.Unwrap()
|
||||
assert.Equal(t, resp, unwrapped)
|
||||
|
||||
// Test with a nested WithCodeResponseWriter
|
||||
nestedWriter := NewWithCodeResponseWriter(writer)
|
||||
unwrappedNested := nestedWriter.Unwrap()
|
||||
assert.Equal(t, resp, unwrappedNested)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user