assert_cli! Macro

Author: killercup

src/lib.rs

@@ -8,14 +8,14 @@
 //! Here's a trivial example:
 //!
 //! ```rust
-//! extern crate assert_cli;
+//! # extern crate assert_cli;
+//!
 //! assert_cli::assert_cli_output("echo", &["42"], "42").unwrap();
 //! ```
 //!
 //! And here is one that will fail:
 //!
 //! ```rust,should_panic
-//! extern crate assert_cli;
 //! assert_cli::assert_cli_output("echo", &["42"], "1337").unwrap();
 //! ```
 //!
@@ -25,6 +25,15 @@
 //! -1337
 //! +42
 //! ```
+//!
+//! Alternatively, you can use the `assert_cli!` macro:
+//!
+//! ```rust,ignore
+//! assert_cli!("echo 42" => Success, "42").unwrap();
+//! ```
+//!
+//! Make sure to include the crate as `#[macro_use] extern crate assert_cli;`.
+
 
 #![cfg_attr(feature = "dev", feature(plugin))]
 #![cfg_attr(feature = "dev", plugin(clippy))]
@@ -127,3 +136,25 @@ pub fn assert_cli_output_error<S>(cmd: &str,
         })
         .map_err(From::from)
 }
+
+/// The `assert_cli!` macro combines the functionality of the other functions in this crate in one
+/// short macro.
+///
+/// ```rust,ignore
+/// assert_cli!("echo 42" => Success, "42").unwrap();
+/// assert_cli!("exit 11" => Error 11, "").unwrap();
+/// ```
+///
+/// Make sure to include the crate as `#[macro_use] extern crate assert_cli;`.
+#[macro_export]
+macro_rules! assert_cli {
+    ($cmd:expr, $args:expr => Success, $output:expr) => {{
+        $crate::assert_cli_output($cmd, $args, $output)
+    }};
+    ($cmd:expr, $args:expr => Error, $output:expr) => {{
+        $crate::assert_cli_output_error($cmd, $args, None, $output)
+    }};
+    ($cmd:expr, $args:expr => Error $err:expr, $output:expr) => {{
+        $crate::assert_cli_output_error($cmd, $args, Some($err), $output)
+    }};
+}

tests/macro.rs

@@ -0,0 +1,24 @@
+#[macro_use]
+extern crate assert_cli;
+
+fn test_helper(exit_code: i32, output: &str) -> Vec<String> {
+    vec!["-c".into(),
+         format!(r#"function test_helper() {{ >&2 echo $1; return {}; }}; test_helper "{}""#,
+                 exit_code,
+                 output)]
+}
+
+#[test]
+fn assert_success() {
+    assert_cli!("echo", &["42"] => Success, "42").unwrap();
+    assert!(assert_cli!("echo", &["1"] => Success, "42").is_err());
+}
+
+#[test]
+fn assert_failure() {
+    assert_cli!("bash", &test_helper(66, "sorry, my bad") => Error, "sorry, my bad\n").unwrap();
+    assert_cli!("bash", &test_helper(42, "error no 42") => Error 42, "error no 42\n").unwrap();
+
+    assert!(assert_cli!("echo", &["good"] => Error, "").is_err());
+    assert!(assert_cli!("echo", &["good"] => Error 11, "").is_err());
+}