add tests for built-in function str

binoyjayan · Sep 17, 2023 · 9a1fe18 · 9a1fe18
1 parent bb9d4a0
commit 9a1fe18
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git a/src/common/object.rs b/src/common/object.rs
@@ -265,10 +265,10 @@ impl fmt::Display for HMap {
         let pairs_str = self
             .pairs
             .iter()
-            .map(|(k, v)| format!("{}: {}, ", k, v))
+            .map(|(k, v)| format!(r#""{}": {}, "#, k, v))
             .collect::<String>();
         let pairs_str = pairs_str.trim_end_matches(|c| c == ' ' || c == ',');
-        write!(f, "{{ {} }}", pairs_str)
+        write!(f, "{{{}}}", pairs_str)
     }
 }
 

diff --git a/src/vm/tests.rs b/src/vm/tests.rs
@@ -828,6 +828,39 @@ fn test_builtin_functions() {
             "#,
             expected: Object::Number(2.0),
         },
+        VmTestCase {
+            // rest([]) is Nil
+            input: "str(rest([]))",
+            expected: Object::Str(String::from("nil")),
+        },
+        VmTestCase {
+            input: r#"str("hello")"#,
+            expected: Object::Str(String::from("hello")),
+        },
+        VmTestCase {
+            input: "str(999)",
+            expected: Object::Str(String::from("999")),
+        },
+        VmTestCase {
+            input: "str(str(999))",
+            expected: Object::Str(String::from("999")),
+        },
+        VmTestCase {
+            input: "str(true)",
+            expected: Object::Str(String::from("true")),
+        },
+        VmTestCase {
+            input: "str([1, 2, 3333, 4, 5])",
+            expected: Object::Str(String::from("[1, 2, 3333, 4, 5]")),
+        },
+        VmTestCase {
+            input: r#"str({"a": 1})"#,
+            expected: Object::Str(String::from(r#"{"a": 1}"#)),
+        },
+        VmTestCase {
+            input: "str({})",
+            expected: Object::Str(String::from("{}")),
+        },
     ];
     run_vm_tests(&tests);
 }
@@ -855,6 +888,10 @@ fn test_builtin_function_failures() {
             input: r#"push(1, 1)"#,
             expected: "argument to 'push' not supported",
         },
+        VmTestCaseErr {
+            input: "str(fn() {})",
+            expected: "argument to 'str' not supported",
+        },
     ];
 
     run_vm_negative_tests(&tests);