44 lines
1 KiB
Go
44 lines
1 KiB
Go
package runtime_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/d5/tengo/objects"
|
|
)
|
|
|
|
type StringArrayIterator struct {
|
|
objectImpl
|
|
strArr *StringArray
|
|
idx int
|
|
}
|
|
|
|
func (i *StringArrayIterator) TypeName() string {
|
|
return "string-array-iterator"
|
|
}
|
|
|
|
func (i *StringArrayIterator) Next() bool {
|
|
i.idx++
|
|
return i.idx <= len(i.strArr.Value)
|
|
}
|
|
|
|
func (i *StringArrayIterator) Key() objects.Object {
|
|
return &objects.Int{Value: int64(i.idx - 1)}
|
|
}
|
|
|
|
func (i *StringArrayIterator) Value() objects.Object {
|
|
return &objects.String{Value: i.strArr.Value[i.idx-1]}
|
|
}
|
|
|
|
func (o *StringArray) Iterate() objects.Iterator {
|
|
return &StringArrayIterator{
|
|
strArr: o,
|
|
}
|
|
}
|
|
|
|
func TestIterable(t *testing.T) {
|
|
strArr := func() *StringArray { return &StringArray{Value: []string{"one", "two", "three"}} }
|
|
|
|
expectWithSymbols(t, `for i, s in arr { out += i }`, 3, SYM{"arr": strArr()})
|
|
expectWithSymbols(t, `for i, s in arr { out += s }`, "onetwothree", SYM{"arr": strArr()})
|
|
expectWithSymbols(t, `for i, s in arr { out += s + i }`, "one0two1three2", SYM{"arr": strArr()})
|
|
}
|