Move to ui_test which supports our testing scheme out of the box

Author: oli-obk

.gitignore

@@ -1,2 +1,2 @@
-/target
+target
 Cargo.lock

Cargo.toml

@@ -22,7 +22,7 @@ default = ["tracing-log"]
 [dev-dependencies]
 tracing = "0.1"
 glob = "0.3"
-assert_cmd = "1"
+ui_test = "0.7"
 log = "0.4"
 
 [[test]]

test_dependencies/Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "test_dependencies"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+tracing = "0.1"
+glob = "0.3"
+log = "0.4"
+tracing-subscriber = { version = "0.3", default-features = false, features = ["registry", "fmt", "std"] }
+tracing-tree = { path = ".." }
+tracing-log = "0.1"

test_dependencies/src/lib.rs

@@ -0,0 +1 @@
+

tests/ui.rs

@@ -1,79 +1,19 @@
-use assert_cmd::prelude::*;
+use ui_test::{color_eyre::Result, run_tests, Config, Mode, OutputConflictHandling};
 
-use std::process::Command;
-
-// Timings are flaky, so tests would spuriously fail.
-// Thus we replace all `/([0-9]+)ms/` with underscores
-fn replace_ms(data: &[u8]) -> Vec<u8> {
-    let mut skip = false;
-    let mut seen_s = false;
-    let mut v: Vec<u8> = data
-        .iter()
-        .rev()
-        .filter_map(|&b| match (b, skip, seen_s) {
-            (b'0'..=b'9', true, _) => None,
-            (_, true, _) => {
-                skip = false;
-                Some(b)
-            }
-            (b's', _, _) => {
-                seen_s = true;
-                Some(b)
-            }
-            (b'm', _, true) => {
-                seen_s = false;
-                skip = true;
-                Some(b)
-            }
-            _ => Some(b),
-        })
-        .collect();
-    v.reverse();
-    v
-}
-
-fn main() {
-    for entry in glob::glob("examples/*.rs").expect("Failed to read glob pattern") {
-        let entry = entry.unwrap();
-        let mut cmd = Command::cargo_bin(entry.with_extension("").to_str().unwrap()).unwrap();
-        let output = cmd.unwrap();
-        let stderr = entry.with_extension("stderr");
-        let stdout = entry.with_extension("stdout");
-
-        if std::env::args().any(|arg| arg == "--bless") {
-            if output.stderr.is_empty() {
-                let _ = std::fs::remove_file(stderr);
-            } else {
-                std::fs::write(stderr, replace_ms(&output.stderr)).unwrap();
-            }
-            if output.stdout.is_empty() {
-                let _ = std::fs::remove_file(stdout);
-            } else {
-                std::fs::write(stdout, replace_ms(&output.stdout)).unwrap();
-            }
-        } else {
-            if output.stderr.is_empty() {
-                assert!(
-                    !stderr.exists(),
-                    "{} exists but there was no stderr output",
-                    stderr.display()
-                );
-            } else {
-                assert!(
-                    std::fs::read(&stderr).unwrap() == replace_ms(&output.stderr),
-                    "{} is not the expected output, rerun the test with `cargo test -- -- --bless`",
-                    stderr.display()
-                );
-            }
-            if output.stdout.is_empty() {
-                assert!(!stdout.exists());
-            } else {
-                assert!(
-                    std::fs::read(&stdout).unwrap() == replace_ms(&output.stdout),
-                    "{} is not the expected output, rerun the test with `cargo test -- -- --bless`",
-                    stdout.display()
-                );
-            }
-        }
-    }
+fn main() -> Result<()> {
+    let mut config = Config::default();
+    config.root_dir = "examples".into();
+    config.dependencies_crate_manifest_path = Some("test_dependencies/Cargo.toml".into());
+    config.args.push("--cfg".into());
+    config.args.push("feature=\"tracing-log\"".into());
+    config.out_dir = Some("target/ui_test".into());
+    config.mode = Mode::Run { exit_code: 0 };
+    config.stdout_filter("[0-9]+(ms|s|m)", "$1");
+    config.stderr_filter("[0-9]+(ms|s|m)", "$1");
+    config.output_conflict_handling = if std::env::args().any(|arg| arg == "--bless") {
+        OutputConflictHandling::Bless
+    } else {
+        OutputConflictHandling::Error
+    };
+    run_tests(config)
 }