tyck: Add testcases mentioned in #677

jinko-core · Sep 13, 2024 · 649c428 · 649c428
1 parent 3edfcce
commit 649c428
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/typecheck/src/lib.rs b/typecheck/src/lib.rs
@@ -665,4 +665,62 @@ mod tests {
 
         assert!(fir.is_ok());
     }
+
+    #[test]
+    fn union_with_constant_valid() {
+        let ast = ast! {
+            type Nothing;
+            type NullableInt = int | Nothing;
+
+            func g(n: NullableInt) {}
+
+            where x = 16;
+
+            g(15);
+            g(x);
+            g(Nothing);
+        };
+
+        let fir = fir!(ast).type_check();
+
+        assert!(fir.is_ok());
+    }
+
+    #[test]
+    fn union_with_constant_invalid() {
+        let ast = ast! {
+            type Nothing;
+            type NullableInt = 15 | Nothing;
+
+            func g(n: NullableInt) {}
+
+            where x = 16;
+
+            g(15);
+            g(x);
+            g(Nothing);
+        };
+
+        let fir = fir!(ast).type_check();
+
+        assert!(fir.is_err());
+    }
+
+    #[test]
+    fn union_primitive_from_ext_fn() {
+        let ast = ast! {
+            ext func magic() -> int;
+
+            type Nothing;
+            type NullableInt = int | Nothing;
+
+            func g(n: NullableInt) {}
+
+            g(magic());
+        };
+
+        let fir = fir!(ast).type_check();
+
+        assert!(fir.is_ok());
+    }
 }