Ignore leading newlines on snapshot comparison

mitsuhiko · Feb 19, 2024 · f356b52 · f356b52
1 parent 181dfaf
commit f356b52
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/snapshot.rs b/src/snapshot.rs
@@ -576,7 +576,13 @@ impl From<SnapshotContents> for String {
 
 impl PartialEq for SnapshotContents {
     fn eq(&self, other: &Self) -> bool {
-        self.0.trim_end() == other.0.trim_end()
+        self.0
+            .trim_start_matches(|x| x == '\r' || x == '\n')
+            .trim_end()
+            == other
+                .0
+                .trim_start_matches(|x| x == '\r' || x == '\n')
+                .trim_end()
     }
 }
 

diff --git a/tests/test_inline.rs b/tests/test_inline.rs
@@ -286,3 +286,15 @@ fn test_inline_test_in_loop() {
         assert_snapshot!(i.to_string(), @"0");
     }
 }
+
+#[test]
+fn test_inline_snapshot_whitespace() {
+    assert_snapshot!("\n\nfoo\n\n    bar\n\n", @r###"
+
+
+    foo
+
+        bar
+
+    "###);
+}