defer seems to be fully working

1 files changed, 23 insertions, 1 deletions

diff --git a/test.toc b/test.toc
index abeebd1..28e639e 100644
--- a/test.toc
+++ b/test.toc
@@ -6,9 +6,31 @@ main ::= fn() {
 		defer io.puts("deferred from for");
 		io.puti(i);
 		if i == 2 {
+			defer io.puts("deferred from if1");
+			defer io.puts("deferred from if2");
+			defer io.puts("deferred from if3");
+			defer io.puts("deferred from if4");
+			defer io.puts("deferred from if5");
+			defer io.puts("deferred from if6");
+			defer io.puts("deferred from if7");
+			defer io.puts("deferred from if8");
+			continue;
+		}
+		if i == 8 {
+			break;
+		}
+	}
+	i := 0;
+	while {
+		defer io.puts("deferred from while");
+		i += 1;
+		io.puti(i);
+		if i % 2 == 0 { continue; }
+		if i == 7 {
 			defer io.puts("deferred from if");
-			return;
+			break;
 		}
+
 	}
 	io.puts("end of main()");
 }