go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2025-07-01 00:55:04 +02:00
Code Issues Releases Activity

Merge pull request #1558 from iwamot/testafter

Browse Source 
Add test for 'func (*Response) After'
This commit is contained in:
Roland Lammel
2020-07-20 18:00:17 +02:00
committed by GitHub
parent 165648d4a0 4aebe68f37
commit 7533c6984b
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
response_test.go
View File

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