Create a more compact diff format

Author: oli-obk

Cargo.lock

@@ -12,7 +12,7 @@ rustc_version = "0.4"
 colored = "2"
 # Features chosen to match those required by env_logger, to avoid rebuilds
 regex = { version = "1.5.5", default-features = false, features = ["perf", "std"] }
-pretty_assertions = "1.2.1"
+diff = "0.1.13"
 crossbeam = "0.8.1"
 lazy_static = "1.4.0"
 serde = { version = "1.0", features = ["derive"] }

ui_test/Cargo.lock

@@ -0,0 +1,111 @@
+use colored::*;
+use diff::{chars, lines, Result, Result::*};
+
+#[derive(Default)]
+struct DiffState<'a> {
+    skipped_lines: usize,
+    /// When we see a removed line, we don't print it, we
+    /// keep it around to compare it with the next added line.
+    prev_left: Option<&'a str>,
+}
+
+impl<'a> DiffState<'a> {
+    fn print_skip(&mut self) {
+        if self.skipped_lines != 0 {
+            eprintln!("... {} lines skipped", self.skipped_lines);
+        }
+        self.skipped_lines = 0;
+    }
+
+    fn print_prev(&mut self) {
+        if let Some(l) = self.prev_left.take() {
+            self.print_left(l);
+        }
+    }
+
+    fn print_left(&self, l: &str) {
+        eprintln!("{}{}", "-".red(), l.red());
+    }
+
+    fn print_right(&self, r: &str) {
+        eprintln!("{}{}", "+".green(), r.green());
+    }
+
+    fn row(&mut self, row: Result<&'a str>) {
+        match row {
+            Left(l) => {
+                self.print_skip();
+                self.print_prev();
+                self.prev_left = Some(l);
+            }
+            Both(..) => {
+                self.print_prev();
+                self.skipped_lines += 1
+            }
+            Right(r) => {
+                if let Some(l) = self.prev_left.take() {
+                    // If the lines only add chars or only remove chars, display an inline diff
+                    let diff = chars(l, r);
+                    let mut seen_l = false;
+                    let mut seen_r = false;
+                    for char in &diff {
+                        match char {
+                            Left(l) if !l.is_whitespace() => seen_l = true,
+                            Right(r) if !r.is_whitespace() => seen_r = true,
+                            _ => {}
+                        }
+                    }
+                    if seen_l && seen_r {
+                        // the line both adds and removes chars, print both lines, but highlight their differences
+                        eprint!("{}", "-".red());
+                        for char in &diff {
+                            match char {
+                                Left(l) => eprint!("{}", l.to_string().red()),
+                                Right(_) => {}
+                                Both(l, _) => eprint!("{}", l),
+                            }
+                        }
+                        eprintln!();
+                        eprint!("{}", "+".green());
+                        for char in &diff {
+                            match char {
+                                Left(_) => {}
+                                Right(r) => eprint!("{}", r.to_string().green()),
+                                Both(l, _) => eprint!("{}", l),
+                            }
+                        }
+                        eprintln!();
+                    } else {
+                        eprint!("{}", "~".yellow());
+                        for char in diff {
+                            match char {
+                                Left(l) => eprint!("{}", l.to_string().red()),
+                                Both(l, _) => eprint!("{}", l),
+                                Right(r) => eprint!("{}", r.to_string().green()),
+                            }
+                        }
+                        eprintln!();
+                    }
+                } else {
+                    self.print_skip();
+                    self.print_right(r);
+                }
+            }
+        }
+    }
+
+    fn finish(self) {
+        if self.skipped_lines != 0 {
+            eprintln!("... {} lines skipped ...", self.skipped_lines);
+        }
+        eprintln!()
+    }
+}
+
+pub fn print_diff(expected: &str, actual: &str) {
+    let mut state = DiffState::default();
+    for row in lines(expected, actual) {
+        state.row(row);
+    }
+    state.finish();
+}

ui_test/Cargo.toml

@@ -16,6 +16,7 @@ use rustc_stderr::{Level, Message};
 
 use crate::parser::{Comments, Condition};
 
+mod diff;
 mod parser;
 mod rustc_stderr;
 #[cfg(test)]
@@ -198,8 +199,7 @@ pub fn run_tests(config: Config) -> Result<()> {
                             dump_stderr = false;
                         }
                         eprintln!("actual output differed from expected {}", path.display());
-                        eprintln!("{}", pretty_assertions::StrComparison::new(expected, actual));
-                        eprintln!()
+                        diff::print_diff(expected, actual);
                     }
                     Error::ErrorsWithoutPattern { path: None, msgs } => {
                         eprintln!(