v2ray-core/common/buf/multi_buffer_test.go

package buf_test

import (
	"crypto/rand"
	"testing"

	"v2ray.com/core/common"
	. "v2ray.com/core/common/buf"
	. "v2ray.com/ext/assert"
)

func TestMultiBufferRead(t *testing.T) {
	assert := With(t)

	b1 := New()
	b1.AppendBytes('a', 'b')

	b2 := New()
	b2.AppendBytes('c', 'd')
	mb := NewMultiBufferValue(b1, b2)

	bs := make([]byte, 32)
	nBytes, err := mb.Read(bs)
	assert(err, IsNil)
	assert(nBytes, Equals, 4)
	assert(bs[:nBytes], Equals, []byte("abcd"))
}

func TestMultiBufferAppend(t *testing.T) {
	assert := With(t)

	var mb MultiBuffer
	b := New()
	b.AppendBytes('a', 'b')
	mb.Append(b)
	assert(mb.Len(), Equals, 2)
}

func TestMultiBufferSliceBySizeLarge(t *testing.T) {
	assert := With(t)

	lb := NewSize(8 * 1024)
	common.Must(lb.Reset(ReadFrom(rand.Reader)))

	var mb MultiBuffer
	mb.Append(lb)

	mb2 := mb.SliceBySize(4 * 1024)
	assert(mb2.Len(), Equals, 4*1024)
}