Function TestFileCacheForZstd

fs_test.go:1324–1385

(t *testing.T)

}

func TestFileCacheForZstd(t *testing.T) {
	f, err := os.CreateTemp(os.TempDir(), "test")
	if err != nil {
		t.Fatal(err)
	}
	data := bytes.Repeat([]byte("1"), 1000)
	changedData := bytes.Repeat([]byte("2"), 1000)
	_, err = f.Write(data)
	if err != nil {
		t.Fatal(err)
	}
	err = f.Sync()
	if err != nil {
		t.Fatal(err)
	}
	fs := FS{Root: os.TempDir(), Compress: true, CompressZstd: true, CacheDuration: time.Second * 60}
	h := fs.NewRequestHandler()
	var ctx RequestCtx
	var req Request
	req.Header.Set("Accept-Encoding", "zstd")
	req.SetRequestURI("http://foobar.com/" + strings.TrimPrefix(f.Name(), os.TempDir()))
	ctx.Init(&req, nil, nil)
	h(&ctx)
	if !bytes.Equal(ctx.Response.Header.ContentEncoding(), []byte("zstd")) {
		t.Fatalf("Unexpected 'Content-Encoding' %q. Expecting %q", ctx.Response.Header.ContentEncoding(), "zstd")
	}
	ctx.Response.Reset()
	_, err = f.Seek(0, io.SeekStart)
	if err != nil {
		t.Fatal(err)
	}
	_, err = f.Write(changedData)
	if err != nil {
		t.Fatal(err)
	}
	f.Close()
	h(&ctx)
	if !bytes.Equal(ctx.Response.Header.ContentEncoding(), []byte("zstd")) {
		t.Fatalf("Unexpected 'Content-Encoding' %q. Expecting %q", ctx.Response.Header.ContentEncoding(), "zstd")
	}
	d, err := acquireZstdReader(strings.NewReader(string(ctx.Response.Body())))
	if err != nil {
		t.Fatalf("invalid zstd reader")
	}
	plainText, err := io.ReadAll(d)
	d.Close()
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Equal(plainText, data) {
		t.Fatalf("Unexpected response body %q. Expecting %q . Zstd cache doesn't work", plainText, data)
	}
	ctx.Request.Header.Del("Accept-Encoding")
	ctx.Response.Reset()
	h(&ctx)
	if !bytes.Equal(ctx.Response.Header.ContentEncoding(), []byte("")) {
		t.Fatalf("Unexpected 'Content-Encoding' %q. Expecting %q", ctx.Response.Header.ContentEncoding(), "")
	}