tests: add test for 'elseif' generation

titan-lang · Oct 17, 2017 · 0ec8521 · 0ec8521 · mascarenhas · Oct 18, 2017
1 parent 0c1e3ab
commit 0ec8521
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/spec/coder_spec.lua b/spec/coder_spec.lua
@@ -305,6 +305,28 @@ describe("Titan code generator ", function()
         assert.truthy(ok, err)
     end)
 
+    it("generates code for 'elseif'", function()
+        local code = [[
+            function getval(a: integer): integer
+                if a == 1 then
+                    return 10
+                elseif a == 2 then
+                    return 20
+                else
+                    return 30
+                end
+            end
+        ]]
+        local ast, err = parser.parse(code)
+        assert.truthy(ast, err)
+        local ok, err = checker.check(ast, code, "test.titan")
+        assert.truthy(ok, err)
+        local ok, err = generate(ast, "titan_test")
+        assert.truthy(ok, err)
+        local ok, err = call("titan_test", "assert(titan_test.getval(1) == 10);assert(titan_test.getval(2) == 20);assert(titan_test.getval(3) == 30)")
+        assert.truthy(ok, err)
+    end)
+
     pending("handles coercion to integer", function()
         local code = [[
             function fn(): integer