rename alloc to buf

common/buf/buffer_test.go

@@ -0,0 +1,105 @@
+package buf_test
+
+import (
+	"testing"
+
+	. "v2ray.com/core/common/buf"
+	"v2ray.com/core/common/serial"
+	"v2ray.com/core/testing/assert"
+)
+
+func TestBufferClear(t *testing.T) {
+	assert := assert.On(t)
+
+	buffer := NewBuffer()
+	defer buffer.Release()
+
+	payload := "Bytes"
+	buffer.Append([]byte(payload))
+	assert.Int(buffer.Len()).Equals(len(payload))
+
+	buffer.Clear()
+	assert.Int(buffer.Len()).Equals(0)
+}
+
+func TestBufferIsEmpty(t *testing.T) {
+	assert := assert.On(t)
+
+	buffer := NewBuffer()
+	defer buffer.Release()
+
+	assert.Bool(buffer.IsEmpty()).IsTrue()
+}
+
+func TestBufferString(t *testing.T) {
+	assert := assert.On(t)
+
+	buffer := NewBuffer()
+	defer buffer.Release()
+
+	buffer.AppendFunc(serial.WriteString("Test String"))
+	assert.String(buffer.String()).Equals("Test String")
+}
+
+func TestBufferWrite(t *testing.T) {
+	assert := assert.On(t)
+
+	buffer := NewLocalBuffer(8)
+	nBytes, err := buffer.Write([]byte("abcd"))
+	assert.Error(err).IsNil()
+	assert.Int(nBytes).Equals(4)
+	nBytes, err = buffer.Write([]byte("abcde"))
+	assert.Error(err).IsNil()
+	assert.Int(nBytes).Equals(4)
+	assert.String(buffer.String()).Equals("abcdabcd")
+}
+
+func BenchmarkNewBuffer8192(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		buffer := NewBuffer()
+		buffer.Release()
+	}
+}
+
+func BenchmarkNewLocalBuffer8192(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		buffer := NewLocalBuffer(8192)
+		buffer.Release()
+	}
+}
+
+func BenchmarkNewBuffer2048(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		buffer := NewSmallBuffer()
+		buffer.Release()
+	}
+}
+
+func BenchmarkNewLocalBuffer2048(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		buffer := NewLocalBuffer(2048)
+		buffer.Release()
+	}
+}
+
+func BenchmarkBufferValue(b *testing.B) {
+	x := Buffer{}
+	doSomething := func(a Buffer) {
+		_ = a.Len()
+	}
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		doSomething(x)
+	}
+}
+
+func BenchmarkBufferPointer(b *testing.B) {
+	x := NewSmallBuffer()
+	doSomething := func(a *Buffer) {
+		_ = a.Len()
+	}
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		doSomething(x)
+	}
+}