test: add unit test for transform_code function

MahadMuhammad · Jul 3, 2024 · 716b6e9 · 716b6e9
1 parent b177ff7
commit 716b6e9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -52,4 +52,13 @@ fn transform_code(code: &String, dg_directive: &str) -> Vec<String> {
     }
     new_code
 }
+
+#[test]
+fn test_transform_code() {
+    let dg_msg = "// { dg-error \"expected one of `:`, `@`, or `|`, found `)`\" }";
+    let rust_msg = "//~^ ERROR expected one of `:`, `@`, or `|`, found `)`";
+    assert_eq!(
+        transform_code(&rust_msg.to_string(), DG_ERROR),
+        vec![dg_msg.to_string()]
+    );
 }