From 000fa5b77bf731dde47bffb471cb79492bf90c69 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vili=20Sinerv=C3=A4?= <vili.m.sinerva@gmail.com>
Date: Fri, 31 Jan 2025 18:28:03 +0200
Subject: [PATCH] Add tests for omitting semicolons

---
 src/compiler/parser/tests.rs | 70 ++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)

diff --git a/src/compiler/parser/tests.rs b/src/compiler/parser/tests.rs
index cb31834..3332905 100644
--- a/src/compiler/parser/tests.rs
+++ b/src/compiler/parser/tests.rs
@@ -429,3 +429,73 @@ fn test_var_chain() {
 fn test_var_embedded() {
     parse(&tokenize("if true then var x = 3"));
 }
+
+#[test]
+fn test_omitting_semicolons() {
+    let result = parse(&tokenize("{ { a } { b } }"));
+    assert_eq!(
+        result,
+        Block(vec![
+            Block(vec![Identifier("a")]),
+            Block(vec![Identifier("b")])
+        ])
+    );
+
+    let result = parse(&tokenize("{ if true then { a } b }"));
+    assert_eq!(
+        result,
+        Block(vec![
+            Conditional(
+                bool_ast!(true),
+                Box::new(Block(vec![Identifier("a")])),
+                None
+            ),
+            Block(vec![Identifier("b")]),
+        ])
+    );
+
+    let result = parse(&tokenize("{ if true then { a }; b }"));
+    assert_eq!(
+        result,
+        Block(vec![
+            Conditional(
+                bool_ast!(true),
+                Box::new(Block(vec![Identifier("a")])),
+                None
+            ),
+            Block(vec![Identifier("b")]),
+        ])
+    );
+
+    let result = parse(&tokenize("{ if true then { a } else { b } c }"));
+    assert_eq!(
+        result,
+        Block(vec![
+            Conditional(
+                bool_ast!(true),
+                Box::new(Block(vec![Identifier("a")])),
+                Some(Box::new(Block(vec![Identifier("b")])))
+            ),
+            Block(vec![Identifier("c")]),
+        ])
+    );
+
+    let result = parse(&tokenize("x = { { f(a) } { b } }"));
+    assert_eq!(
+        result,
+        BinaryOp(
+            id_ast!("x"),
+            "=",
+            Box::new(Block(vec![
+                Block(vec![FunCall("f", vec![Identifier("a")])]),
+                Block(vec![Identifier("b")]),
+            ]))
+        )
+    );
+}
+
+#[test]
+#[should_panic]
+fn test_omitting_semicolons_invalid() {
+    parse(&tokenize("{ if true then { a } b c }"));
+}