diff options
author | Dan Davison <dandavison7@gmail.com> | 2020-06-24 10:46:36 -0400 |
---|---|---|
committer | Dan Davison <dandavison7@gmail.com> | 2020-06-25 12:35:07 -0400 |
commit | 88454b0f35831f068447fb391c02d143292658f8 (patch) | |
tree | 647df6a275aafa62a187e5c70001f08e0230f418 | |
parent | 96bacfbf0a5212149d8053653c049ffdb14857c5 (diff) |
Add test of number coloring
-rw-r--r-- | src/features/numbers.rs | 29 |
1 files changed, 26 insertions, 3 deletions
diff --git a/src/features/numbers.rs b/src/features/numbers.rs index da2265f8..fd369634 100644 --- a/src/features/numbers.rs +++ b/src/features/numbers.rs @@ -116,10 +116,33 @@ pub mod tests { "%lp│", ]); let output = run_delta(TWO_MINUS_LINES_DIFF, &config); - let output = strip_ansi_codes(&output); let mut lines = output.lines().skip(4); - assert_eq!(lines.next().unwrap(), " 1 ⋮ │a = 1"); - assert_eq!(lines.next().unwrap(), " 2 ⋮ │b = 2"); + let (line_1, line_2) = (lines.next().unwrap(), lines.next().unwrap()); + assert_eq!(strip_ansi_codes(line_1), " 1 ⋮ │a = 1"); + assert_eq!(strip_ansi_codes(line_2), " 2 ⋮ │b = 2"); + + assert!(line_1.starts_with( + &ansi_term::ANSIStrings(&[ + config.number_left_format_style.paint(" "), + config.number_minus_style.paint("1 "), + config.number_left_format_style.paint("⋮"), + config.number_right_format_style.paint(" "), + config.number_plus_style.paint(" "), + config.number_right_format_style.paint("│"), + ]) + .to_string() + )); + assert!(line_2.starts_with( + &ansi_term::ANSIStrings(&[ + config.number_left_format_style.paint(" "), + config.number_minus_style.paint("2 "), + config.number_left_format_style.paint("⋮"), + config.number_right_format_style.paint(" "), + config.number_plus_style.paint(" "), + config.number_right_format_style.paint("│"), + ]) + .to_string() + )); } #[test] |