From d2def28751b9dbdf3cc47b00850c3dc0e728e361 Mon Sep 17 00:00:00 2001
From: Jeff <dev@jeffa.io>
Date: Sun, 31 Dec 2023 08:36:39 -0500
Subject: [PATCH] Add tests covering for loops

---
 tests/interpret.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 46 insertions(+), 2 deletions(-)

diff --git a/tests/interpret.rs b/tests/interpret.rs
index 9c9bfc0..ae27ea0 100644
--- a/tests/interpret.rs
+++ b/tests/interpret.rs
@@ -47,6 +47,50 @@ mod assignment {
     }
 }
 
+mod for_loop {
+    use dust_lang::*;
+
+    #[test]
+    fn simple_for_loop() {
+        let result = interpret("for i in [1 2 3] { output(i) }");
+
+        assert_eq!(Ok(Value::none()), result);
+    }
+
+    #[test]
+    fn modify_value() {
+        let result = interpret(
+            "
+            list = []
+            for i in [1 2 3] { list += i }
+            list
+            ",
+        );
+
+        assert_eq!(
+            Ok(Value::List(List::with_items(vec![
+                Value::Integer(1),
+                Value::Integer(2),
+                Value::Integer(3),
+            ]))),
+            result
+        );
+    }
+
+    #[test]
+    fn modify_value_async() {
+        let result = interpret(
+            "
+            list = []
+            async for i in [1 2 3] { list += i }
+            length(list)
+            ",
+        );
+
+        assert_eq!(Ok(Value::Integer(3)), result);
+    }
+}
+
 mod logic {
     use dust_lang::*;
 
@@ -154,7 +198,7 @@ mod value {
         let function = value.as_function().unwrap();
 
         assert_eq!(&Vec::<Identifier>::with_capacity(0), function.parameters());
-        assert_eq!(Ok(&Type::Integer), function.return_type());
+        assert_eq!(&Type::Integer, function.return_type());
 
         let result = interpret("(x <bool>) -> <bool> {true}");
         let value = result.unwrap();
@@ -164,7 +208,7 @@ mod value {
             &vec![Identifier::new("x".to_string())],
             function.parameters()
         );
-        assert_eq!(Ok(&Type::Boolean), function.return_type());
+        assert_eq!(&Type::Boolean, function.return_type());
     }
 
     #[test]