neo-go/pkg/core/interops_test.go
Evgenii Stratonikov efddcf3bfe core: remove interop methods from interopContext
If interops are defined as a separate functions
they can be implemented in a separate packages
which can help us to structure core.
2020-04-11 10:56:35 +03:00

92 lines
2 KiB
Go

package core
import (
"reflect"
"runtime"
"testing"
"github.com/nspcc-dev/neo-go/pkg/core/dao"
"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(*interopContext, *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(*interopContext, *vm.VM) error{
accountGetBalance,
accountGetScriptHash,
accountGetVotes,
assetGetAdmin,
assetGetAmount,
assetGetAssetID,
assetGetAssetType,
assetGetAvailable,
assetGetIssuer,
assetGetOwner,
assetGetPrecision,
assetRenew,
attrGetData,
attrGetUsage,
blockGetTransaction,
blockGetTransactionCount,
blockGetTransactions,
contractGetScript,
contractGetStorageContext,
contractIsPayable,
headerGetConsensusData,
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)
})
}
}
}