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
|
package bpool
import (
"bytes"
"testing"
)
func TestBufferPool(t *testing.T) {
var size int = 4
bufPool := NewBufferPool(size)
// Test Get/Put
b := bufPool.Get()
bufPool.Put(b)
// Add some additional buffers beyond the pool size.
for i := 0; i < size*2; i++ {
bufPool.Put(bytes.NewBuffer([]byte{}))
}
// Close the channel so we can iterate over it.
close(bufPool.c)
// Check the size of the pool.
if len(bufPool.c) != size {
t.Fatalf("bufferpool size invalid: got %v want %v", len(bufPool.c), size)
}
}
|
