buffer_test: make use of the new API.
parent
0f6db15c59
commit
6e5247d16d
|
@ -101,7 +101,7 @@ func TestOutMessageAppend(t *testing.T) {
|
||||||
om.Append(wantPayload[4:])
|
om.Append(wantPayload[4:])
|
||||||
|
|
||||||
// The result should be a zeroed header followed by the desired payload.
|
// The result should be a zeroed header followed by the desired payload.
|
||||||
const wantLen = int(OutMessageInitialSize) + len(wantPayloadStr)
|
const wantLen = OutMessageHeaderSize + len(wantPayloadStr)
|
||||||
|
|
||||||
if got, want := om.Len(), wantLen; got != want {
|
if got, want := om.Len(), wantLen; got != want {
|
||||||
t.Errorf("om.Len() = %d, want %d", got, want)
|
t.Errorf("om.Len() = %d, want %d", got, want)
|
||||||
|
@ -113,7 +113,7 @@ func TestOutMessageAppend(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
want := append(
|
want := append(
|
||||||
make([]byte, OutMessageInitialSize),
|
make([]byte, OutMessageHeaderSize),
|
||||||
wantPayload...)
|
wantPayload...)
|
||||||
|
|
||||||
if !bytes.Equal(b, want) {
|
if !bytes.Equal(b, want) {
|
||||||
|
@ -131,7 +131,7 @@ func TestOutMessageAppendString(t *testing.T) {
|
||||||
om.AppendString(wantPayload[4:])
|
om.AppendString(wantPayload[4:])
|
||||||
|
|
||||||
// The result should be a zeroed header followed by the desired payload.
|
// The result should be a zeroed header followed by the desired payload.
|
||||||
const wantLen = int(OutMessageInitialSize) + len(wantPayload)
|
const wantLen = OutMessageHeaderSize + len(wantPayload)
|
||||||
|
|
||||||
if got, want := om.Len(), wantLen; got != want {
|
if got, want := om.Len(), wantLen; got != want {
|
||||||
t.Errorf("om.Len() = %d, want %d", got, want)
|
t.Errorf("om.Len() = %d, want %d", got, want)
|
||||||
|
@ -143,7 +143,7 @@ func TestOutMessageAppendString(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
want := append(
|
want := append(
|
||||||
make([]byte, OutMessageInitialSize),
|
make([]byte, OutMessageHeaderSize),
|
||||||
wantPayload...)
|
wantPayload...)
|
||||||
|
|
||||||
if !bytes.Equal(b, want) {
|
if !bytes.Equal(b, want) {
|
||||||
|
@ -159,10 +159,10 @@ func TestOutMessageShrinkTo(t *testing.T) {
|
||||||
om.AppendString("burrito")
|
om.AppendString("burrito")
|
||||||
|
|
||||||
// Shrink it.
|
// Shrink it.
|
||||||
om.ShrinkTo(OutMessageInitialSize + uintptr(len("taco")))
|
om.ShrinkTo(OutMessageHeaderSize + len("taco"))
|
||||||
|
|
||||||
// The result should be a zeroed header followed by "taco".
|
// The result should be a zeroed header followed by "taco".
|
||||||
const wantLen = int(OutMessageInitialSize) + len("taco")
|
const wantLen = OutMessageHeaderSize + len("taco")
|
||||||
|
|
||||||
if got, want := om.Len(), wantLen; got != want {
|
if got, want := om.Len(), wantLen; got != want {
|
||||||
t.Errorf("om.Len() = %d, want %d", got, want)
|
t.Errorf("om.Len() = %d, want %d", got, want)
|
||||||
|
@ -174,7 +174,7 @@ func TestOutMessageShrinkTo(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
want := append(
|
want := append(
|
||||||
make([]byte, OutMessageInitialSize),
|
make([]byte, OutMessageHeaderSize),
|
||||||
"taco"...)
|
"taco"...)
|
||||||
|
|
||||||
if !bytes.Equal(b, want) {
|
if !bytes.Equal(b, want) {
|
||||||
|
@ -233,7 +233,7 @@ func TestOutMessageReset(t *testing.T) {
|
||||||
om.Reset()
|
om.Reset()
|
||||||
|
|
||||||
// Check that the length was updated.
|
// Check that the length was updated.
|
||||||
if got, want := int(om.Len()), int(OutMessageInitialSize); got != want {
|
if got, want := om.Len(), OutMessageHeaderSize; got != want {
|
||||||
t.Fatalf("om.Len() = %d, want %d", got, want)
|
t.Fatalf("om.Len() = %d, want %d", got, want)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -269,7 +269,7 @@ func TestOutMessageGrow(t *testing.T) {
|
||||||
t.Fatalf("fillWithGarbage: %v", err)
|
t.Fatalf("fillWithGarbage: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
om.ShrinkTo(OutMessageInitialSize)
|
om.ShrinkTo(OutMessageHeaderSize)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Call Grow.
|
// Call Grow.
|
||||||
|
@ -278,7 +278,7 @@ func TestOutMessageGrow(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check the resulting length in two ways.
|
// Check the resulting length in two ways.
|
||||||
const wantLen = int(payloadSize + OutMessageInitialSize)
|
const wantLen = payloadSize + OutMessageHeaderSize
|
||||||
if got, want := om.Len(), wantLen; got != want {
|
if got, want := om.Len(), wantLen; got != want {
|
||||||
t.Errorf("om.Len() = %d, want %d", got)
|
t.Errorf("om.Len() = %d, want %d", got)
|
||||||
}
|
}
|
||||||
|
@ -289,7 +289,7 @@ func TestOutMessageGrow(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check that the payload was zeroed.
|
// Check that the payload was zeroed.
|
||||||
for i, x := range b[OutMessageInitialSize:] {
|
for i, x := range b[OutMessageHeaderSize:] {
|
||||||
if x != 0 {
|
if x != 0 {
|
||||||
t.Fatalf("non-zero byte 0x%02x at payload offset %d", x, i)
|
t.Fatalf("non-zero byte 0x%02x at payload offset %d", x, i)
|
||||||
}
|
}
|
||||||
|
@ -331,7 +331,7 @@ func BenchmarkOutMessageGrowShrink(b *testing.B) {
|
||||||
var om OutMessage
|
var om OutMessage
|
||||||
for i := 0; i < b.N; i++ {
|
for i := 0; i < b.N; i++ {
|
||||||
om.Grow(MaxReadSize)
|
om.Grow(MaxReadSize)
|
||||||
om.ShrinkTo(OutMessageInitialSize)
|
om.ShrinkTo(OutMessageHeaderSize)
|
||||||
}
|
}
|
||||||
|
|
||||||
b.SetBytes(int64(MaxReadSize))
|
b.SetBytes(int64(MaxReadSize))
|
||||||
|
@ -349,7 +349,7 @@ func BenchmarkOutMessageGrowShrink(b *testing.B) {
|
||||||
|
|
||||||
for i := 0; i < b.N; i++ {
|
for i := 0; i < b.N; i++ {
|
||||||
oms[i%numMessages].Grow(MaxReadSize)
|
oms[i%numMessages].Grow(MaxReadSize)
|
||||||
oms[i%numMessages].ShrinkTo(OutMessageInitialSize)
|
oms[i%numMessages].ShrinkTo(OutMessageHeaderSize)
|
||||||
}
|
}
|
||||||
|
|
||||||
b.SetBytes(int64(MaxReadSize))
|
b.SetBytes(int64(MaxReadSize))
|
||||||
|
|
Loading…
Reference in New Issue