63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
|
package luar
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/yuin/gopher-lua"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_slice(t *testing.T) {
|
||
|
|
L := lua.NewState()
|
||
|
|
defer L.Close()
|
||
|
|
|
||
|
|
things := []string{
|
||
|
|
"cake",
|
||
|
|
"wallet",
|
||
|
|
"calendar",
|
||
|
|
"phone",
|
||
|
|
"speaker",
|
||
|
|
}
|
||
|
|
|
||
|
|
L.SetGlobal("things", New(L, things))
|
||
|
|
|
||
|
|
testReturn(t, L, `return #things, things[1], things[5]`, "5", "cake", "speaker")
|
||
|
|
if err := L.DoString(`things[1] = "cookie"`); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if things[0] != "cookie" {
|
||
|
|
t.Fatalf(`expected things[0] = "cookie", got %s`, things[0])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_slice_2(t *testing.T) {
|
||
|
|
L := lua.NewState()
|
||
|
|
defer L.Close()
|
||
|
|
|
||
|
|
items := make([]string, 0, 10)
|
||
|
|
|
||
|
|
L.SetGlobal("items", New(L, items))
|
||
|
|
|
||
|
|
testReturn(t, L, `return #items`, "0")
|
||
|
|
testReturn(t, L, `items = items + "hello" + "world"; return #items`, "2")
|
||
|
|
testReturn(t, L, `return items[1]`, "hello")
|
||
|
|
testReturn(t, L, `return items[2]`, "world")
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_slice_iterator(t *testing.T) {
|
||
|
|
L := lua.NewState()
|
||
|
|
defer L.Close()
|
||
|
|
|
||
|
|
s := []string{
|
||
|
|
"hello",
|
||
|
|
"there",
|
||
|
|
}
|
||
|
|
|
||
|
|
e := []string{}
|
||
|
|
|
||
|
|
L.SetGlobal("s", New(L, s))
|
||
|
|
L.SetGlobal("e", New(L, e))
|
||
|
|
|
||
|
|
testReturn(t, L, `local itr = s(); local a, b = itr(); local c, d = itr(); return a, b, c, d`, "1", "hello", "2", "there")
|
||
|
|
testReturn(t, L, `local itr = e(); local a, b = itr(); return a, b`, "nil", "nil")
|
||
|
|
}
|