81 lines
2.7 KiB
Go
81 lines
2.7 KiB
Go
package contract
|
|
|
|
import (
|
|
"path"
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/contract-coverage-primer/covertest"
|
|
"github.com/davecgh/go-spew/spew"
|
|
"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/trigger"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const ctrPath = "../impulse"
|
|
|
|
// Key for tests
|
|
var (
|
|
validKey = []byte{1, 2, 3, 4, 5}
|
|
invalidKey = []byte{1, 2, 3}
|
|
)
|
|
|
|
func newExecutor(t *testing.T) *neotest.Executor {
|
|
bc, acc := chain.NewSingle(t)
|
|
return neotest.NewExecutor(t, bc, acc, acc)
|
|
}
|
|
|
|
func TestContract(t *testing.T) {
|
|
e := newExecutor(t)
|
|
ctrDI := covertest.CompileFile(t, e.CommitteeHash, ctrPath, path.Join(ctrPath, "config.yml"))
|
|
ctr := ctrDI.Contract
|
|
e.DeployContract(t, ctr, nil)
|
|
inv := e.CommitteeInvoker(ctr.Hash)
|
|
|
|
// test get without put
|
|
inv.InvokeFail(t, "Cannot get number", "getNumber", validKey)
|
|
|
|
// test put-get with valid key
|
|
inv.Invoke(t, stackitem.Null{}, "putNumber", validKey, 42)
|
|
inv.Invoke(t, 42, "getNumber", validKey)
|
|
|
|
// test invalid key
|
|
inv.InvokeFail(t, "Invalid key size", "putNumber", invalidKey, 42)
|
|
inv.InvokeFail(t, "Invalid key size", "getNumber", invalidKey)
|
|
}
|
|
|
|
func TestRun(t *testing.T) {
|
|
e := newExecutor(t)
|
|
ctrDI := covertest.CompileFile(t, e.CommitteeHash, ctrPath, path.Join(ctrPath, "config.yml"))
|
|
e.DeployContract(t, ctrDI.Contract, nil)
|
|
|
|
// setting up a VM for covertest.Run()
|
|
covertestRunVM := setUpVMForPut(t, e, ctrDI.Contract, false, 101, 2, invalidKey)
|
|
res, err := covertest.Run(covertestRunVM)
|
|
spew.Println("Printing collected instructions:")
|
|
spew.Dump(res)
|
|
spew.Println("covertest.Run() returned an error: ", err)
|
|
|
|
// setting up a VM for vm.Run()
|
|
origRunVM := setUpVMForPut(t, e, ctrDI.Contract, false, 101, 2, invalidKey)
|
|
runerr := origRunVM.Run()
|
|
spew.Println("vm.Run() returned an error: ", err)
|
|
|
|
//check if errors are the same
|
|
spew.Println("Are errors the same? ", runerr.Error() == runerr.Error())
|
|
|
|
//check if the number of elements on the stack is the same
|
|
spew.Println("Is the number of elements on the stack the same? ", origRunVM.Estack().Len() == covertestRunVM.Estack().Len())
|
|
}
|
|
|
|
func setUpVMForPut(t *testing.T, e *neotest.Executor, contract *neotest.Contract, hasResult bool, methodOff int, num int, key []byte) (v *vm.VM) {
|
|
ic, err := e.Chain.GetTestVM(trigger.Application, nil, nil)
|
|
require.NoError(t, err)
|
|
ic.VM.LoadNEFMethod(contract.NEF, contract.Hash, contract.Hash, callflag.All, hasResult, methodOff, -1, nil)
|
|
ic.VM.Context().Estack().PushVal(num)
|
|
ic.VM.Context().Estack().PushVal(key)
|
|
return ic.VM
|
|
}
|