28 lines
477 B
Go
28 lines
477 B
Go
|
package pool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBytes(t *testing.T) {
|
||
|
|
p := NewPool(func() *bytes.Buffer { return bytes.NewBuffer(nil) })
|
||
|
|
b := p.Get()
|
||
|
|
b.Write([]byte(`test`))
|
||
|
|
if b.String() != "test" {
|
||
|
|
t.Fatal("pool not works")
|
||
|
|
}
|
||
|
|
p.Put(b)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStrings(t *testing.T) {
|
||
|
|
p := NewPool(func() *strings.Builder { return &strings.Builder{} })
|
||
|
|
b := p.Get()
|
||
|
|
b.Write([]byte(`test`))
|
||
|
|
if b.String() != "test" {
|
||
|
|
t.Fatal("pool not works")
|
||
|
|
}
|
||
|
|
p.Put(b)
|
||
|
|
}
|