From ef615feac5e7fc178d3a6888d4e817a1891a62f3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sergio=20Mart=C3=ADnez=20Portela?=
 <sergio@codigoparallevar.com>
Date: Tue, 30 Jul 2024 10:52:37 +0200
Subject: [PATCH] Allow trailing whitespace changes on raw content update.

---
 tests/test_org.py         | 2 +-
 tests/utils/assertions.py | 7 +++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/test_org.py b/tests/test_org.py
index c7cfd6e..e66e338 100644
--- a/tests/test_org.py
+++ b/tests/test_org.py
@@ -903,7 +903,7 @@ class TestSerde(unittest.TestCase):
         hl.update_raw_contents(content)
 
         # Check after update
-        ex.assert_matches(self, hl)
+        ex.assert_matches(self, hl, accept_trailing_whitespace_changes=True)
 
 
 def print_tree(tree, indentation=0, headline=None):
diff --git a/tests/utils/assertions.py b/tests/utils/assertions.py
index 59dc658..732bc37 100644
--- a/tests/utils/assertions.py
+++ b/tests/utils/assertions.py
@@ -58,7 +58,7 @@ class HL:
         self.content = content
         self.children = children
 
-    def assert_matches(self, test_case: unittest.TestCase, doc):
+    def assert_matches(self, test_case: unittest.TestCase, doc, accept_trailing_whitespace_changes=False):
         test_case.assertEqual(self.title, get_raw(doc.title))
 
         # Check properties
@@ -75,7 +75,10 @@ class HL:
                         timestamp_to_datetime(doc_props[i].value), prop[1]
                     )
 
-        test_case.assertEqual(get_raw_contents(doc), self.get_raw())
+        if accept_trailing_whitespace_changes:
+            test_case.assertEqual(get_raw_contents(doc).rstrip(), self.get_raw().rstrip())
+        else:
+            test_case.assertEqual(get_raw_contents(doc), self.get_raw())
 
         # Check children
         if self.children is None: