fire: Add more complex return tests

jinko-core · Oct 19, 2023 · c82473f · c82473f
1 parent bbeb5c0
commit c82473f
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/fire/src/lib.rs b/fire/src/lib.rs
@@ -530,4 +530,40 @@ mod tests {
         let result = fir!(ast).interpret();
         assert_eq!(result, Some(Instance::from("boo")))
     }
+
+    #[test]
+    fn nested_return() {
+        let ast = ast! {
+            func halloween() -> string {
+                { { { {
+                    return "boo";
+                } } } }
+            }
+
+            halloween()
+        };
+
+        let result = fir!(ast).interpret();
+        assert_eq!(result, Some(Instance::from("boo")))
+    }
+
+    #[test]
+    fn nested_return_inner_fn() {
+        let ast = ast! {
+            func halloween() -> string {
+                func inner() -> string {
+                    { { { {
+                        return "boo";
+                    } } } }
+                };
+
+                "different string"
+            }
+
+            halloween()
+        };
+
+        let result = fir!(ast).interpret();
+        assert_eq!(result, Some(Instance::from("different string")))
+    }
 }