Remove assert_cmd and predicates

1 files changed, 131 insertions, 76 deletions

diff --git a/tests/integration_tests.rs b/tests/integration_tests.rs
index 7d30467..d5afd2c 100644
--- a/tests/integration_tests.rs
+++ b/tests/integration_tests.rs
@@ -1,134 +1,189 @@
-use assert_cmd::prelude::*;
-use std::process::Command;
+use std::{
+    env::{self, consts::EXE_SUFFIX},
+    process::{Command, Stdio},
+    str::from_utf8,
+};
+
+#[derive(Default)]
+struct Cmd<'a> {
+    current_dir: Option<&'a str>,
+    args: &'a [&'a str],
+    stdout: Option<&'a str>,
+    full_stdout: bool,
+}
+
+impl<'a> Cmd<'a> {
+    #[inline]
+    fn current_dir(&mut self, current_dir: &'a str) -> &mut Self {
+        self.current_dir = Some(current_dir);
+        self
+    }
+
+    #[inline]
+    fn args(&mut self, args: &'a [&'a str]) -> &mut Self {
+        self.args = args;
+        self
+    }
+
+    #[inline]
+    fn stdout(&mut self, stdout: &'a str) -> &mut Self {
+        self.stdout = Some(stdout);
+        self
+    }
+
+    #[inline]
+    fn full_stdout(&mut self) -> &mut Self {
+        self.full_stdout = true;
+        self
+    }
+
+    fn assert(&self, success: bool) {
+        let rustlings_bin = {
+            let mut path = env::current_exe().unwrap();
+            // Pop test binary name
+            path.pop();
+            // Pop `/deps`
+            path.pop();
+
+            path.push("rustlings");
+            let mut path = path.into_os_string();
+            path.push(EXE_SUFFIX);
+            path
+        };
+
+        let mut cmd = Command::new(rustlings_bin);
+
+        if let Some(current_dir) = self.current_dir {
+            cmd.current_dir(current_dir);
+        }
+
+        cmd.args(self.args)
+            .stdin(Stdio::null())
+            .stderr(Stdio::null());
+
+        let status = if let Some(expected_stdout) = self.stdout {
+            let output = cmd.output().unwrap();
+            let stdout = from_utf8(&output.stdout).unwrap();
+
+            if self.full_stdout {
+                assert_eq!(stdout, expected_stdout);
+            } else {
+                assert!(stdout.contains(expected_stdout));
+            }
+
+            output.status
+        } else {
+            cmd.stdout(Stdio::null()).status().unwrap()
+        };
+
+        assert_eq!(status.success(), success);
+    }
+
+    #[inline]
+    fn success(&self) {
+        self.assert(true);
+    }
+
+    #[inline]
+    fn fail(&self) {
+        self.assert(false);
+    }
+}
 
 #[test]
 fn fails_when_in_wrong_dir() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .current_dir("tests/")
-        .assert()
-        .code(1);
+    Cmd::default().current_dir("tests").fail();
 }
 
 #[test]
 fn run_single_compile_success() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "compSuccess"])
-        .current_dir("tests/fixture/success/")
-        .assert()
+    Cmd::default()
+        .current_dir("tests/fixture/success")
+        .args(&["run", "compSuccess"])
         .success();
 }
 
 #[test]
 fn run_single_compile_failure() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "compFailure"])
-        .current_dir("tests/fixture/failure/")
-        .assert()
-        .code(1);
+    Cmd::default()
+        .current_dir("tests/fixture/failure")
+        .args(&["run", "compFailure"])
+        .fail();
 }
 
 #[test]
 fn run_single_test_success() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "testSuccess"])
-        .current_dir("tests/fixture/success/")
-        .assert()
+    Cmd::default()
+        .current_dir("tests/fixture/success")
+        .args(&["run", "testSuccess"])
         .success();
 }
 
 #[test]
 fn run_single_test_failure() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "testFailure"])
-        .current_dir("tests/fixture/failure/")
-        .assert()
-        .code(1);
+    Cmd::default()
+        .current_dir("tests/fixture/failure")
+        .args(&["run", "testFailure"])
+        .fail();
 }
 
 #[test]
 fn run_single_test_not_passed() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "testNotPassed.rs"])
-        .current_dir("tests/fixture/failure/")
-        .assert()
-        .code(1);
+    Cmd::default()
+        .current_dir("tests/fixture/failure")
+        .args(&["run", "testNotPassed.rs"])
+        .fail();
 }
 
 #[test]
 fn run_single_test_no_exercise() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "compNoExercise.rs"])
+    Cmd::default()
         .current_dir("tests/fixture/failure")
-        .assert()
-        .code(1);
+        .args(&["run", "compNoExercise.rs"])
+        .fail();
 }
 
 #[test]
 fn reset_single_exercise() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["reset", "intro1"])
-        .assert()
-        .code(0);
+    Cmd::default().args(&["reset", "intro1"]).success();
 }
 
 #[test]
 fn reset_no_exercise() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .arg("reset")
-        .assert()
-        .code(2)
-        .stderr(predicates::str::contains(
-            "required arguments were not provided",
-        ));
+    Cmd::default().args(&["reset"]).fail();
 }
 
 #[test]
 fn get_hint_for_single_test() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["hint", "testFailure"])
+    Cmd::default()
         .current_dir("tests/fixture/failure")
-        .assert()
-        .code(0)
-        .stdout("Hello!\n");
+        .args(&["hint", "testFailure"])
+        .stdout("Hello!\n")
+        .full_stdout()
+        .success();
 }
 
 #[test]
 fn run_compile_exercise_does_not_prompt() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "pending_exercise"])
+    Cmd::default()
         .current_dir("tests/fixture/state")
-        .assert()
-        .code(0);
+        .args(&["run", "pending_exercise"])
+        .success();
 }
 
 #[test]
 fn run_test_exercise_does_not_prompt() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "pending_test_exercise"])
+    Cmd::default()
         .current_dir("tests/fixture/state")
-        .assert()
-        .code(0);
+        .args(&["run", "pending_test_exercise"])
+        .success();
 }
 
 #[test]
 fn run_single_test_success_with_output() {
-    Command::cargo_bin("rustlings")
-        .unwrap()
-        .args(["run", "testSuccess"])
-        .current_dir("tests/fixture/success/")
-        .assert()
-        .code(0)
-        .stdout(predicates::str::contains("THIS TEST TOO SHALL PASS"));
+    Cmd::default()
+        .current_dir("tests/fixture/success")
+        .args(&["run", "testSuccess"])
+        .stdout("\nTHIS TEST TOO SHALL PASS\n")
+        .success();
 }