mirror of https://github.com/q191201771/naza
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
// Copyright 2021, Chef. All rights reserved.
|
|
// https://github.com/q191201771/lal
|
|
//
|
|
// Use of this source code is governed by a MIT-style license
|
|
// that can be found in the License file.
|
|
//
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
package nazabytes
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/q191201771/naza/pkg/assert"
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
)
|
|
|
|
func TestBuffer(t *testing.T) {
|
|
golden := []byte("1234567890")
|
|
|
|
b := NewBuffer(8)
|
|
assert.Equal(t, nil, b.Bytes())
|
|
assert.Equal(t, 8, len(b.WritableBytes()))
|
|
assert.Equal(t, 0, b.Len())
|
|
assert.Equal(t, 8, b.Cap())
|
|
|
|
// 简单写读
|
|
b.Grow(5)
|
|
buf := b.WritableBytes()[:5]
|
|
assert.Equal(t, nil, b.Bytes())
|
|
copy(buf, golden[:5])
|
|
b.Flush(5)
|
|
buf = b.Bytes()
|
|
assert.Equal(t, golden[:5], buf)
|
|
assert.Equal(t, 5, b.Len())
|
|
b.Skip(5)
|
|
assert.Equal(t, nil, b.Bytes())
|
|
assert.Equal(t, 8, len(b.WritableBytes()))
|
|
assert.Equal(t, 0, b.Len())
|
|
assert.Equal(t, 8, b.Cap())
|
|
|
|
// 发生扩容
|
|
buf = b.ReserveBytes(10)
|
|
copy(buf, golden)
|
|
b.Flush(10)
|
|
buf = b.Bytes()
|
|
assert.Equal(t, golden, buf)
|
|
b.Skip(10)
|
|
assert.Equal(t, nil, b.Bytes())
|
|
assert.Equal(t, 128, len(b.WritableBytes()))
|
|
assert.Equal(t, 0, b.Len())
|
|
assert.Equal(t, 128, b.Cap())
|
|
|
|
// 利用头部空闲空间扩容
|
|
buf = b.ReserveBytes(10)
|
|
copy(buf, golden)
|
|
b.Flush(10)
|
|
b.Skip(2)
|
|
buf = b.ReserveBytes(7)
|
|
copy(buf, golden[:7])
|
|
b.Flush(7)
|
|
nazalog.Debugf("%s", string(b.Bytes()))
|
|
assert.Equal(t, golden[2:], b.Bytes()[:8])
|
|
assert.Equal(t, golden[:7], b.Bytes()[8:])
|
|
assert.Equal(t, 15, b.Len())
|
|
assert.Equal(t, 128, b.Cap())
|
|
|
|
// Truncate
|
|
b.Reset()
|
|
buf = b.ReserveBytes(10)
|
|
copy(buf, golden)
|
|
b.Flush(10)
|
|
b.Truncate(4)
|
|
assert.Equal(t, golden[:6], b.Bytes())
|
|
|
|
// 特殊值
|
|
b.Reset()
|
|
b.Flush(b.Cap())
|
|
assert.Equal(t, nil, b.WritableBytes())
|
|
|
|
// 一些错误
|
|
b.Reset()
|
|
b.Skip(1)
|
|
assert.Equal(t, nil, b.Bytes())
|
|
b.Truncate(1)
|
|
assert.Equal(t, nil, b.Bytes())
|
|
b.Flush(b.Cap() + 1)
|
|
assert.Equal(t, b.Cap(), b.Len())
|
|
|
|
// 特殊值，极小的扩容
|
|
b = NewBuffer(1)
|
|
buf = b.ReserveBytes(2)
|
|
|
|
b = NewBuffer(4096)
|
|
b.Write(bytes.Repeat([]byte{'1'}, 1400))
|
|
b.Skip(20)
|
|
b.Write(bytes.Repeat([]byte{'1'}, 1400))
|
|
b.Write(bytes.Repeat([]byte{'1'}, 1400))
|
|
b.Bytes()
|
|
assert.Equal(t, 4180, b.Len())
|
|
}
|