[IMP] edi_voxel_oca: pre-commit stuff for unit tests.

macagua · macagua · commit 55c77cc52f85 · 2025-01-31T18:53:12.000+01:00
diff --git a/edi_voxel_oca/tests/test_queue_job.py b/edi_voxel_oca/tests/test_queue_job.py
@@ -4,23 +4,28 @@
 
 from odoo.tests.common import TransactionCase
 
-class TestQueueJob(TransactionCase):
 
+class TestQueueJob(TransactionCase):
     def setUp(self):
         super(TestQueueJob, self).setUp()
-        self.queue_job = self.env['queue.job'].create({
-            'name': 'Test Job',
-            'state': 'pending',
-        })
+        self.queue_job = self.env["queue.job"].create(
+            {
+                "name": "Test Job",
+                "state": "pending",
+            }
+        )
 
     def test_voxel_do_now(self):
         self.queue_job.voxel_do_now()
         self.assertFalse(self.queue_job.eta, "ETA should be set to False")
 
     def test_voxel_cancel_now(self):
         self.queue_job.voxel_cancel_now()
-        self.assertFalse(self.queue_job.exists(), "Job should be unlinked if in pending, enqueued, or failed state")
+        self.assertFalse(
+            self.queue_job.exists(),
+            "Job should be unlinked if in pending, enqueued, or failed state",
+        )
 
     def test_voxel_requeue_sudo(self):
         self.queue_job.voxel_requeue_sudo()
-        self.assertEqual(self.queue_job.state, 'enqueued', "Job should be requeued")
+        self.assertEqual(self.queue_job.state, "enqueued", "Job should be requeued")
diff --git a/edi_voxel_oca/tests/test_res_company.py b/edi_voxel_oca/tests/test_res_company.py
@@ -2,43 +2,54 @@
 # Copyright 2025 Guavana - Leonardo J. Caballero G.
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from odoo.tests.common import TransactionCase
 from datetime import datetime, timedelta
+
 import pytz
 
-class TestResCompany(TransactionCase):
+from odoo.tests.common import TransactionCase
+
 
+class TestResCompany(TransactionCase):
     def setUp(self):
         super(TestResCompany, self).setUp()
-        self.company = self.env['res.company'].create({
-            'name': 'Test Company',
-            'voxel_enabled': True,
-            'voxel_send_mode': 'auto',
-            'voxel_sent_time': 12.0,
-            'voxel_delay_time': 2.0,
-        })
+        self.company = self.env["res.company"].create(
+            {
+                "name": "Test Company",
+                "voxel_enabled": True,
+                "voxel_send_mode": "auto",
+                "voxel_sent_time": 12.0,
+                "voxel_delay_time": 2.0,
+            }
+        )
 
     def test_get_voxel_report_eta_auto(self):
-        self.company.voxel_send_mode = 'auto'
+        self.company.voxel_send_mode = "auto"
         eta = self.company._get_voxel_report_eta()
         self.assertIsNone(eta, "ETA should be None for auto send mode")
 
     def test_get_voxel_report_eta_fixed(self):
-        self.company.voxel_send_mode = 'fixed'
-        tz = self.env.context.get("tz", self.env.user.partner_id.tz) or 'UTC'
+        self.company.voxel_send_mode = "fixed"
+        tz = self.env.context.get("tz", self.env.user.partner_id.tz) or "UTC"
         offset = datetime.now(pytz.timezone(tz)).strftime("%z")
         hour_diff = int(offset[:3])
         expected_hour = int(12 - hour_diff)
         now = datetime.now()
         if now.hour > expected_hour:
             now += timedelta(days=1)
-        expected_eta = now.replace(hour=expected_hour, minute=0, second=0, microsecond=0)
+        expected_eta = now.replace(
+            hour=expected_hour, minute=0, second=0, microsecond=0
+        )
         eta = self.company._get_voxel_report_eta()
         self.assertEqual(eta, expected_eta, "ETA should match the expected fixed time")
 
     def test_get_voxel_report_eta_delayed(self):
