package compiler_test import ( "math/big" "testing" "github.com/nspcc-dev/neo-go/pkg/core/interop/interopnames" istorage "github.com/nspcc-dev/neo-go/pkg/core/interop/storage" "github.com/nspcc-dev/neo-go/pkg/interop/contract" "github.com/nspcc-dev/neo-go/pkg/interop/storage" "github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag" "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" ) // Checks that changes in `smartcontract` are reflected in compiler interop package. func TestCallFlags(t *testing.T) { require.EqualValues(t, contract.ReadStates, callflag.ReadStates) require.EqualValues(t, contract.WriteStates, callflag.WriteStates) require.EqualValues(t, contract.AllowCall, callflag.AllowCall) require.EqualValues(t, contract.AllowNotify, callflag.AllowNotify) require.EqualValues(t, contract.States, callflag.States) require.EqualValues(t, contract.ReadOnly, callflag.ReadOnly) require.EqualValues(t, contract.All, callflag.All) require.EqualValues(t, contract.NoneFlag, callflag.NoneFlag) } func TestFindFlags(t *testing.T) { require.EqualValues(t, storage.None, istorage.FindDefault) require.EqualValues(t, storage.KeysOnly, istorage.FindKeysOnly) require.EqualValues(t, storage.RemovePrefix, istorage.FindRemovePrefix) require.EqualValues(t, storage.ValuesOnly, istorage.FindValuesOnly) require.EqualValues(t, storage.DeserializeValues, istorage.FindDeserialize) require.EqualValues(t, storage.PickField0, istorage.FindPick0) require.EqualValues(t, storage.PickField1, istorage.FindPick1) } func TestStoragePutGet(t *testing.T) { src := ` package foo import "github.com/nspcc-dev/neo-go/pkg/interop/storage" func Main() string { ctx := storage.GetContext() key := []byte("token") storage.Put(ctx, key, []byte("foo")) x := storage.Get(ctx, key) return x.(string) } ` eval(t, src, []byte("foo")) } func TestNotify(t *testing.T) { src := `package foo import "github.com/nspcc-dev/neo-go/pkg/interop/runtime" func Main(arg int) { runtime.Notify("Event1", arg, "sum", arg+1) runtime.Notify("single") }` v, s := vmAndCompileInterop(t, src) v.Estack().PushVal(11) require.NoError(t, v.Run()) require.Equal(t, 2, len(s.events)) exp0 := []stackitem.Item{stackitem.NewBigInteger(big.NewInt(11)), stackitem.NewByteArray([]byte("sum")), stackitem.NewBigInteger(big.NewInt(12))} assert.Equal(t, "Event1", s.events[0].Name) assert.Equal(t, exp0, s.events[0].Item.Value()) assert.Equal(t, "single", s.events[1].Name) assert.Equal(t, []stackitem.Item{}, s.events[1].Item.Value()) } func TestSyscallInGlobalInit(t *testing.T) { src := `package foo import "github.com/nspcc-dev/neo-go/pkg/interop/binary" var a = binary.Base58Decode([]byte("5T")) func Main() []byte { return a }` v, s := vmAndCompileInterop(t, src) s.interops[interopnames.ToID([]byte(interopnames.SystemBinaryBase58Decode))] = func(v *vm.VM) error { s := v.Estack().Pop().Value().([]byte) require.Equal(t, "5T", string(s)) v.Estack().PushVal([]byte{1, 2}) return nil } require.NoError(t, v.Run()) require.Equal(t, []byte{1, 2}, v.Estack().Pop().Value()) }