test: fix failed test

This commit is contained in:
HFO4
2022-04-15 15:53:10 +08:00
parent 478d390867
commit 18f5bffed1
4 changed files with 22 additions and 14 deletions

View File

@@ -76,3 +76,12 @@ func TestIsFunctionEnabled(t *testing.T) {
}
}
func TestCacheControl(t *testing.T) {
a := assert.New(t)
TestFunc := CacheControl()
rec := httptest.NewRecorder()
c, _ := gin.CreateTestContext(rec)
TestFunc(c)
a.Contains(c.Writer.Header().Get("Cache-Control"), "no-cache")
}