tests fixes OCA#2

Author: KDRdev

document_page_reference/models/document_page.py

@@ -50,22 +50,26 @@ def _get_document(self, code):
 
     def get_content(self):
         self.ensure_one()
-        raw = self.content or ""
+        raw = str(self.content or "")
         content_parsed = Markup(raw)
         for text in re.findall(r"\{\{.*?\}\}", raw):
             reference = re.sub(r"<[^>]*>", "", text).replace("{{", "").replace("}}", "")
             content_parsed = content_parsed.replace(
                 text, self._resolve_reference(reference)
             )
+        # Find all oe_direct_line links and re-resolve them
+        # We look for a <a> tag with class='oe_direct_line' and its 'name' attribute
+        # name attribute contains the reference code
         link_regex = (
             r"<a[^>]*class=['\"][^'\"]*oe_direct_line[^'\"]*['\"]"
             r"[^>]*name=['\"]([^'\"]*)['\"][^>]*>.*?</a>"
         )
         for match in re.finditer(link_regex, raw):
             full_link = match.group(0)
             reference = match.group(1)
+            # Ensure we replace exactly what was found in raw
             content_parsed = content_parsed.replace(
-                full_link, self._resolve_reference(reference)
+                Markup(full_link), self._resolve_reference(reference)
             )
         return content_parsed
 
@@ -91,7 +95,7 @@ def _resolve_reference(self, code):
         )
 
     def get_raw_content(self):
-        return Markup(self.with_context(raw_reference=True).get_content())
+        return str(self.with_context(raw_reference=True).get_content())
 
     @api.model_create_multi
     def create(self, vals_list):

document_page_reference/tests/test_document_reference.py

@@ -37,7 +37,9 @@ def test_no_constrains(self):
         self.assertFalse(self.page2.reference)
 
     def test_check_raw(self):
-        self.assertEqual(self.page2.display_name, self.page1.get_raw_content())
+        self.assertEqual(
+            str(self.page2.display_name), str(self.page1.get_raw_content())
+        )
 
     def test_auto_reference(self):
         """Test if reference is proposed when saving a page without one."""