From a46f0ca71c914b47624cec224f2950d1b52c7698 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Duy=20=28=C4=90=E1=BB=97=20Anh=29?=
 <doanhduyxavie@gmail.com>
Date: Thu, 10 Oct 2024 19:50:54 +0700
Subject: [PATCH] [ADD] base_import_async: add tests

---
 base_import_async/tests/__init__.py           |  1 +
 .../tests/test_base_import_import.py          | 99 +++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 base_import_async/tests/__init__.py
 create mode 100644 base_import_async/tests/test_base_import_import.py

diff --git a/base_import_async/tests/__init__.py b/base_import_async/tests/__init__.py
new file mode 100644
index 0000000000..629d1e39e1
--- /dev/null
+++ b/base_import_async/tests/__init__.py
@@ -0,0 +1 @@
+from . import test_base_import_import
diff --git a/base_import_async/tests/test_base_import_import.py b/base_import_async/tests/test_base_import_import.py
new file mode 100644
index 0000000000..4a815d6707
--- /dev/null
+++ b/base_import_async/tests/test_base_import_import.py
@@ -0,0 +1,99 @@
+# Copyright 2024 Camptocamp
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+from odoo.tests.common import RecordCapturer, TransactionCase
+
+from ..models.base_import_import import OPT_USE_QUEUE
+
+
+class TestBaseImportImport(TransactionCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.res_partners = cls.env["res.partner"]
+        cls.import_wizard = cls.env["base_import.import"]
+
+    def test_normal_import_res_partners(self):
+        values = [
+            [
+                "name",
+                "email",
+                "is_company",
+            ],
+            [
+                "partner 1",
+                "partner1@example.com",
+                "1",
+            ],
+            [
+                "partner 2",
+                "partner2@example.com",
+                "0",
+            ],
+        ]
+        import_vals = {
+            "res_model": self.res_partners._name,
+            "file": "\n".join([";".join(values) for values in values]),
+            "file_type": "text/csv",
+        }
+        self.import_wizard |= self.import_wizard.create(import_vals)
+        opts = {"quoting": '"', "separator": ";", "has_headers": True}
+        preview = self.import_wizard.parse_preview(opts)
+        self.assertEqual(
+            preview["matches"],
+            {
+                0: ["name"],
+                1: ["email"],
+                2: ["is_company"],
+            },
+        )
+        with RecordCapturer(self.res_partners, []) as capture:
+            results = self.import_wizard.execute_import(
+                [fnames[0] for fnames in preview["matches"].values()],
+                [],
+                opts,
+            )
+        # if result is empty, no import error
+        self.assertItemsEqual(results["messages"], [])
+        records_created = capture.records
+        self.assertEqual(len(records_created), 2)
+        self.assertIn("partner1", records_created[0].email)
+
+    def test_wrong_import_res_partners(self):
+        values = [
+            [
+                "name",
+                "email",
+                "date",  # Adding date field to trigger parsing error
+            ],
+            [
+                "partner 1",
+                "partner1@example.com",
+                "21-13-2024",
+            ],
+            [
+                "partner 2",
+                "partner2@example.com",
+                "2024-13-45",
+            ],
+        ]
+        opts = {
+            "quoting": '"',
+            "separator": ";",
+            "has_headers": True,
+            "date_format": "%Y-%m-%d",  # Set specific date format
+            OPT_USE_QUEUE: True,
+        }
+        import_vals = {
+            "res_model": self.res_partners._name,
+            "file": "\n".join([";".join(row) for row in values]),
+            "file_type": "text/csv",
+        }
+        import_wizard = self.import_wizard.create(import_vals)
+        preview = import_wizard.parse_preview(opts)
+        results = import_wizard.execute_import(
+            [field[0] for field in preview["matches"].values()],
+            ["name", "email", "date"],  # Include date in fields to import
+            opts,
+        )
+        self.assertTrue(any(msg["type"] == "error" for msg in results["messages"]))