Initial commit: Odoo 18.0-20251222 extra-addons

sale_partner_primeship/tests/test_res_partner_primeship.py

@@ -0,0 +1,61 @@
+# Copyright 2024 Akretion - Olivier Nibart
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from datetime import date, timedelta
+
+from .common import TestCommon
+
+
+class TestResPartnerPrimeship(TestCommon):
+    def test_compute_active_primeship(self):
+        active_primeship = self.make_primeship(
+            date.today() - timedelta(days=1), date.today() + timedelta(days=30)
+        )
+        self.assertTrue(active_primeship.current)
+        self.partner._compute_active_primeship()
+        self.assertTrue(self.partner.active_primeship)
+
+        expired_primeship = self.make_primeship(
+            date.today() - timedelta(days=60), date.today() - timedelta(days=30)
+        )
+        self.assertFalse(expired_primeship.current)
+        self.partner._compute_active_primeship()
+        # Still true due to active primeship
+        self.assertTrue(self.partner.active_primeship)
+
+        active_primeship.active = False
+        self.partner._compute_active_primeship()
+        self.assertFalse(self.partner.active_primeship)
+
+    def test_compute_primeship_count(self):
+        self.assertEqual(self.partner.primeship_count, 0)
+
+        self.make_primeship(date.today(), date.today() + timedelta(days=30))
+        self.partner._compute_primeship_count()
+        self.assertEqual(self.partner.primeship_count, 1)
+
+        self.make_primeship(
+            date.today() + timedelta(days=31), date.today() + timedelta(days=60)
+        )
+        self.partner._compute_primeship_count()
+        self.assertEqual(self.partner.primeship_count, 2)
+
+    def test_check_expired_primeships(self):
+        primeship = self.make_primeship(
+            date.today() - timedelta(days=1), date.today() + timedelta(days=30)
+        )
+        self.assertTrue(self.partner.active_primeship)
+        self.Partner._check_expired_primeships()
+        self.assertTrue(self.partner.active_primeship)
+
+        primeship.start_date = date.today() + timedelta(days=1)
+        self.assertFalse(self.partner.active_primeship)
+        self.partner.active_primeship = True
+        self.Partner._check_expired_primeships()
+        self.assertFalse(self.partner.active_primeship)
+
+        # work also on inactive partners
+        self.partner.active = False
+        self.partner.active_primeship = True
+        self.Partner._check_expired_primeships()
+        self.assertFalse(self.partner.active_primeship)