From acc1a27224ff6a03d34ab4ebf108ffb806a03ae9 Mon Sep 17 00:00:00 2001
From: Hisham Muhammad <hisham@gobolinux.org>
Date: Mon, 16 Oct 2017 16:22:36 -0700
Subject: [PATCH] tests: more type tests

---
 spec/checker_spec.lua | 65 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/spec/checker_spec.lua b/spec/checker_spec.lua
index bf53a08..8e22e5d 100644
--- a/spec/checker_spec.lua
+++ b/spec/checker_spec.lua
@@ -113,6 +113,56 @@ describe("Titan type checker", function()
         assert.truthy(ok)
     end)
 
+    it("type-checks 'while'", function()
+        local code = [[
+            function fn(x: integer): integer
+                local i: integer = 15
+                while x < 100 do
+                    x = x + i
+                end
+                return x
+            end
+        ]]
+        local ast, err = parser.parse(code)
+        local ok, err = checker.check(ast, code, "test.titan")
+        assert.truthy(ok)
+    end)
+
+    it("type-checks 'if'", function()
+        local code = [[
+            function fn(x: integer): integer
+                local i: integer = 15
+                if x < 100 then
+                    x = x + i
+                elseif x > 100 then
+                    x = x - i
+                else
+                    x = 100
+                end
+                return x
+            end
+        ]]
+        local ast, err = parser.parse(code)
+        local ok, err = checker.check(ast, code, "test.titan")
+        assert.truthy(ok)
+    end)
+
+    it("checks code inside the 'while' black", function()
+        local code = [[
+            function fn(x: integer): integer
+                local i: integer = 15
+                while i do
+                    local s: string = i
+                end
+                return x
+            end
+        ]]
+        local ast, err = parser.parse(code)
+        local ok, err = checker.check(ast, code, "test.titan")
+        assert.falsy(ok)
+        assert.match("expected string but found integer", err)
+    end)
+
     it("type-checks 'for' with a step", function()
         local code = [[
             function fn(x: integer): integer
@@ -144,6 +194,21 @@ describe("Titan type checker", function()
         assert.match("'for' start expression", err)
     end)
 
+    it("catches 'for' errors in the control variable", function()
+        local code = [[
+            function fn(x: integer, s: string): integer
+                for i: string = 1, s, 2 do
+                    x = x + i
+                end
+                return x
+            end
+        ]]
+        local ast, err = parser.parse(code)
+        local ok, err = checker.check(ast, code, "test.titan")
+        assert.falsy(ok)
+        assert.match("control variable", err)
+    end)
+
     it("catches 'for' errors in the finish expression", function()
         local code = [[
             function fn(x: integer, s: string): integer