fix while bytecode generation and unit test

b3b00 · Dec 29, 2021 · 815db22 · 815db22
1 parent 2420cf4
commit 815db22
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/ParserTests/samples/IndentedWhileTests.cs b/ParserTests/samples/IndentedWhileTests.cs
@@ -203,6 +203,32 @@ public void TestIfThenElse()
             Assert.Equal("world", (elseAssign.Value as StringConstant).Value);
         }
 
+        [Fact]
+        public void TestNestedIfThenElse()
+        {
+            var program = @"
+# TestIfThenElse
+a := -111
+if true then
+    if true then
+        a := 1
+    else
+        a := 2
+else
+    a := 3
+    b := ""world""
+return a
+";
+            var compiler = new IndentedWhileCompiler();
+            var func = compiler.CompileToFunction(program,true);
+            Assert.NotNull(func);
+            var f = func();
+            Assert.Equal(1, f);
+
+
+        }
+
+
         [Fact]
         public void TestInfiniteWhile()
         {

diff --git a/samples/while/compiler/SemanticChecker.cs b/samples/while/compiler/SemanticChecker.cs
@@ -80,11 +80,11 @@ private void SemanticCheck(IfStatement ast, CompilerContext context)
             ast.CompilerScope = context.CurrentScope;
 
             context.OpenNewScope();
-            SemanticCheck(ast.ThenStmt);
+            SemanticCheck(ast.ThenStmt,context);
             context.CloseScope();
 
             context.OpenNewScope();
-            SemanticCheck(ast.ElseStmt);
+            SemanticCheck(ast.ElseStmt,context);
             context.CloseScope();
         }