-        self.company.voxel_send_mode = 'delayed'
+        self.company.voxel_send_mode = "delayed"
         expected_eta = datetime.now() + timedelta(hours=2)
         eta = self.company._get_voxel_report_eta()
         self.assertIsNotNone(eta, "ETA should not be None for delayed send mode")
-        self.assertAlmostEqual(eta, expected_eta, msg="ETA should be 2 hours from now for delayed send mode", delta=timedelta(seconds=1))
+        self.assertAlmostEqual(
+            eta,
+            expected_eta,
+            msg="ETA should be 2 hours from now for delayed send mode",
+            delta=timedelta(seconds=1),
+        )
diff --git a/edi_voxel_oca/tests/test_res_partner.py b/edi_voxel_oca/tests/test_res_partner.py
@@ -4,25 +4,33 @@
 
 from odoo.tests.common import TransactionCase
 
-class TestResPartner(TransactionCase):
 
+class TestResPartner(TransactionCase):
     def setUp(self):
         super(TestResPartner, self).setUp()
-        self.partner = self.env['res.partner'].create({
-            'name': 'Test Partner',
-            'vat': 'ES12345678',
-            'voxel_enabled': True,
-        })
+        self.partner = self.env["res.partner"].create(
+            {
+                "name": "Test Partner",
+                "vat": "ES12345678",
+                "voxel_enabled": True,
+            }
+        )
 
     def test_commercial_fields(self):
         commercial_fields = self.partner._commercial_fields()
-        self.assertIn('voxel_enabled', commercial_fields, "voxel_enabled should be in the commercial fields")
+        self.assertIn(
+            "voxel_enabled",
+            commercial_fields,
+            "voxel_enabled should be in the commercial fields",
+        )
 
     def test_get_voxel_vat_with_prefix(self):
         vat = self.partner._get_voxel_vat()
-        self.assertEqual(vat, '12345678', "VAT should be stripped of the 'ES' prefix")
+        self.assertEqual(vat, "12345678", "VAT should be stripped of the 'ES' prefix")
 
     def test_get_voxel_vat_without_prefix(self):
-        self.partner.vat = '12345678'
+        self.partner.vat = "12345678"
         vat = self.partner._get_voxel_vat()
-        self.assertEqual(vat, '12345678', "VAT should remain unchanged if no 'ES' prefix")
+        self.assertEqual(
+            vat, "12345678", "VAT should remain unchanged if no 'ES' prefix"
+        )
diff --git a/edi_voxel_oca/tests/test_voxel_mixin.py b/edi_voxel_oca/tests/test_voxel_mixin.py
@@ -2,68 +2,87 @@
 # Copyright 2025 Guavana - Leonardo J. Caballero G.
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
+from unittest.mock import MagicMock, patch
+
 from odoo.tests.common import TransactionCase
-from unittest.mock import patch, MagicMock
-from datetime import datetime, timedelta
 
-class TestVoxelMixin(TransactionCase):
 
+class TestVoxelMixin(TransactionCase):
     def setUp(self):
         super(TestVoxelMixin, self).setUp()
-        self.company = self.env['res.company'].create({
-            'name': 'Test Company',
-            'voxel_enabled': True,
-        })
-        self.mixin = self.env['voxel.mixin'].create({
-            'voxel_state': 'not_sent',
-            # 'company_id': self.company.id,
-        })
+        self.company = self.env["res.company"].create(
+            {
+                "name": "Test Company",
+                "voxel_enabled": True,
+            }
+        )
+        self.mixin = self.env["voxel.mixin"].create(
+            {
+                "voxel_state": "not_sent",
+                # 'company_id': self.company.id,
+            }
+        )
 
-    @patch('odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.put')
+    @patch("odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.put")
     def test_send_voxel_report_success(self, mock_put):
         mock_put.return_value.status_code = 200
-        self.mixin._send_voxel_report('Outbox', 'test.xml', b'<xml></xml>')
-        self.assertEqual(self.mixin.voxel_state, 'sent', "Voxel state should be 'sent'")
+        self.mixin._send_voxel_report("Outbox", "test.xml", b"<xml></xml>")
+        self.assertEqual(self.mixin.voxel_state, "sent", "Voxel state should be 'sent'")
 
