TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-05-03 23:02:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

compiler: make Notify accept varargs

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-04-05 18:05:40 +03:00
parent c0b5271386
commit efad66aee1
3 changed files with 46 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
pkg/compiler/vm_test.go
View file

@ -53,6 +53,11 @@ func assertResult(t *testing.T, vm *vm.VM, result interface{}) {
}
func vmAndCompile(t *testing.T, src string) *vm.VM {
v, _ := vmAndCompileInterop(t, src)
return v
}
func vmAndCompileInterop(t *testing.T, src string) (*vm.VM, *storagePlugin) {
vm := vm.New()
storePlugin := newStoragePlugin()
@ -61,12 +66,13 @@ func vmAndCompile(t *testing.T, src string) *vm.VM {
b, err := compiler.Compile(strings.NewReader(src))
require.NoError(t, err)
vm.Load(b)
return vm
return vm, storePlugin
}
type storagePlugin struct {
mem map[string][]byte
interops map[uint32]vm.InteropFunc
events []vm.StackItem
}
func newStoragePlugin() *storagePlugin {
@ -77,6 +83,7 @@ func newStoragePlugin() *storagePlugin {
s.interops[vm.InteropNameToID([]byte("Neo.Storage.Get"))] = s.Get
s.interops[vm.InteropNameToID([]byte("Neo.Storage.Put"))] = s.Put
s.interops[vm.InteropNameToID([]byte("Neo.Storage.GetContext"))] = s.GetContext
s.interops[vm.InteropNameToID([]byte("Neo.Runtime.Notify"))] = s.Notify
return s
}
@ -89,6 +96,11 @@ func (s *storagePlugin) getInterop(id uint32) *vm.InteropFuncPrice {
return nil
}
func (s *storagePlugin) Notify(v *vm.VM) error {
s.events = append(s.events, v.Estack().Pop().Item())
return nil
}
func (s *storagePlugin) Delete(vm *vm.VM) error {
vm.Estack().Pop()
key := vm.Estack().Pop().Bytes()