TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

compiler: add test for util.AbortMsg and util.AssertMsg

Browse source 
Signed-off-by: Anna Shaleva <shaleva.ann@nspcc.ru>
This commit is contained in:
Anna Shaleva 2023-07-20 11:51:18 +03:00
parent de43b39c2a
commit 5b12be2ac7
1 changed files with 54 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
pkg/compiler/interop_test.go
View file

@ -192,6 +192,60 @@ func TestAbort(t *testing.T) {
require.True(t, v.HasFailed())
}
func TestAbortMsg(t *testing.T) {
src := `package foo
import "github.com/nspcc-dev/neo-go/pkg/interop/util"
func Main() int {
util.AbortMsg("some message")
return 1
}`
v := vmAndCompile(t, src)
err := v.Run()
require.Error(t, err)
require.True(t, v.HasFailed())
require.True(t, strings.Contains(err.Error(), "ABORTMSG is executed. Reason: some message"), err)
}
func TestAssert(t *testing.T) {
src := `package foo
import "github.com/nspcc-dev/neo-go/pkg/interop/util"
func Main(ok bool) int {
util.Assert(ok)
return 1
}`
// assert OK
evalWithArgs(t, src, nil, []stackitem.Item{stackitem.Make(true)}, big.NewInt(1))
// assert FALSE
v := vmAndCompile(t, src)
v.Estack().PushVal(false)
err := v.Run()
require.Error(t, err)
require.True(t, v.HasFailed())
require.True(t, strings.Contains(err.Error(), "ASSERT"), err)
}
func TestAssertMsg(t *testing.T) {
src := `package foo
import "github.com/nspcc-dev/neo-go/pkg/interop/util"
func Main(ok bool) int {
util.AssertMsg(ok, "some message")
return 1
}`
// assert OK
evalWithArgs(t, src, nil, []stackitem.Item{stackitem.Make(true)}, big.NewInt(1))
// assert FALSE
v := vmAndCompile(t, src)
v.Estack().PushVal(false)
err := v.Run()
require.Error(t, err)
require.True(t, v.HasFailed())
require.True(t, strings.Contains(err.Error(), "ASSERTMSG is executed with false result. Reason: some message"), err)
}
func spawnVM(t *testing.T, ic *interop.Context, src string) *vm.VM {
b, di, err := compiler.CompileWithOptions("foo.go", strings.NewReader(src), nil)
require.NoError(t, err)