[T-8592][FIX] partner_address_hide_pdf_fields: fix test

Author: Anxo82

partner_address_hide_pdf_fields/tests/fake_models.py

@@ -0,0 +1,20 @@
+from odoo import models
+
+
+class ResPartner(models.Model):
+    _name = "res.partner"
+    _inherit = "res.partner"
+    _description = "Fake ResPartner for Testing"
+
+    def _hided_in_pdf_address_fields(self):
+        return super()._hided_in_pdf_address_fields() + ["field_id", "field_name"]
+
+    # This FakeModel overrides `_prepare_display_address` to simulate
+    # additional address fields (`field_id`, `field_name`) being included
+    # and hidden depending on the `inside_pdf` context.
+    def _prepare_display_address(self, without_company=False):
+        address_format, args = super()._prepare_display_address(without_company)
+        args.update({"field_id": "123", "field_name": "Field Name"})
+        if self._display_address_inside_pdf():
+            args.update({key: "" for key in self._hided_in_pdf_address_fields()})
+        return address_format, args

partner_address_hide_pdf_fields/tests/test_partner_address_hide_pdf_fields.py

@@ -1,63 +1,39 @@
+from odoo_test_helper import FakeModelLoader
+
 from odoo.tests.common import TransactionCase
 
 
 class TestPartnerAddressHidePdfFields(TransactionCase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
+        cls.loader = FakeModelLoader(cls.env, cls.__module__)
+        cls.loader.backup_registry()
+        from .fake_models import ResPartner
+
+        cls.loader.update_registry((ResPartner,))
+
         cls.partner = cls.env["res.partner"].create(
             {
-                "name": "Test Partner",
-                "country_id": cls.env.ref("base.es").id,
-                "city": "Test city",
+                "name": "Partner Test",
+                "city": "Test City",
                 "zip": "12345",
-                "street": "Test street",
+                "street": "Test Street",
             }
         )
 
-        cls._original_hided = cls.env[
-            "res.partner"
-        ].__class__._hided_in_pdf_address_fields
-
-        def fake_hided_in_pdf_address_fields(cls):
-            return ["field_id", "field_name"]
-
-        cls.env["res.partner"].__class__._hided_in_pdf_address_fields = classmethod(
-            fake_hided_in_pdf_address_fields
-        )
-
-        cls._original_prepare = cls.env[
-            "res.partner"
-        ].__class__._prepare_display_address
-
-        def test_prepare_display_address(self, without_company=False):
-            address_format, args = cls._original_prepare(self, without_company)
-            args.update({"field_id": "123", "field_name": "Name"})
-            if self._display_address_inside_pdf():
-                args.update({key: "" for key in self._hided_in_pdf_address_fields()})
-            return address_format, args
-
-        cls.env[
-            "res.partner"
-        ].__class__._prepare_display_address = test_prepare_display_address
-
     @classmethod
     def tearDownClass(cls):
-        cls.env[
-            "res.partner"
-        ].__class__._hided_in_pdf_address_fields = cls._original_hided
-        cls.env[
-            "res.partner"
-        ].__class__._prepare_display_address = cls._original_prepare
+        cls.loader.restore_registry()
         super().tearDownClass()
 
     def test_fields_hidden_inside_pdf(self):
-        partner_with_ctx = self.partner.with_context(inside_pdf=True)
-        _, args = partner_with_ctx._prepare_display_address()
-        self.assertEqual(args.get("field_id"), "")
-        self.assertEqual(args.get("field_name"), "")
+        partner_ctx = self.partner.with_context(inside_pdf=True)
+        _, args = partner_ctx._prepare_display_address()
+        self.assertEqual(args["field_id"], "")
+        self.assertEqual(args["field_name"], "")
 
     def test_fields_visible_outside_pdf(self):
         _, args = self.partner._prepare_display_address()
-        self.assertEqual(args.get("field_id"), "123")
-        self.assertEqual(args.get("field_name"), "Name")
+        self.assertEqual(args["field_id"], "123")
+        self.assertEqual(args["field_name"], "Field Name")

test-requirements.txt

@@ -0,0 +1 @@
+odoo_test_helper