83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package compiler_test
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
)
|
|
|
|
func TestInit(t *testing.T) {
|
|
t.Run("Simple", func(t *testing.T) {
|
|
src := `package foo
|
|
var a int
|
|
func init() {
|
|
a = 42
|
|
}
|
|
func Main() int {
|
|
return a
|
|
}`
|
|
eval(t, src, big.NewInt(42))
|
|
})
|
|
t.Run("Multi", func(t *testing.T) {
|
|
src := `package foo
|
|
var m = map[int]int{}
|
|
var a = 2
|
|
func init() {
|
|
m[1] = 11
|
|
}
|
|
func init() {
|
|
a = 1
|
|
m[3] = 30
|
|
}
|
|
func Main() int {
|
|
return m[1] + m[3] + a
|
|
}`
|
|
eval(t, src, big.NewInt(42))
|
|
})
|
|
t.Run("WithCall", func(t *testing.T) {
|
|
src := `package foo
|
|
var m = map[int]int{}
|
|
func init() {
|
|
initMap(m)
|
|
}
|
|
func initMap(m map[int]int) {
|
|
m[11] = 42
|
|
}
|
|
func Main() int {
|
|
return m[11]
|
|
}`
|
|
eval(t, src, big.NewInt(42))
|
|
})
|
|
t.Run("InvalidSignature", func(t *testing.T) {
|
|
src := `package foo
|
|
type Foo int
|
|
var a int
|
|
func (f Foo) init() {
|
|
a = 2
|
|
}
|
|
func Main() int {
|
|
return a
|
|
}`
|
|
eval(t, src, big.NewInt(0))
|
|
})
|
|
}
|
|
|
|
func TestImportOrder(t *testing.T) {
|
|
t.Run("1,2", func(t *testing.T) {
|
|
src := `package foo
|
|
import _ "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg1"
|
|
import _ "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg2"
|
|
import "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg3"
|
|
func Main() int { return pkg3.A }`
|
|
v := vmAndCompile(t, src)
|
|
v.PrintOps()
|
|
eval(t, src, big.NewInt(2))
|
|
})
|
|
t.Run("2,1", func(t *testing.T) {
|
|
src := `package foo
|
|
import _ "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg2"
|
|
import _ "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg1"
|
|
import "github.com/nspcc-dev/neo-go/pkg/compiler/testdata/pkg3"
|
|
func Main() int { return pkg3.A }`
|
|
eval(t, src, big.NewInt(1))
|
|
})
|
|
}
|