1
0
mirror of https://github.com/labstack/echo.git synced 2025-07-17 01:43:02 +02:00

Add test for 'func (*Response) After'

This commit is contained in:
Takashi Iwamoto
2020-04-26 22:01:24 +09:00
parent c08f30359b
commit 4aebe68f37

View File

@ -19,8 +19,13 @@ func TestResponse(t *testing.T) {
res.Before(func() {
c.Response().Header().Set(HeaderServer, "echo")
})
// After
res.After(func() {
c.Response().Header().Set(HeaderXFrameOptions, "DENY")
})
res.Write([]byte("test"))
assert.Equal(t, "echo", rec.Header().Get(HeaderServer))
assert.Equal(t, "DENY", rec.Header().Get(HeaderXFrameOptions))
}
func TestResponse_Write_FallsBackToDefaultStatus(t *testing.T) {