core: add tests for local extension of System.Storage.* interops

A part of #3979.

Signed-off-by: Tural Devrishev <[email protected]>

Author: Tural Devrishev

pkg/compiler/syscall_test.go

@@ -9,7 +9,9 @@ import (
 	"testing"
 
 	"github.com/nspcc-dev/neo-go/pkg/compiler"
+	"github.com/nspcc-dev/neo-go/pkg/config"
 	"github.com/nspcc-dev/neo-go/pkg/core"
+	"github.com/nspcc-dev/neo-go/pkg/core/block"
 	"github.com/nspcc-dev/neo-go/pkg/core/interop"
 	"github.com/nspcc-dev/neo-go/pkg/core/interop/interopnames"
 	istorage "github.com/nspcc-dev/neo-go/pkg/core/interop/storage"
@@ -95,12 +97,21 @@ func TestSyscallExecution(t *testing.T) {
 		"storage.Get":                      {interopnames.SystemStorageGet, []string{sctx, b}, false},
 		"storage.GetContext":               {interopnames.SystemStorageGetContext, nil, false},
 		"storage.GetReadOnlyContext":       {interopnames.SystemStorageGetReadOnlyContext, nil, false},
+		"storage.LocalDelete":              {interopnames.SystemStorageLocalDelete, []string{b}, true},
+		"storage.LocalFind":                {interopnames.SystemStorageLocalFind, []string{b, "storage.None"}, false},
+		"storage.LocalGet":                 {interopnames.SystemStorageLocalGet, []string{b}, false},
+		"storage.LocalPut":                 {interopnames.SystemStorageLocalPut, []string{b, b}, true},
 		"storage.Put":                      {interopnames.SystemStoragePut, []string{sctx, b, b}, true},
 		"storage.ConvertContextToReadOnly": {interopnames.SystemStorageAsReadOnly, []string{sctx}, false},
 		"crypto.CheckMultisig":             {interopnames.SystemCryptoCheckMultisig, []string{pubs, sigs}, false},
 		"crypto.CheckSig":                  {interopnames.SystemCryptoCheckSig, []string{pub, sig}, false},
 	}
-	ic := &interop.Context{}
+	ic := &interop.Context{
+		Hardforks: map[string]uint32{
+			config.HFFaun.String(): 0,
+		},
+		Block: &block.Block{},
+	}
 	core.SpawnVM(ic) // set Functions field
 	for _, fs := range ic.Functions {
 		// It will be set in test and we want to fail if calling invalid syscall.

pkg/core/interop/storage/storage_test.go

@@ -2,8 +2,11 @@ package storage_test
 
 import (
 	"math/big"
+	"strings"
 	"testing"
 
+	"github.com/nspcc-dev/neo-go/pkg/compiler"
+	"github.com/nspcc-dev/neo-go/pkg/config"
 	"github.com/nspcc-dev/neo-go/pkg/config/limits"
 	"github.com/nspcc-dev/neo-go/pkg/core"
 	"github.com/nspcc-dev/neo-go/pkg/core/block"
@@ -14,6 +17,7 @@ import (
 	"github.com/nspcc-dev/neo-go/pkg/core/state"
 	"github.com/nspcc-dev/neo-go/pkg/core/transaction"
 	"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
+	"github.com/nspcc-dev/neo-go/pkg/neotest"
 	"github.com/nspcc-dev/neo-go/pkg/neotest/chain"
 	"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
 	"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
@@ -334,6 +338,52 @@ func createVMAndContractState(t testing.TB) (*vm.VM, *state.Contract, *interop.C
 	return v, contractState, context, chain
 }
 
+func TestStorage_LocalInteropAPI(t *testing.T) {
+	bc, acc := chain.NewSingleWithCustomConfig(t, func(c *config.Blockchain) {
+		c.Hardforks = map[string]uint32{
+			config.HFFaun.String(): 6,
+		}
+	})
+	e := neotest.NewExecutor(t, bc, acc, acc)
+
+	src := `package foo
+    import (
+      "github.com/nspcc-dev/neo-go/pkg/interop/storage"
+      "github.com/nspcc-dev/neo-go/pkg/interop/iterator"
+    )
+
+    func LocalPut() {
+      storage.LocalPut([]byte("key"), []byte("val"))
+    }
+
+    func LocalGet() any {
+      return storage.LocalGet([]byte("key"))
+    }
+
+    func LocalDelete() {
+      storage.LocalDelete([]byte("key"))
+    }
+
+    func LocalFind() bool {
+      return iterator.Next(storage.LocalFind([]byte("key"), storage.None))
+    }`
+
+	ctr := neotest.CompileSource(t, e.Validator.ScriptHash(), strings.NewReader(src), &compiler.Options{
+		Name: "testpolicy_contract",
+	})
+	e.DeployContract(t, ctr, nil)
+
+	ctrInvoker := e.NewInvoker(ctr.Hash, e.Committee)
+	ctrInvoker.InvokeFail(t, "System.Storage.Local.Put failed: syscall not found", "localPut")
+	ctrInvoker.InvokeFail(t, "System.Storage.Local.Get failed: syscall not found", "localGet")
+	ctrInvoker.InvokeFail(t, "System.Storage.Local.Delete failed: syscall not found", "localDelete")
+	ctrInvoker.InvokeFail(t, "System.Storage.Local.Find failed: syscall not found", "localFind")
+	ctrInvoker.Invoke(t, nil, "localPut")
+	ctrInvoker.Invoke(t, stackitem.Make([]byte("val")), "localGet")
+	ctrInvoker.Invoke(t, nil, "localDelete")
+	ctrInvoker.Invoke(t, stackitem.Make(false), "localFind")
+}
+
 func TestStorage_LocalErrors(t *testing.T) {
 	_, ic, _ := createVM(t)
 	for _, f := range []func(*interop.Context) error{istorage.LocalDelete, istorage.LocalFind, istorage.LocalGet, istorage.LocalPut} {