diff options
Diffstat (limited to 'pkg/lang/compiler/compiler_test.go')
| -rw-r--r-- | pkg/lang/compiler/compiler_test.go | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/pkg/lang/compiler/compiler_test.go b/pkg/lang/compiler/compiler_test.go index 4fd4bac..326ad68 100644 --- a/pkg/lang/compiler/compiler_test.go +++ b/pkg/lang/compiler/compiler_test.go @@ -291,6 +291,60 @@ func TestForCond(t *testing.T) { mustCompileTo(t, src, expected) } +func TestForIn(t *testing.T) { + src := ` + for x in [1, 2, 3] { + "say"(x) + } + ` + + expected := ` + push_array + + get_local 0 + get_member "$push" + push_int 1 + call 1 + + get_local 0 + get_member "$push" + push_int 2 + call 1 + + get_local 0 + get_member "$push" + push_int 3 + call 1 + + push_int 0 + push_null + + @check: + get_local 1 + get_local 0 + get_member "$length" + call 0 + lt + + jf @end + + get_local 0 + get_local 1 + index + set_local 2 + + push_string "say" + get_local 2 + call 1 + + jmp @check + @end: + halt + ` + + mustCompileTo(t, src, expected) +} + func mustCompileTo(t *testing.T, src, expected string) { scanner := scanner.New(strings.NewReader(src)) tokens, err := scanner.Scan() |