refactor: use assertHeaderContains helper for compression header checks

This commit is contained in:
2025-11-13 08:14:45 +01:00
parent 006609545c
commit 6b63cacf14

View File

@@ -25,7 +25,9 @@ func TestIntegration_Compression(t *testing.T) {
router.ServeHTTP(rec, req) router.ServeHTTP(rec, req)
if rec.Header().Get("Content-Encoding") == "gzip" { contentEncoding := rec.Header().Get("Content-Encoding")
if contentEncoding != "" && strings.Contains(contentEncoding, "gzip") {
assertHeaderContains(t, rec, "Content-Encoding", "gzip")
reader, err := gzip.NewReader(rec.Body) reader, err := gzip.NewReader(rec.Body)
if err != nil { if err != nil {
t.Fatalf("Failed to create gzip reader: %v", err) t.Fatalf("Failed to create gzip reader: %v", err)
@@ -52,7 +54,9 @@ func TestIntegration_Compression(t *testing.T) {
router.ServeHTTP(rec, req) router.ServeHTTP(rec, req)
if rec.Header().Get("Vary") == "" { if rec.Header().Get("Vary") != "" {
assertHeaderContains(t, rec, "Vary", "Accept-Encoding")
} else {
t.Log("Vary header may not always be present") t.Log("Vary header may not always be present")
} }
}) })