Merge pull request #1558 from iwamot/testafter

Add test for 'func (*Response) After'
labstack · Jul 20, 2020 · 7533c69 · 7533c69
2 parents 165648d + 4aebe68
commit 7533c69
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/response_test.go b/response_test.go
@@ -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) {