core: move System.Binary.* interops to binary/ package

Also extend test suite.
2020-12-02 12:29:14 +03:00 · 2020-12-02 12:29:14 +03:00 · d136569ac8
commit d136569ac8
parent 2eb256014e
5 changed files with 155 additions and 100 deletions
--- a/pkg/core/interop/binary/encode_test.go
+++ b/pkg/core/interop/binary/encode_test.go
@ -0,0 +1,92 @@
+package binary
+
+import (
+	"encoding/base64"
+	"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/require"
+)
+
+func TestRuntimeSerialize(t *testing.T) {
+	t.Run("recursive", func(t *testing.T) {
+		arr := stackitem.NewArray(nil)
+		arr.Append(arr)
+		ic := &interop.Context{VM: vm.New()}
+		ic.VM.Estack().PushVal(arr)
+		require.Error(t, Serialize(ic))
+	})
+	t.Run("big item", func(t *testing.T) {
+		ic := &interop.Context{VM: vm.New()}
+		ic.VM.Estack().PushVal(make([]byte, stackitem.MaxSize))
+		require.Error(t, Serialize(ic))
+	})
+	t.Run("good", func(t *testing.T) {
+		ic := &interop.Context{VM: vm.New()}
+		ic.VM.Estack().PushVal(42)
+		require.NoError(t, Serialize(ic))
+
+		w := io.NewBufBinWriter()
+		stackitem.EncodeBinaryStackItem(stackitem.Make(42), w.BinWriter)
+		require.NoError(t, w.Err)
+
+		encoded := w.Bytes()
+		require.Equal(t, encoded, ic.VM.Estack().Pop().Bytes())
+
+		ic.VM.Estack().PushVal(encoded)
+		require.NoError(t, Deserialize(ic))
+		require.Equal(t, big.NewInt(42), ic.VM.Estack().Pop().BigInt())
+
+		t.Run("bad", func(t *testing.T) {
+			encoded[0] ^= 0xFF
+			ic.VM.Estack().PushVal(encoded)
+			require.Error(t, Deserialize(ic))
+		})
+	})
+}
+
+func TestRuntimeEncodeDecode(t *testing.T) {
+	original := []byte("my pretty string")
+	encoded64 := base64.StdEncoding.EncodeToString(original)
+	encoded58 := base58.Encode(original)
+	v := vm.New()
+	ic := &interop.Context{VM: v}
+
+	t.Run("Encode64", func(t *testing.T) {
+		v.Estack().PushVal(original)
+		require.NoError(t, EncodeBase64(ic))
+		actual := v.Estack().Pop().Bytes()
+		require.Equal(t, []byte(encoded64), actual)
+	})
+	t.Run("Encode58", func(t *testing.T) {
+		v.Estack().PushVal(original)
+		require.NoError(t, EncodeBase58(ic))
+		actual := v.Estack().Pop().Bytes()
+		require.Equal(t, []byte(encoded58), actual)
+	})
+	t.Run("Decode64/positive", func(t *testing.T) {
+		v.Estack().PushVal(encoded64)
+		require.NoError(t, DecodeBase64(ic))
+		actual := v.Estack().Pop().Bytes()
+		require.Equal(t, original, actual)
+	})
+	t.Run("Decode64/error", func(t *testing.T) {
+		v.Estack().PushVal(encoded64 + "%")
+		require.Error(t, DecodeBase64(ic))
+	})
+	t.Run("Decode58/positive", func(t *testing.T) {
+		v.Estack().PushVal(encoded58)
+		require.NoError(t, DecodeBase58(ic))
+		actual := v.Estack().Pop().Bytes()
+		require.Equal(t, original, actual)
+	})
+	t.Run("Decode58/error", func(t *testing.T) {
+		v.Estack().PushVal(encoded58 + "%")
+		require.Error(t, DecodeBase58(ic))
+	})
+}