From c23c5c47aa6ab36814170ec2918ee01697bda26c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?= <41315874+fumiama@users.noreply.github.com> Date: Sat, 1 Mar 2025 01:04:30 +0900 Subject: [PATCH] feat(pbuf): add copy test --- pbuf/buffer_test.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/pbuf/buffer_test.go b/pbuf/buffer_test.go index 9cbdb32..45fb60a 100644 --- a/pbuf/buffer_test.go +++ b/pbuf/buffer_test.go @@ -17,6 +17,22 @@ func TestBuffer(t *testing.T) { testBuffer(InvolveBuffer(bytes.NewBuffer(make([]byte, 0, 8192))), t) } +func TestUserBufferCopy(t *testing.T) { + p := NewBufferPool[int64]() + buf := p.NewBuffer(nil) + buf.P(func(ub *UserBuffer[int64]) { + ub.DAT = 123456 + ub.WriteString("0987654321") + }) + cpd := buf.Copy().Trans() + if cpd.DAT != 123456 { + t.Fatal("exp", 123456, "got", cpd.DAT) + } + if !bytes.Equal(cpd.Bytes(), []byte("0987654321")) { + t.Fail() + } +} + func testBuffer(buf *OBuffer, t *testing.T) { buf.P(func(buf *Buffer) { if buf.Len() != 4096 {