5bf00db2c9
The logic here is that we'll have all binary encoding/decoding done via our io package, which simplifies error handling. This functionality doesn't belong to util, so it's moved. This also expands BufBinWriter with Reset() method to fit the needs of core package.
194 lines
2.7 KiB
Go
194 lines
2.7 KiB
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Mock structure to test getting size of an array of serializable things
|
|
type smthSerializable struct {
|
|
}
|
|
|
|
func (*smthSerializable) DecodeBinary(*io.BinReader) error {
|
|
return nil
|
|
}
|
|
|
|
func (*smthSerializable) EncodeBinary(*io.BinWriter) error {
|
|
return nil
|
|
}
|
|
|
|
func (*smthSerializable) Size() int {
|
|
return 42
|
|
}
|
|
|
|
func TestVarSize(t *testing.T) {
|
|
testCases := []struct {
|
|
variable interface{}
|
|
name string
|
|
expected int
|
|
}{
|
|
{
|
|
252,
|
|
"test_int_1",
|
|
1,
|
|
},
|
|
{
|
|
253,
|
|
"test_int_2",
|
|
3,
|
|
},
|
|
{
|
|
65535,
|
|
"test_int_3",
|
|
3,
|
|
},
|
|
{
|
|
65536,
|
|
"test_int_4",
|
|
5,
|
|
},
|
|
{
|
|
4294967295,
|
|
"test_int_5",
|
|
5,
|
|
},
|
|
{
|
|
uint(252),
|
|
"test_uint_1",
|
|
1,
|
|
},
|
|
{
|
|
uint(253),
|
|
"test_uint_2",
|
|
3,
|
|
},
|
|
{
|
|
uint(65535),
|
|
"test_uint_3",
|
|
3,
|
|
},
|
|
{
|
|
uint(65536),
|
|
"test_uint_4",
|
|
5,
|
|
},
|
|
{
|
|
uint(4294967295),
|
|
"test_uint_5",
|
|
5,
|
|
},
|
|
{
|
|
[]byte{1, 2, 4, 5, 6},
|
|
"test_[]byte_1",
|
|
6,
|
|
},
|
|
{
|
|
// The neo C# implementation doe not allowed this!
|
|
Uint160{1, 2, 4, 5, 6},
|
|
"test_Uint160_1",
|
|
21,
|
|
},
|
|
|
|
{[20]uint8{1, 2, 3, 4, 5, 6},
|
|
"test_uint8_1",
|
|
21,
|
|
},
|
|
{[20]uint8{1, 2, 3, 4, 5, 6, 8, 9},
|
|
"test_uint8_2",
|
|
21,
|
|
},
|
|
|
|
{[32]uint8{1, 2, 3, 4, 5, 6},
|
|
"test_uint8_3",
|
|
33,
|
|
},
|
|
{[10]uint16{1, 2, 3, 4, 5, 6},
|
|
"test_uint16_1",
|
|
21,
|
|
},
|
|
|
|
{[10]uint16{1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_uint16_2",
|
|
21,
|
|
},
|
|
{[30]uint32{1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_uint32_2",
|
|
121,
|
|
},
|
|
{[30]uint64{1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_uint64_2",
|
|
241,
|
|
},
|
|
{[20]int8{1, 2, 3, 4, 5, 6},
|
|
"test_int8_1",
|
|
21,
|
|
},
|
|
{[20]int8{-1, 2, 3, 4, 5, 6, 8, 9},
|
|
"test_int8_2",
|
|
21,
|
|
},
|
|
|
|
{[32]int8{-1, 2, 3, 4, 5, 6},
|
|
"test_int8_3",
|
|
33,
|
|
},
|
|
{[10]int16{-1, 2, 3, 4, 5, 6},
|
|
"test_int16_1",
|
|
21,
|
|
},
|
|
|
|
{[10]int16{-1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_int16_2",
|
|
21,
|
|
},
|
|
{[30]int32{-1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_int32_2",
|
|
121,
|
|
},
|
|
{[30]int64{-1, 2, 3, 4, 5, 6, 10, 21},
|
|
"test_int64_2",
|
|
241,
|
|
},
|
|
// The neo C# implementation doe not allowed this!
|
|
{Uint256{1, 2, 3, 4, 5, 6},
|
|
"test_Uint256_1",
|
|
33,
|
|
},
|
|
|
|
{"abc",
|
|
"test_string_1",
|
|
4,
|
|
},
|
|
{"abcà",
|
|
"test_string_2",
|
|
6,
|
|
},
|
|
{"2d3b96ae1bcc5a585e075e3b81920210dec16302",
|
|
"test_string_3",
|
|
41,
|
|
},
|
|
{[]*smthSerializable{{}, {}},
|
|
"test_Serializable",
|
|
2*42 + 1,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(fmt.Sprintf("run: %s", tc.name), func(t *testing.T) {
|
|
result := GetVarSize(tc.variable)
|
|
assert.Equal(t, tc.expected, result)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestVarSizePanic(t *testing.T) {
|
|
defer func() {
|
|
r := recover()
|
|
assert.NotNil(t, r)
|
|
}()
|
|
|
|
_ = GetVarSize(t)
|
|
}
|