neo-go/pkg/core/interops_test.go

package core

import (
	"reflect"
	"runtime"
	"testing"

	"github.com/nspcc-dev/neo-go/pkg/core/dao"
	"github.com/nspcc-dev/neo-go/pkg/core/interop"
	"github.com/nspcc-dev/neo-go/pkg/core/storage"
	"github.com/nspcc-dev/neo-go/pkg/smartcontract/trigger"
	"github.com/nspcc-dev/neo-go/pkg/vm"
	"github.com/stretchr/testify/require"
)

func testNonInterop(t *testing.T, value interface{}, f func(*interop.Context, *vm.VM) error) {
	v := vm.New()
	v.Estack().PushVal(value)
	chain := newTestChain(t)
	defer chain.Close()
	context := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore()), nil, nil)
	require.Error(t, f(context, v))
}

func TestUnexpectedNonInterops(t *testing.T) {
	vals := map[string]interface{}{
		"int":    1,
		"bool":   false,
		"string": "smth",
		"array":  []int{1, 2, 3},
	}

	// All of these functions expect an interop item on the stack.
	funcs := []func(*interop.Context, *vm.VM) error{
		accountGetBalance,
		accountGetScriptHash,
		assetGetAdmin,
		assetGetAmount,
		assetGetAssetID,
		assetGetAssetType,
		assetGetAvailable,
		assetGetIssuer,
		assetGetOwner,
		assetGetPrecision,
		assetRenew,
		attrGetData,
		attrGetUsage,
		blockGetTransaction,
		blockGetTransactionCount,
		blockGetTransactions,
		contractGetScript,
		contractGetStorageContext,
		contractIsPayable,
		headerGetHash,
		headerGetIndex,
		headerGetMerkleRoot,
		headerGetNextConsensus,
		headerGetPrevHash,
		headerGetTimestamp,
		headerGetVersion,
		inputGetHash,
		inputGetIndex,
		invocationTxGetScript,
		outputGetAssetID,
		outputGetScriptHash,
		outputGetValue,
		storageContextAsReadOnly,
		storageDelete,
		storageFind,
		storageGet,
		storagePut,
		storagePutEx,
		txGetAttributes,
		txGetHash,
		txGetInputs,
		txGetOutputs,
		txGetReferences,
		txGetType,
		txGetUnspentCoins,
		txGetWitnesses,
		witnessGetVerificationScript,
	}
	for _, f := range funcs {
		for k, v := range vals {
			fname := runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name()
			t.Run(k+"/"+fname, func(t *testing.T) {
				testNonInterop(t, v, f)
			})
		}
	}
}