neo-go/pkg/vm/stack_item_test.go

package vm

import (
	"testing"

	"github.com/nspcc-dev/neo-go/pkg/smartcontract"
	"github.com/stretchr/testify/assert"
)

var toContractParameterTestCases = []struct {
	input  StackItem
	result smartcontract.Parameter
}{
	{
		input: NewStructItem([]StackItem{
			NewBigIntegerItem(1),
			NewBoolItem(true),
		}),
		result: smartcontract.Parameter{Type: smartcontract.ArrayType, Value: []smartcontract.Parameter{
			{Type: smartcontract.IntegerType, Value: int64(1)},
			{Type: smartcontract.BoolType, Value: true},
		}},
	},
	{
		input:  NewBoolItem(false),
		result: smartcontract.Parameter{Type: smartcontract.BoolType, Value: false},
	},
	{
		input:  NewByteArrayItem([]byte{0x01, 0x02, 0x03}),
		result: smartcontract.Parameter{Type: smartcontract.ByteArrayType, Value: []byte{0x01, 0x02, 0x03}},
	},
	{
		input: NewArrayItem([]StackItem{NewBigIntegerItem(2), NewBoolItem(true)}),
		result: smartcontract.Parameter{Type: smartcontract.ArrayType, Value: []smartcontract.Parameter{
			{Type: smartcontract.IntegerType, Value: int64(2)},
			{Type: smartcontract.BoolType, Value: true},
		}},
	},
	{
		input:  NewInteropItem(nil),
		result: smartcontract.Parameter{Type: smartcontract.InteropInterfaceType, Value: nil},
	},
	{
		input: &MapItem{value: map[interface{}]StackItem{
			toMapKey(NewBigIntegerItem(1)):               NewBoolItem(true),
			toMapKey(NewByteArrayItem([]byte("qwerty"))): NewBigIntegerItem(3),
			toMapKey(NewBoolItem(true)):                  NewBoolItem(false),
		}},
		result: smartcontract.Parameter{
			Type: smartcontract.MapType,
			Value: map[smartcontract.Parameter]smartcontract.Parameter{
				{Type: smartcontract.IntegerType, Value: int64(1)}:   {Type: smartcontract.BoolType, Value: true},
				{Type: smartcontract.ByteArrayType, Value: "qwerty"}: {Type: smartcontract.IntegerType, Value: int64(3)},
				{Type: smartcontract.BoolType, Value: true}:          {Type: smartcontract.BoolType, Value: false},
			},
		},
	},
}

func TestToContractParameter(t *testing.T) {
	for _, tc := range toContractParameterTestCases {
		seen := make(map[StackItem]bool)
		res := tc.input.ToContractParameter(seen)
		assert.Equal(t, res, tc.result)
	}
}

var fromMapKeyTestCases = []struct {
	input  interface{}
	result StackItem
}{
	{
		input:  true,
		result: NewBoolItem(true),
	},
	{
		input:  int64(4),
		result: NewBigIntegerItem(4),
	},
	{
		input:  "qwerty",
		result: NewByteArrayItem([]byte("qwerty")),
	},
}

func TestFromMapKey(t *testing.T) {
	for _, tc := range fromMapKeyTestCases {
		res := fromMapKey(tc.input)
		assert.Equal(t, res, tc.result)
	}
}