Function TestZstdLevels

gzhttp/compress_test.go:2084–2123

TestZstdLevels tests zstd compression levels.

(t *testing.T)

// TestZstdLevels tests zstd compression levels.
func TestZstdLevels(t *testing.T) {
	levels := []int{
		int(zstd.SpeedFastest),
		int(zstd.SpeedDefault),
		int(zstd.SpeedBetterCompression),
		int(zstd.SpeedBestCompression),
	}
	for _, level := range levels {
		t.Run(fmt.Sprintf("level-%d", level), func(t *testing.T) {
			wrapper, err := NewWrapper(EnableGzip(false), ZstdCompressionLevel(level))
			if err != nil {
				t.Fatal(err)
			}
			handler := wrapper(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.Write(testBody)
			}))

			req, _ := http.NewRequest("GET", "/", nil)
			req.Header.Set("Accept-Encoding", "zstd")
			res := httptest.NewRecorder()
			handler.ServeHTTP(res, req)

			assertEqual(t, "zstd", res.Header().Get("Content-Encoding"))

			// Decompress and verify
			dec, err := zstd.NewReader(res.Body)
			if err != nil {
				t.Fatalf("zstd.NewReader failed: %v", err)
			}
			got, err := io.ReadAll(dec)
			dec.Close()
			if err != nil {
				t.Fatalf("io.ReadAll failed: %v", err)
			}
			if !bytes.Equal(testBody, got) {
				t.Errorf("response body mismatch at level %d", level)
			}
		})
	}
}

// TestPreferGzip tests that gzip is preferred when PreferZstd(false) is set.
func TestPreferGzip(t *testing.T) {