Allow blessing test run for collapse too

jonhoo · Jul 14, 2024 · edaba3b · edaba3b
1 parent bd14b1f
commit edaba3b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/common/collapse.rs b/tests/common/collapse.rs
@@ -89,7 +89,15 @@ where
         eprintln!("test output in {}", tm.display());
     }
     // and then compare
-    compare_results(result, expected, expected_filename, strip_quotes);
+    if let Err(e) = std::panic::catch_unwind(|| {
+        compare_results(result.clone(), expected, expected_filename, strip_quotes)
+    }) {
+        if std::env::var("INFERNO_BLESS_TESTS").is_ok() {
+            fs::write(expected_filename, result.get_ref()).unwrap();
+        } else {
+            std::panic::resume_unwind(e);
+        }
+    }
     Ok(())
 }