1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
package fasthttp
import (
"bytes"
"fmt"
"io"
"testing"
)
func TestZstdBytesSerial(t *testing.T) {
t.Parallel()
if err := testZstdBytes(); err != nil {
t.Fatal(err)
}
}
func TestZstdBytesConcurrent(t *testing.T) {
t.Parallel()
if err := testConcurrent(10, testZstdBytes); err != nil {
t.Fatal(err)
}
}
func testZstdBytes() error {
for _, s := range compressTestcases {
if err := testZstdBytesSingleCase(s); err != nil {
return err
}
}
return nil
}
func testZstdBytesSingleCase(s string) error {
prefix := []byte("foobar")
ZstdpedS := AppendZstdBytes(prefix, []byte(s))
if !bytes.Equal(ZstdpedS[:len(prefix)], prefix) {
return fmt.Errorf("unexpected prefix when compressing %q: %q. Expecting %q", s, ZstdpedS[:len(prefix)], prefix)
}
unZstdedS, err := AppendUnzstdBytes(prefix, ZstdpedS[len(prefix):])
if err != nil {
return fmt.Errorf("unexpected error when uncompressing %q: %w", s, err)
}
if !bytes.Equal(unZstdedS[:len(prefix)], prefix) {
return fmt.Errorf("unexpected prefix when uncompressing %q: %q. Expecting %q", s, unZstdedS[:len(prefix)], prefix)
}
unZstdedS = unZstdedS[len(prefix):]
if string(unZstdedS) != s {
return fmt.Errorf("unexpected uncompressed string %q. Expecting %q", unZstdedS, s)
}
return nil
}
func TestZstdCompressSerial(t *testing.T) {
t.Parallel()
if err := testZstdCompress(); err != nil {
t.Fatal(err)
}
}
func TestZstdCompressConcurrent(t *testing.T) {
t.Parallel()
if err := testConcurrent(10, testZstdCompress); err != nil {
t.Fatal(err)
}
}
func testZstdCompress() error {
for _, s := range compressTestcases {
if err := testZstdCompressSingleCase(s); err != nil {
return err
}
}
return nil
}
func testZstdCompressSingleCase(s string) error {
var buf bytes.Buffer
zw := acquireStacklessZstdWriter(&buf, CompressZstdDefault)
if _, err := zw.Write([]byte(s)); err != nil {
return fmt.Errorf("unexpected error: %w. s=%q", err, s)
}
releaseStacklessZstdWriter(zw, CompressZstdDefault)
zr, err := acquireZstdReader(&buf)
if err != nil {
return fmt.Errorf("unexpected error: %w. s=%q", err, s)
}
body, err := io.ReadAll(zr)
if err != nil {
return fmt.Errorf("unexpected error: %w. s=%q", err, s)
}
if string(body) != s {
return fmt.Errorf("unexpected string after decompression: %q. Expecting %q", body, s)
}
releaseZstdReader(zr)
return nil
}
|
