Compile ForCond statements

1 files changed, 31 insertions, 0 deletions

diff --git a/pkg/lang/compiler/compiler_test.go b/pkg/lang/compiler/compiler_test.go
index 88413c7..1a759f8 100644
--- a/pkg/lang/compiler/compiler_test.go
+++ b/pkg/lang/compiler/compiler_test.go
@@ -234,6 +234,37 @@ func TestNestedIfs(t *testing.T) {
 	mustCompileTo(t, src, expected)
 }
 
+func TestForCond(t *testing.T) {
+	src := `
+	for true {
+		for false {
+			1
+		}
+	}
+	`
+
+	expected := `
+	@start:
+	push_true
+	jf @end
+
+	@inner_start:
+	push_false
+	jf @inner_end
+
+	push_int 1
+	jmp @inner_start
+
+	@inner_end:
+	jmp @start
+
+	@end
+	halt
+	`
+
+	mustCompileTo(t, src, expected)
+}
+
 func mustCompileTo(t *testing.T, src, expected string) {
 	scanner := scanner.New(strings.NewReader(src))
 	tokens, err := scanner.Scan()