From 85d71a1645010cbc097a6dd982d8a9baaf9146e8 Mon Sep 17 00:00:00 2001
From: GuillemCForgeFlow <guillem.casassas@forgeflow.com>
Date: Tue, 3 Oct 2023 13:44:22 +0200
Subject: [PATCH] [IMP]hs_code_link: add tests to cover new changes

---
 hs_code_link/tests/__init__.py          |  1 +
 hs_code_link/tests/test_hs_code_link.py | 90 +++++++++++++++++++++++++
 2 files changed, 91 insertions(+)
 create mode 100644 hs_code_link/tests/__init__.py
 create mode 100644 hs_code_link/tests/test_hs_code_link.py

diff --git a/hs_code_link/tests/__init__.py b/hs_code_link/tests/__init__.py
new file mode 100644
index 000000000..e28dd936a
--- /dev/null
+++ b/hs_code_link/tests/__init__.py
@@ -0,0 +1 @@
+from . import test_hs_code_link
diff --git a/hs_code_link/tests/test_hs_code_link.py b/hs_code_link/tests/test_hs_code_link.py
new file mode 100644
index 000000000..bc6055afe
--- /dev/null
+++ b/hs_code_link/tests/test_hs_code_link.py
@@ -0,0 +1,90 @@
+# Copyright 2023 ForgeFlow S.L. (https://www.forgeflow.com)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+from odoo.tests.common import SavepointCase
+
+
+class TestHSCodeLink(SavepointCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        # Models
+        cls.company_model = cls.env["res.company"]
+        cls.hs_code_model = cls.env["hs.code"]
+        cls.product_template_model = cls.env["product.template"]
+        cls.property_model = cls.env["ir.property"]
+
+        # Instances
+        cls.company_1 = cls._create_company("Company 1")
+        cls.company_2 = cls._create_company("Company 2")
+
+        cls.hs_code_1 = cls._create_hs_code()
+        cls.hs_code_2 = cls._create_hs_code("12345678910")
+
+        cls.product_template = cls._create_product_template("Product Template 1")
+
+    @classmethod
+    def _create_company(cls, name):
+        return cls.company_model.create({"name": name})
+
+    @classmethod
+    def _create_hs_code(cls, local_code="123456789", description=False):
+        return cls.hs_code_model.with_context(skip_check_hs_code_heading=True).create(
+            {"local_code": local_code, "description": description}
+        )
+
+    @classmethod
+    def _create_product_template(cls, name):
+        return cls.product_template_model.create({"name": name})
+
+    def test_01_check_correctly_sync_values_in_different_companies(self):
+        """
+        Check that the values are correctly synced in different companies.
+        """
+        # First Company
+        template_company_1 = self.product_template.with_context(
+            force_company=self.company_1.id
+        )
+        template_company_1.hs_code_id = self.hs_code_1
+        fields_to_sync = self.hs_code_model._get_fields_to_sync_from_hs_code_id()
+        field_name = "hs_code"
+        self.assertTrue(
+            field_name
+            in [hc_field_name for hc_field_name, _pt_field_name in fields_to_sync]
+        )
+        self.assertEqual(
+            template_company_1.hs_code,
+            self.hs_code_1.hs_code,
+            "The H.S. Codes should be equal.",
+        )
+        # Second Company
+        template_company_2 = self.product_template.with_context(
+            force_company=self.company_2.id
+        )
+        template_company_2.hs_code_id = self.hs_code_2
+        self.assertEqual(
+            template_company_2.hs_code,
+            self.hs_code_2.hs_code,
+            "The H.S. Codes should be equal.",
+        )
+
+    def test_02_delete_property(self):
+        """
+        First create a Company Property and then delete it so check if the related
+        Company Property is also deleted.
+        """
+        template_company_1 = self.product_template.with_context(
+            force_company=self.company_1.id
+        )
+        template_company_1.hs_code_id = self.hs_code_1
+        res_id = "product.template,%s" % template_company_1.id
+        hs_code_id_property = self.property_model.search(
+            [("res_id", "=", res_id), ("name", "=", "hs_code_id")], limit=1
+        )
+        hs_code_property = self.property_model.search(
+            [("res_id", "=", res_id), ("name", "=", "hs_code")], limit=1
+        )
+        self.assertTrue(hs_code_id_property)
+        self.assertTrue(hs_code_property)
+        hs_code_id_property.unlink()
+        self.assertFalse(hs_code_id_property.exists())
+        self.assertFalse(hs_code_property.exists())