You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
873 B
Go
39 lines
873 B
Go
package iox
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBufferPool(t *testing.T) {
|
|
capacity := 1024
|
|
pool := NewBufferPool(capacity)
|
|
pool.Put(bytes.NewBuffer(make([]byte, 0, 2*capacity)))
|
|
assert.True(t, pool.Get().Cap() <= capacity)
|
|
}
|
|
|
|
func TestBufferPool_Put(t *testing.T) {
|
|
t.Run("with nil buf", func(t *testing.T) {
|
|
pool := NewBufferPool(1024)
|
|
pool.Put(nil)
|
|
val := pool.Get()
|
|
assert.IsType(t, new(bytes.Buffer), val)
|
|
})
|
|
|
|
t.Run("with less-cap buf", func(t *testing.T) {
|
|
pool := NewBufferPool(1024)
|
|
pool.Put(bytes.NewBuffer(make([]byte, 0, 512)))
|
|
val := pool.Get()
|
|
assert.IsType(t, new(bytes.Buffer), val)
|
|
})
|
|
|
|
t.Run("with more-cap buf", func(t *testing.T) {
|
|
pool := NewBufferPool(1024)
|
|
pool.Put(bytes.NewBuffer(make([]byte, 0, 1024<<1)))
|
|
val := pool.Get()
|
|
assert.IsType(t, new(bytes.Buffer), val)
|
|
})
|
|
}
|