Add test for trailing newline in diff.

Credits to @afshinm for most of the code.

diff --git a/src/rustfmt_diff.rs b/src/rustfmt_diff.rs
index daad73dc6ab524613115eb9084b64aee9ab46556..c2dac6c041689d3a6e3c36780c4b8056008a6cab 100644 (file)
--- a/src/rustfmt_diff.rs
+++ b/src/rustfmt_diff.rs
@@ -234,4 +234,20 @@ fn diff_zerocontext() {
             ]
         );
     }
+
+    #[test]
+    fn diff_trailing_newline() {
+        let src = "one\ntwo\nthree\nfour\nfive";
+        let dest = "one\ntwo\nthree\nfour\nfive\n";
+        let diff = make_diff(src, dest, 1);
+        assert_eq!(
+            diff,
+            vec![
+                Mismatch {
+                    line_number: 5,
+                    lines: vec![Context("five".into()), Expected("".into())],
+                },
+            ]
+        );
+    }
 }