76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package compiler_test
|
|
|
|
import (
|
|
"fmt"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/compiler"
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/hash"
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAppCall(t *testing.T) {
|
|
srcInner := `
|
|
package foo
|
|
func Main(args []interface{}) int {
|
|
a := args[0].(int)
|
|
b := args[1].(int)
|
|
return a + b
|
|
}
|
|
`
|
|
|
|
inner, err := compiler.Compile(strings.NewReader(srcInner))
|
|
require.NoError(t, err)
|
|
|
|
ih := hash.Hash160(inner)
|
|
getScript := func(u util.Uint160) []byte {
|
|
if u.Equals(ih) {
|
|
return inner
|
|
}
|
|
return nil
|
|
}
|
|
|
|
t.Run("valid script", func(t *testing.T) {
|
|
src := getAppCallScript(fmt.Sprintf("%#v", ih.BytesBE()))
|
|
v := vmAndCompile(t, src)
|
|
v.SetScriptGetter(getScript)
|
|
|
|
require.NoError(t, v.Run())
|
|
|
|
assertResult(t, v, big.NewInt(42))
|
|
})
|
|
|
|
t.Run("missing script", func(t *testing.T) {
|
|
h := ih
|
|
h[0] = ^h[0]
|
|
|
|
src := getAppCallScript(fmt.Sprintf("%#v", h.BytesBE()))
|
|
v := vmAndCompile(t, src)
|
|
v.SetScriptGetter(getScript)
|
|
|
|
require.Error(t, v.Run())
|
|
})
|
|
|
|
t.Run("invalid script address", func(t *testing.T) {
|
|
src := getAppCallScript("[]byte{1, 2, 3}")
|
|
|
|
_, err := compiler.Compile(strings.NewReader(src))
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func getAppCallScript(h string) string {
|
|
return `
|
|
package foo
|
|
import "github.com/CityOfZion/neo-go/pkg/interop/engine"
|
|
func Main() int {
|
|
x := 13
|
|
y := 29
|
|
result := engine.AppCall(` + h + `, []interface{}{x, y})
|
|
return result.(int)
|
|
}
|
|
`
|
|
}
|