Fix creusot argument command

creusot-rs · Aug 3, 2024 · 12ca32c · 12ca32c
1 parent e13ca38
commit 12ca32c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/creusot/tests/ui.rs b/creusot/tests/ui.rs
@@ -91,8 +91,14 @@ fn run_creusot(
     // Find comment chunks of the form CREUSOT_ARG=ARGUMENT. Does not support spaces in arguments currently (would require real parser)
     let args: Vec<_> = header_line
         .split(" ")
-        .filter(|chunk| chunk.contains("CREUSOT_ARG"))
-        .flat_map(|chunk| chunk.split("=").nth(1))
+        .filter_map(|chunk| {
+            let (first, rest) = chunk.split_once("=")?;
+            if first != "CREUSOT_ARG" {
+                None
+            } else {
+                Some(rest)
+            }
+        })
         .collect();
 
     cmd.args(&[