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
|
//nolint:testpackage
package bufpool
import "testing"
func TestBorrowBufferResizeAndAppend(t *testing.T) {
t.Parallel()
b := Borrow(1)
defer b.Release()
if cap(b.buf) < DefaultBufferCap {
t.Fatalf("expected capacity >= %d, got %d", DefaultBufferCap, cap(b.buf))
}
b.Append([]byte("alpha"))
b.Append([]byte("beta"))
if got := string(b.Bytes()); got != "alphabeta" {
t.Fatalf("unexpected contents: %q", got)
}
b.Resize(3)
if got := string(b.Bytes()); got != "alp" {
t.Fatalf("resize shrink mismatch: %q", got)
}
b.Resize(8)
if len(b.Bytes()) != 8 {
t.Fatalf("expected len 8 after grow, got %d", len(b.Bytes()))
}
if prefix := string(b.Bytes()[:3]); prefix != "alp" {
t.Fatalf("prefix lost after grow: %q", prefix)
}
}
func TestBorrowBufferRelease(t *testing.T) {
t.Parallel()
b := Borrow(DefaultBufferCap / 2)
b.Append([]byte("data"))
b.Release()
if b.buf != nil {
t.Fatal("expected buffer cleared after release")
}
}
func TestBorrowUsesLargerPools(t *testing.T) {
t.Parallel()
const request = DefaultBufferCap * 4
classIdx, classCap, pooled := classFor(request)
if !pooled {
t.Fatalf("expected %d to map to a pooled class", request)
}
b := Borrow(request)
if b.pool != poolIndex(classIdx) { //#nosec:G115
t.Fatalf("expected pooled buffer in class %d, got %d", classIdx, b.pool)
}
if cap(b.buf) != classCap {
t.Fatalf("expected capacity %d, got %d", classCap, cap(b.buf))
}
b.Release()
b2 := Borrow(request)
defer b2.Release()
if b2.pool != poolIndex(classIdx) { //#nosec:G115
t.Fatalf("expected pooled buffer in class %d on reuse, got %d", classIdx, b2.pool)
}
if cap(b2.buf) != classCap {
t.Fatalf("expected capacity %d on reuse, got %d", classCap, cap(b2.buf))
}
}
func TestGrowingBufferStaysPooled(t *testing.T) {
t.Parallel()
b := Borrow(DefaultBufferCap)
defer b.Release()
b.Append(make([]byte, DefaultBufferCap*3))
if b.pool == unpooled {
t.Fatal("buffer should stay pooled after growth within limit")
}
}
|
