[IMP] extendable_fastapi: Add class to ease tests with fastapi and ex…

…tendable
OCA · Jun 26, 2023 · af9fff5 · af9fff5
1 parent fdc7cc3
commit af9fff5
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/extendable_fastapi/tests/__init__.py b/extendable_fastapi/tests/__init__.py
diff --git a/extendable_fastapi/tests/common.py b/extendable_fastapi/tests/common.py
@@ -0,0 +1,21 @@
+# Copyright 2023 ACSONE SA/NV
+# License LGPL-3.0 or later (http://www.gnu.org/licenses/LGPL).
+
+from odoo.addons.extendable.tests.common import ExtendableMixin
+from odoo.addons.fastapi.tests.common import (
+    FastAPITransactionCase as BaseFastAPITransactionCase,
+)
+
+
+class FastAPITransactionCase(BaseFastAPITransactionCase, ExtendableMixin):
+    """Base class for FastAPI tests using extendable classes."""
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        super().setUpClass()
+        cls.init_extendable_registry()
+
+    @classmethod
+    def tearDownClass(cls) -> None:
+        super().tearDownClass()
+        cls.reset_extendable_registry()