From 040aa2d49b11182d3fa3fdd5e008c0bb29de5969 Mon Sep 17 00:00:00 2001
From: David Tolnay <dtolnay@gmail.com>
Date: Sun, 3 Apr 2022 23:57:26 -0700
Subject: [PATCH] Compile tests simultaneously using --keep-going

---
 src/cargo.rs | 24 +++++++++++++++++++++
 src/run.rs   | 59 ++++++++++++++++++++++++++++++++++++++++++++++------
 2 files changed, 77 insertions(+), 6 deletions(-)

diff --git a/src/cargo.rs b/src/cargo.rs
index 5cee8f7..ea94e78 100644
--- a/src/cargo.rs
+++ b/src/cargo.rs
@@ -100,6 +100,30 @@ pub fn build_test(project: &Project, name: &Name) -> Result<Output> {
         .map_err(Error::Cargo)
 }
 
+pub fn build_all_tests(project: &Project) -> Result<Output> {
+    let _ = cargo(project)
+        .arg("clean")
+        .arg("--package")
+        .arg(&project.name)
+        .arg("--color=never")
+        .stdout(Stdio::null())
+        .stderr(Stdio::null())
+        .status();
+
+    cargo(project)
+        .arg(if project.has_pass { "build" } else { "check" })
+        .args(target())
+        .arg("--bins")
+        .args(features(project))
+        .arg("--quiet")
+        .arg("--color=never")
+        .arg("--message-format=json")
+        .arg("-Zunstable-options")
+        .arg("--keep-going")
+        .output()
+        .map_err(Error::Cargo)
+}
+
 pub fn run_test(project: &Project, name: &Name) -> Result<Output> {
     cargo(project)
         .arg("run")
diff --git a/src/run.rs b/src/run.rs
index 4a2ab05..6232127 100644
--- a/src/run.rs
+++ b/src/run.rs
@@ -62,6 +62,14 @@ impl Runner {
 
         if tests.is_empty() {
             message::no_tests_enabled();
+        } else if project.keep_going && !project.has_pass {
+            failures = match self.run_all(&project, tests) {
+                Ok(failures) => failures,
+                Err(err) => {
+                    message::test_fail(err);
+                    len
+                }
+            }
         } else {
             for test in tests {
                 if let Err(err) = test.run(&project) {
@@ -260,6 +268,35 @@ impl Runner {
             },
         }
     }
+
+    fn run_all(&self, project: &Project, tests: Vec<ExpandedTest>) -> Result<usize> {
+        let mut failures = 0;
+        let output = cargo::build_all_tests(project)?;
+        let parsed = parse_cargo_json(&output.stdout);
+        let fallback = Stderr::default();
+
+        for mut t in tests {
+            let show_expected = false;
+            message::begin_test(&t.test, show_expected);
+
+            if t.error.is_none() {
+                t.error = check_exists(&t.test.path).err();
+            }
+
+            if t.error.is_none() {
+                let src_path = project.source_dir.join(&t.test.path);
+                let this_test = parsed.stderrs.get(&src_path).unwrap_or(&fallback);
+                t.error = t.test.check(project, &t.name, this_test, "").err();
+            }
+
+            if let Some(err) = t.error {
+                failures += 1;
+                message::test_fail(err);
+            }
+        }
+
+        Ok(failures)
+    }
 }
 
 impl Test {
@@ -269,13 +306,23 @@ impl Test {
         check_exists(&self.path)?;
 
         let output = cargo::build_test(project, name)?;
+        let parsed = parse_cargo_json(&output.stdout);
         let src_path = project.source_dir.join(&self.path);
-        let mut outputs = parse_cargo_json(&output.stdout);
-        let this_test = outputs.stderrs.remove(&src_path).unwrap_or_default();
-        let success = this_test.success;
-        let stdout = &outputs.stdout;
+        let fallback = Stderr::default();
+        let this_test = parsed.stderrs.get(&src_path).unwrap_or(&fallback);
+        self.check(project, name, this_test, &parsed.stdout)
+    }
+
+    fn check(
+        &self,
+        project: &Project,
+        name: &Name,
+        result: &Stderr,
+        build_stdout: &str,
+    ) -> Result<()> {
+        let success = result.success;
         let stderr = normalize::diagnostics(
-            &this_test.stderr,
+            &result.stderr,
             Context {
                 krate: &name.0,
                 source_dir: &project.source_dir,
@@ -291,7 +338,7 @@ impl Test {
             Expected::CompileFail => Test::check_compile_fail,
         };
 
-        check(self, project, name, success, stdout, stderr)
+        check(self, project, name, success, build_stdout, stderr)
     }
 
     fn check_pass(