mirror of https://github.com/v2ray/v2ray-core
61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package alloc_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
. "github.com/v2ray/v2ray-core/common/alloc"
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
|
)
|
|
|
|
func TestBufferClear(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
buffer := NewBuffer().Clear()
|
|
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 TestBufferIsFull(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
buffer := NewBuffer()
|
|
defer buffer.Release()
|
|
|
|
assert.Bool(buffer.IsFull()).IsTrue()
|
|
|
|
buffer.Clear()
|
|
assert.Bool(buffer.IsFull()).IsFalse()
|
|
}
|
|
|
|
func TestBufferPrepend(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
buffer := NewBuffer().Clear()
|
|
defer buffer.Release()
|
|
|
|
buffer.Append([]byte{'a', 'b', 'c'})
|
|
buffer.Prepend([]byte{'x', 'y', 'z'})
|
|
|
|
assert.Int(buffer.Len()).Equals(6)
|
|
assert.Bytes(buffer.Value).Equals([]byte("xyzabc"))
|
|
|
|
buffer.Prepend([]byte{'u', 'v', 'w'})
|
|
assert.Bytes(buffer.Value).Equals([]byte("uvwxyzabc"))
|
|
}
|
|
|
|
func TestBufferString(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
buffer := NewBuffer().Clear()
|
|
defer buffer.Release()
|
|
|
|
buffer.AppendString("Test String")
|
|
assert.String(buffer.String()).Equals("Test String")
|
|
}
|