mirror of
https://github.com/fumiama/orbyte.git
synced 2026-06-05 10:10:30 +08:00
52 lines
1.0 KiB
Go
52 lines
1.0 KiB
Go
package pbuf
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"io"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/fumiama/orbyte"
|
|
)
|
|
|
|
func TestBuffer(t *testing.T) {
|
|
testBuffer(NewBuffer(nil), t)
|
|
testBuffer(NewBuffer(make([]byte, 0, 8192)), t)
|
|
testBuffer(ParseBuffer(bytes.NewBuffer(nil)), t)
|
|
testBuffer(ParseBuffer(bytes.NewBuffer(make([]byte, 0, 8192))), t)
|
|
testBuffer(InvolveBuffer(bytes.NewBuffer(nil)), t)
|
|
testBuffer(InvolveBuffer(bytes.NewBuffer(make([]byte, 0, 8192))), t)
|
|
}
|
|
|
|
func testBuffer(buf *orbyte.Item[bytes.Buffer], t *testing.T) {
|
|
buf.P(func(buf *bytes.Buffer) {
|
|
if buf.Len() != 4096 {
|
|
io.CopyN(buf, rand.Reader, 4096)
|
|
if buf.Len() != 4096 {
|
|
t.Fatal("got", buf.Len())
|
|
}
|
|
}
|
|
})
|
|
bufcp := buf.Copy()
|
|
dat := buf.Trans()
|
|
bufcp.P(func(bufcp *bytes.Buffer) {
|
|
if bufcp.Len() != 4096 {
|
|
t.Fatal("got", bufcp.Len())
|
|
}
|
|
if !bytes.Equal(bufcp.Bytes(), dat.Bytes()) {
|
|
t.Fatal("unexpected")
|
|
}
|
|
})
|
|
|
|
runtime.GC()
|
|
runtime.Gosched()
|
|
runtime.GC()
|
|
|
|
out, in := bufferPool.p.CountItems()
|
|
t.Log(out, in)
|
|
if out != 0 {
|
|
t.Fail()
|
|
}
|
|
}
|