Add 1-argument ensure!($expr)

eyre-rs · Nov 21, 2023 · ec98ce3 · ec98ce3
1 parent 949e845
commit ec98ce3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/eyre/src/macros.rs b/eyre/src/macros.rs
@@ -107,6 +107,11 @@ macro_rules! bail {
 /// ```
 #[macro_export]
 macro_rules! ensure {
+    ($cond:expr $(,)?) => {
+        if !$cond {
+            $crate::ensure!($cond, concat!("Condition failed: `", stringify!($cond), "`"))
+        }
+    };
     ($cond:expr, $msg:literal $(,)?) => {
         if !$cond {
             return $crate::private::Err($crate::eyre!($msg));

diff --git a/eyre/tests/test_macros.rs b/eyre/tests/test_macros.rs
@@ -39,6 +39,15 @@ fn test_ensure() {
         Ok(())
     };
     assert!(f().is_err());
+
+    let f = || {
+        ensure!(v + v == 1);
+        Ok(())
+    };
+    assert_eq!(
+        f().unwrap_err().to_string(),
+        "Condition failed: `v + v == 1`",
+    );
 }
 
 #[test]