test: add return eval

hd1fernando · May 23, 2023 · e1bc9ea · e1bc9ea
1 parent 87f5621
commit e1bc9ea
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/DotMonkey.ParserTest/EvaluatorTest.cs b/tests/DotMonkey.ParserTest/EvaluatorTest.cs
@@ -97,6 +97,19 @@ public void TestIfElsExpression(string input, object expected)
         }
     }
 
+
+    [Theory(DisplayName = "Return Statement")]
+    [InlineData("return 10;", 10)]
+    [InlineData("return 10; 9;", 10)]
+    [InlineData("return 2 * 5; 9;", 10)]
+    [InlineData("9; return 2 * 5; 9;", 10)]
+    [InlineData("if (10 > 1) { if (10 > 1) { return 10; } return 1; }", 10)]
+    public void TestReturnStatement(string input, int expected)
+    {
+        var evaluated = TestEval(input);
+        TestIntergerObject(evaluated, expected);
+    }
+
     private void TestNullObject(IObject evalueated)
     {
         evalueated.Should().BeOfType<NULL>();