mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-25 13:56:35 +00:00
328 lines
9.5 KiB
Go
328 lines
9.5 KiB
Go
package native
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"math"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestStdLibItoaAtoi(t *testing.T) {
|
|
s := newStd()
|
|
ic := &interop.Context{VM: vm.New()}
|
|
var actual stackitem.Item
|
|
|
|
t.Run("itoa-atoi", func(t *testing.T) {
|
|
var testCases = []struct {
|
|
num *big.Int
|
|
base *big.Int
|
|
result string
|
|
}{
|
|
{big.NewInt(0), big.NewInt(10), "0"},
|
|
{big.NewInt(0), big.NewInt(16), "0"},
|
|
{big.NewInt(1), big.NewInt(10), "1"},
|
|
{big.NewInt(-1), big.NewInt(10), "-1"},
|
|
{big.NewInt(1), big.NewInt(16), "1"},
|
|
{big.NewInt(7), big.NewInt(16), "7"},
|
|
{big.NewInt(8), big.NewInt(16), "08"},
|
|
{big.NewInt(65535), big.NewInt(16), "0FFFF"},
|
|
{big.NewInt(15), big.NewInt(16), "0F"},
|
|
{big.NewInt(-1), big.NewInt(16), "F"},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
require.NotPanics(t, func() {
|
|
actual = s.itoa(ic, []stackitem.Item{stackitem.Make(tc.num), stackitem.Make(tc.base)})
|
|
})
|
|
require.Equal(t, stackitem.Make(tc.result), actual)
|
|
|
|
require.NotPanics(t, func() {
|
|
actual = s.atoi(ic, []stackitem.Item{stackitem.Make(tc.result), stackitem.Make(tc.base)})
|
|
})
|
|
require.Equal(t, stackitem.Make(tc.num), actual)
|
|
}
|
|
|
|
t.Run("-1", func(t *testing.T) {
|
|
for _, str := range []string{"FF", "FFF", "FFFF"} {
|
|
require.NotPanics(t, func() {
|
|
actual = s.atoi(ic, []stackitem.Item{stackitem.Make(str), stackitem.Make(16)})
|
|
})
|
|
|
|
require.Equal(t, stackitem.Make(-1), actual)
|
|
}
|
|
})
|
|
})
|
|
|
|
t.Run("itoa error", func(t *testing.T) {
|
|
var testCases = []struct {
|
|
num *big.Int
|
|
base *big.Int
|
|
err error
|
|
}{
|
|
{big.NewInt(1), big.NewInt(13), ErrInvalidBase},
|
|
{big.NewInt(-1), new(big.Int).Add(big.NewInt(math.MaxInt64), big.NewInt(10)), ErrInvalidBase},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
require.PanicsWithError(t, tc.err.Error(), func() {
|
|
_ = s.itoa(ic, []stackitem.Item{stackitem.Make(tc.num), stackitem.Make(tc.base)})
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("atoi error", func(t *testing.T) {
|
|
var testCases = []struct {
|
|
num string
|
|
base *big.Int
|
|
err error
|
|
}{
|
|
{"1", big.NewInt(13), ErrInvalidBase},
|
|
{"1", new(big.Int).Add(big.NewInt(math.MaxInt64), big.NewInt(16)), ErrInvalidBase},
|
|
{"1_000", big.NewInt(10), ErrInvalidFormat},
|
|
{"FE", big.NewInt(10), ErrInvalidFormat},
|
|
{"XD", big.NewInt(16), ErrInvalidFormat},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
require.PanicsWithError(t, tc.err.Error(), func() {
|
|
_ = s.atoi(ic, []stackitem.Item{stackitem.Make(tc.num), stackitem.Make(tc.base)})
|
|
})
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestStdLibJSON(t *testing.T) {
|
|
s := newStd()
|
|
ic := &interop.Context{VM: vm.New()}
|
|
var actual stackitem.Item
|
|
|
|
t.Run("JSONSerialize", func(t *testing.T) {
|
|
t.Run("Good", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.jsonSerialize(ic, []stackitem.Item{stackitem.Make(42)})
|
|
})
|
|
|
|
require.Equal(t, stackitem.Make([]byte("42")), actual)
|
|
})
|
|
|
|
t.Run("Bad", func(t *testing.T) {
|
|
arr := stackitem.NewArray([]stackitem.Item{
|
|
stackitem.NewByteArray(make([]byte, stackitem.MaxSize/2)),
|
|
stackitem.NewByteArray(make([]byte, stackitem.MaxSize/2)),
|
|
})
|
|
require.Panics(t, func() {
|
|
_ = s.jsonSerialize(ic, []stackitem.Item{arr})
|
|
})
|
|
})
|
|
})
|
|
|
|
t.Run("JSONDeserialize", func(t *testing.T) {
|
|
t.Run("Good", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.jsonDeserialize(ic, []stackitem.Item{stackitem.Make("42")})
|
|
})
|
|
|
|
require.Equal(t, stackitem.Make(42), actual)
|
|
})
|
|
t.Run("Bad", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
_ = s.jsonDeserialize(ic, []stackitem.Item{stackitem.Make("{]")})
|
|
})
|
|
require.Panics(t, func() {
|
|
_ = s.jsonDeserialize(ic, []stackitem.Item{stackitem.NewInterop(nil)})
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestStdLibEncodeDecode(t *testing.T) {
|
|
s := newStd()
|
|
original := []byte("my pretty string")
|
|
encoded64 := base64.StdEncoding.EncodeToString(original)
|
|
encoded58 := base58.Encode(original)
|
|
ic := &interop.Context{VM: vm.New()}
|
|
var actual stackitem.Item
|
|
|
|
t.Run("Encode64", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.base64Encode(ic, []stackitem.Item{stackitem.Make(original)})
|
|
})
|
|
require.Equal(t, stackitem.Make(encoded64), actual)
|
|
})
|
|
t.Run("Encode58", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.base58Encode(ic, []stackitem.Item{stackitem.Make(original)})
|
|
})
|
|
require.Equal(t, stackitem.Make(encoded58), actual)
|
|
})
|
|
t.Run("Decode64/positive", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.base64Decode(ic, []stackitem.Item{stackitem.Make(encoded64)})
|
|
})
|
|
require.Equal(t, stackitem.Make(original), actual)
|
|
})
|
|
t.Run("Decode64/error", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
_ = s.base64Decode(ic, []stackitem.Item{stackitem.Make(encoded64 + "%")})
|
|
})
|
|
require.Panics(t, func() {
|
|
_ = s.base64Decode(ic, []stackitem.Item{stackitem.NewInterop(nil)})
|
|
})
|
|
})
|
|
t.Run("Decode58/positive", func(t *testing.T) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.base58Decode(ic, []stackitem.Item{stackitem.Make(encoded58)})
|
|
})
|
|
require.Equal(t, stackitem.Make(original), actual)
|
|
})
|
|
t.Run("Decode58/error", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
_ = s.base58Decode(ic, []stackitem.Item{stackitem.Make(encoded58 + "%")})
|
|
})
|
|
require.Panics(t, func() {
|
|
_ = s.base58Decode(ic, []stackitem.Item{stackitem.NewInterop(nil)})
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestStdLibSerialize(t *testing.T) {
|
|
s := newStd()
|
|
ic := &interop.Context{VM: vm.New()}
|
|
|
|
t.Run("recursive", func(t *testing.T) {
|
|
arr := stackitem.NewArray(nil)
|
|
arr.Append(arr)
|
|
require.Panics(t, func() {
|
|
_ = s.serialize(ic, []stackitem.Item{arr})
|
|
})
|
|
})
|
|
t.Run("big item", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
_ = s.serialize(ic, []stackitem.Item{stackitem.NewByteArray(make([]byte, stackitem.MaxSize))})
|
|
})
|
|
})
|
|
t.Run("good", func(t *testing.T) {
|
|
var (
|
|
actualSerialized stackitem.Item
|
|
actualDeserialized stackitem.Item
|
|
)
|
|
require.NotPanics(t, func() {
|
|
actualSerialized = s.serialize(ic, []stackitem.Item{stackitem.Make(42)})
|
|
})
|
|
|
|
w := io.NewBufBinWriter()
|
|
stackitem.EncodeBinaryStackItem(stackitem.Make(42), w.BinWriter)
|
|
require.NoError(t, w.Err)
|
|
|
|
encoded := w.Bytes()
|
|
require.Equal(t, stackitem.Make(encoded), actualSerialized)
|
|
|
|
require.NotPanics(t, func() {
|
|
actualDeserialized = s.deserialize(ic, []stackitem.Item{actualSerialized})
|
|
})
|
|
require.Equal(t, stackitem.Make(42), actualDeserialized)
|
|
|
|
t.Run("bad", func(t *testing.T) {
|
|
encoded[0] ^= 0xFF
|
|
require.Panics(t, func() {
|
|
_ = s.deserialize(ic, []stackitem.Item{stackitem.Make(encoded)})
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestStdLibSerializeDeserialize(t *testing.T) {
|
|
s := newStd()
|
|
ic := &interop.Context{VM: vm.New()}
|
|
var actual stackitem.Item
|
|
|
|
checkSerializeDeserialize := func(t *testing.T, value interface{}, expected stackitem.Item) {
|
|
require.NotPanics(t, func() {
|
|
actual = s.serialize(ic, []stackitem.Item{stackitem.Make(value)})
|
|
})
|
|
require.NotPanics(t, func() {
|
|
actual = s.deserialize(ic, []stackitem.Item{actual})
|
|
})
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
|
|
t.Run("Bool", func(t *testing.T) {
|
|
checkSerializeDeserialize(t, true, stackitem.NewBool(true))
|
|
})
|
|
t.Run("ByteArray", func(t *testing.T) {
|
|
checkSerializeDeserialize(t, []byte{1, 2, 3}, stackitem.NewByteArray([]byte{1, 2, 3}))
|
|
})
|
|
t.Run("Integer", func(t *testing.T) {
|
|
checkSerializeDeserialize(t, 48, stackitem.NewBigInteger(big.NewInt(48)))
|
|
})
|
|
t.Run("Array", func(t *testing.T) {
|
|
arr := stackitem.NewArray([]stackitem.Item{
|
|
stackitem.Make(true),
|
|
stackitem.Make(123),
|
|
stackitem.NewMap()})
|
|
checkSerializeDeserialize(t, arr, arr)
|
|
})
|
|
t.Run("Struct", func(t *testing.T) {
|
|
st := stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.Make(true),
|
|
stackitem.Make(123),
|
|
stackitem.NewMap(),
|
|
})
|
|
checkSerializeDeserialize(t, st, st)
|
|
})
|
|
t.Run("Map", func(t *testing.T) {
|
|
item := stackitem.NewMap()
|
|
item.Add(stackitem.Make(true), stackitem.Make([]byte{1, 2, 3}))
|
|
item.Add(stackitem.Make([]byte{0}), stackitem.Make(false))
|
|
checkSerializeDeserialize(t, item, item)
|
|
})
|
|
t.Run("Serialize MapCompat", func(t *testing.T) {
|
|
resHex := "480128036b6579280576616c7565"
|
|
res, err := hex.DecodeString(resHex)
|
|
require.NoError(t, err)
|
|
|
|
item := stackitem.NewMap()
|
|
item.Add(stackitem.Make([]byte("key")), stackitem.Make([]byte("value")))
|
|
require.NotPanics(t, func() {
|
|
actual = s.serialize(ic, []stackitem.Item{stackitem.Make(item)})
|
|
})
|
|
bytes, err := actual.TryBytes()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, res, bytes)
|
|
})
|
|
t.Run("Serialize Interop", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
actual = s.serialize(ic, []stackitem.Item{stackitem.NewInterop("kek")})
|
|
})
|
|
})
|
|
t.Run("Serialize Array bad", func(t *testing.T) {
|
|
item := stackitem.NewArray([]stackitem.Item{stackitem.NewBool(true), stackitem.NewBool(true)})
|
|
item.Value().([]stackitem.Item)[1] = item
|
|
require.Panics(t, func() {
|
|
actual = s.serialize(ic, []stackitem.Item{item})
|
|
})
|
|
})
|
|
t.Run("Deserialize unknown", func(t *testing.T) {
|
|
data, err := stackitem.SerializeItem(stackitem.NewBigInteger(big.NewInt(123)))
|
|
require.NoError(t, err)
|
|
|
|
data[0] = 0xFF
|
|
require.Panics(t, func() {
|
|
actual = s.deserialize(ic, []stackitem.Item{stackitem.Make(data)})
|
|
})
|
|
})
|
|
t.Run("Deserialize not a byte array", func(t *testing.T) {
|
|
require.Panics(t, func() {
|
|
actual = s.deserialize(ic, []stackitem.Item{stackitem.NewInterop(nil)})
|
|
})
|
|
})
|
|
}
|