diff --git a/automation_oca/tests/test_automation_date.py b/automation_oca/tests/test_automation_date.py index 07b7178..24a8391 100644 --- a/automation_oca/tests/test_automation_date.py +++ b/automation_oca/tests/test_automation_date.py @@ -10,21 +10,19 @@ class TestAutomationDate(AutomationTestCase): - @classmethod - def setUpClass(cls): - super().setUpClass() - cls.loader = FakeModelLoader(cls.env, cls.__module__) - cls.loader.backup_registry() + def setUp(self): + super().setUp() + self.loader = FakeModelLoader(self.env, self.__module__) + self.loader.backup_registry() # The fake class is imported here !! After the backup_registry from .models import ResPartner - cls.loader.update_registry((ResPartner,)) + self.loader.update_registry((ResPartner,)) - @classmethod - def tearDownClass(cls): - cls.loader.restore_registry() - super().tearDownClass() + def tearDown(self): + self.loader.restore_registry() + return super().tearDown() def test_schedule_date_force(self): partner_01 = self.env["res.partner"].create(