mq_coder/encoding/nvarint_test.go

package encoding

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNInt32(t *testing.T) {
	testNVarInt32(t, -1, []byte{0xFF})
	testNVarInt32(t, -5, []byte{0xFB})
	testNVarInt32(t, 63, []byte{0x3F})
	testNVarInt32(t, -65, []byte{0xBF})
	testNVarInt32(t, 127, []byte{0x7F})
	testNVarInt32(t, 255, []byte{0x00, 0xFF})
	testNVarInt32(t, -4097, []byte{0xEF, 0xFF})
	testNVarInt32(t, -8193, []byte{0xDF, 0xFF})
	testNVarInt32(t, -2097152, []byte{0xE0, 0x00, 0x00})
	testNVarInt32(t, -134217729, []byte{0xF7, 0xFF, 0xFF, 0xFF})
	testNVarInt32(t, -2147483648, []byte{0x80, 0x00, 0x00, 0x00})
}

func TestNUInt32(t *testing.T) {
	testNVarUInt32(t, 1, []byte{0x01})
	testNVarUInt32(t, 127, []byte{0x7F})
	testNVarUInt32(t, 128, []byte{0x00, 0x80})
	testNVarUInt32(t, 130, []byte{0x00, 0x82})
	testNVarUInt32(t, 1048576, []byte{0x10, 0x00, 0x00})
	testNVarUInt32(t, 134217728, []byte{0x08, 0x00, 0x00, 0x00})
	testNVarUInt32(t, 4294967295, []byte{0xFF})
}

func TestNInt64(t *testing.T) {
	testNVarInt64(t, 0, []byte{0x00})
	testNVarInt64(t, 1, []byte{0x01})
	testNVarInt64(t, -1, []byte{0xFF})
}

func TestNUInt64(t *testing.T) {
	testNVarUInt64(t, 0, []byte{0x00})
	testNVarUInt64(t, 1, []byte{0x01})
	testNVarUInt64(t, 18446744073709551615, []byte{0xFF})
}

func testNVarInt32(t *testing.T, value int32, bytes []byte) {
	var msg = fmt.Sprintf("tester %v (%X): %X", value, uint32(value), bytes)
	var size = SizeOfNVarInt32(value)
	assert.Equal(t, len(bytes), size, msg)

	buffer := make([]byte, len(bytes))
	codec := VarCodec{Size: size}
	assert.Nil(t, codec.EncodeNVarInt32(buffer, value), msg)
	assert.Equal(t, bytes, buffer, msg)

	var val int32
	codec = VarCodec{Size: len(bytes)}
	assert.Nil(t, codec.DecodeNVarInt32(bytes, &val), msg)
	assert.Equal(t, value, val, msg)
}

func testNVarUInt32(t *testing.T, value uint32, bytes []byte) {
	var msg = fmt.Sprintf("tester %v (%X): %X", value, value, bytes)
	var size = SizeOfNVarUInt32(value)
	assert.Equal(t, len(bytes), size, msg)

	buffer := make([]byte, len(bytes))
	codec := VarCodec{Size: size}
	assert.Nil(t, codec.EncodeNVarUInt32(buffer, value), msg)
	assert.Equal(t, bytes, buffer, msg)

	var val uint32
	codec = VarCodec{Size: len(bytes)}
	assert.Nil(t, codec.DecodeNVarUInt32(bytes, &val), msg)
	assert.Equal(t, value, val, msg)
}

func testNVarInt64(t *testing.T, value int64, bytes []byte) {
	var msg = fmt.Sprintf("tester %v (%X): %X", value, uint64(value), bytes)
	var size = SizeOfNVarInt64(value)
	assert.Equal(t, len(bytes), size, msg)

	buffer := make([]byte, len(bytes))
	codec := VarCodec{Size: size}
	assert.Nil(t, codec.EncodeNVarInt64(buffer, value), msg)
	assert.Equal(t, bytes, buffer, msg)

	var val int64
	codec = VarCodec{Size: len(bytes)}
	assert.Nil(t, codec.DecodeNVarInt64(bytes, &val), msg)
	assert.Equal(t, value, val, msg)
}

func testNVarUInt64(t *testing.T, value uint64, bytes []byte) {
	var msg = fmt.Sprintf("tester %v (%X): %X", value, value, bytes)
	var size = SizeOfNVarUInt64(value)
	assert.Equal(t, len(bytes), size, msg)

	buffer := make([]byte, len(bytes))
	codec := VarCodec{Size: size}
	assert.Nil(t, codec.EncodeNVarUInt64(buffer, value), msg)
	assert.Equal(t, bytes, buffer, msg)

	var val uint64
	codec = VarCodec{Size: len(bytes)}
	assert.Nil(t, codec.DecodeNVarUInt64(bytes, &val), msg)
	assert.Equal(t, value, val)
}