84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package luar
|
|
|
|
import (
|
|
"runtime/debug"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
type StructTestPerson struct {
|
|
Name string
|
|
Age int
|
|
Friend *StructTestPerson
|
|
LastAddSum int
|
|
}
|
|
|
|
func (p StructTestPerson) Hello() string {
|
|
return "Hello, " + p.Name
|
|
}
|
|
|
|
func (p StructTestPerson) String() string {
|
|
return p.Name + " (" + strconv.Itoa(p.Age) + ")"
|
|
}
|
|
|
|
func (p *StructTestPerson) AddNumbers(L *LState) int {
|
|
sum := 0
|
|
for i := L.GetTop(); i >= 1; i-- {
|
|
sum += L.CheckInt(i)
|
|
}
|
|
L.Push(lua.LString(p.Name + " counts: " + strconv.Itoa(sum)))
|
|
p.LastAddSum = sum
|
|
return 1
|
|
}
|
|
|
|
func (p *StructTestPerson) IncreaseAge() {
|
|
p.Age++
|
|
}
|
|
|
|
func testReturn(t *testing.T, L *lua.LState, code string, values ...string) {
|
|
top := L.GetTop()
|
|
if err := L.DoString(code); err != nil {
|
|
t.Fatalf("%s\n\n%s", err, debug.Stack())
|
|
}
|
|
|
|
valid := true
|
|
newTop := L.GetTop()
|
|
|
|
if newTop-top != len(values) {
|
|
valid = false
|
|
} else {
|
|
for i, expect := range values {
|
|
// TODO: strong typing
|
|
val := L.Get(top + i + 1).String()
|
|
if val != expect {
|
|
valid = false
|
|
}
|
|
}
|
|
}
|
|
|
|
if !valid {
|
|
got := make([]string, newTop-top)
|
|
for i := 0; i < len(got); i++ {
|
|
got[i] = L.Get(top + i + 1).String()
|
|
}
|
|
|
|
t.Fatalf("bad return values: expecting %#v, got %#v\n\n%s", values, got, debug.Stack())
|
|
}
|
|
|
|
L.SetTop(top)
|
|
}
|
|
|
|
func testError(t *testing.T, L *lua.LState, code, error string) {
|
|
err := L.DoString(code)
|
|
if err == nil {
|
|
t.Fatalf("expecting error, got nil\n\n%s", debug.Stack())
|
|
}
|
|
|
|
if s := err.Error(); strings.Index(s, error) == -1 {
|
|
t.Fatalf("error substring '%s' not found in '%s'\n\n%s", error, s, debug.Stack())
|
|
}
|
|
}
|