-    @patch('odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.put')
+    @patch("odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.put")
     def test_send_voxel_report_failure(self, mock_put):
         mock_put.return_value.status_code = 500
-        with self.assertRaises(Exception):
-            self.mixin._send_voxel_report('Outbox', 'test.xml', b'<xml></xml>')
-        self.assertEqual(self.mixin.voxel_state, 'sent_errors', "Voxel state should be 'sent_errors'")
+        with self.assertRaises(Exception) as context:
+            self.mixin._send_voxel_report("Outbox", "test.xml", b"<xml></xml>")
+        self.assertTrue("specific error message" in str(context.exception))
+        self.assertEqual(
+            self.mixin.voxel_state, "sent_errors", "Voxel state should be 'sent_errors'"
+        )
 
-    @patch('odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.get')
+    @patch("odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.get")
     def test_list_voxel_document_filenames(self, mock_get):
         mock_get.return_value.status_code = 200
-        mock_get.return_value.content = b'test1.xml\ntest2.xml\n'
-        filenames = self.mixin._list_voxel_document_filenames('Outbox', self.company)
-        self.assertEqual(filenames, ['test1.xml', 'test2.xml'], "Filenames should match the expected list")
+        mock_get.return_value.content = b"test1.xml\ntest2.xml\n"
+        filenames = self.mixin._list_voxel_document_filenames("Outbox", self.company)
+        self.assertEqual(
+            filenames,
+            ["test1.xml", "test2.xml"],
+            "Filenames should match the expected list",
+        )
 
-    @patch('odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.get')
+    @patch("odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.get")
     def test_read_voxel_document(self, mock_get):
         mock_get.return_value.status_code = 200
-        mock_get.return_value.content = b'<xml></xml>'
-        content = self.mixin._read_voxel_document('Inbox', self.company, 'test.xml')
-        self.assertEqual(content, '<xml></xml>', "Content should match the expected XML")
+        mock_get.return_value.content = b"<xml></xml>"
+        content = self.mixin._read_voxel_document("Inbox", self.company, "test.xml")
+        self.assertEqual(
+            content, "<xml></xml>", "Content should match the expected XML"
+        )
 
-    @patch('odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.delete')
+    @patch("odoo.addons.edi_voxel_oca.models.voxel_mixin.requests.delete")
     def test_delete_voxel_document(self, mock_delete):
         mock_delete.return_value.status_code = 200
-        self.mixin._delete_voxel_document('Inbox', 'test.xml', self.company)
+        self.mixin._delete_voxel_document("Inbox", "test.xml", self.company)
         mock_delete.assert_called_once()
 
     def test_get_voxel_filename(self):
-        self.mixin.get_document_type = MagicMock(return_value='INV')
+        self.mixin.get_document_type = MagicMock(return_value="INV")
         filename = self.mixin._get_voxel_filename()
-        self.assertTrue(filename.startswith('INV_'), "Filename should start with 'INV_'")
-        self.assertTrue(filename.endswith('.xml'), "Filename should end with '.xml'")
+        self.assertTrue(
+            filename.startswith("INV_"), "Filename should start with 'INV_'"
+        )
+        self.assertTrue(filename.endswith(".xml"), "Filename should end with '.xml'")
 
     def test_cancel_voxel_jobs(self):
-        job = self.env['queue.job'].create({
-            'name': 'Test Job',
-            'state': 'pending',
-        })
+        job = self.env["queue.job"].create(
+            {
+                "name": "Test Job",
+                "state": "pending",
+            }
+        )
         self.mixin.voxel_job_ids = [(4, job.id)]
         self.mixin._cancel_voxel_jobs()
-        self.assertEqual(self.mixin.voxel_state, 'cancelled', "Voxel state should be 'cancelled'")
+        self.assertEqual(
+            self.mixin.voxel_state, "cancelled", "Voxel state should be 'cancelled'"
+        )
         self.assertFalse(job.exists(), "Job should be unlinked")
