Function TestZstdHandler

gzhttp/compress_test.go:1997–2021

TestZstdHandler tests basic zstd compression.

(t *testing.T)

// TestZstdHandler tests basic zstd compression.
func TestZstdHandler(t *testing.T) {
	handler := newTestHandler(testBody)

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

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

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

// TestZstdPreferredOverGzip tests that zstd is preferred when both are accepted with equal qvalues.
func TestZstdPreferredOverGzip(t *testing.T) {