Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-rest-framework-fastapi/fastapi/tests/test_fastapi_demo.py

@@ -0,0 +1,111 @@
+# Copyright 2022 ACSONE SA/NV
+# License LGPL-3.0 or later (http://www.gnu.org/licenses/LGPL).
+
+from functools import partial
+
+from requests import Response
+
+from odoo.exceptions import UserError
+from odoo.tools.misc import mute_logger
+
+from fastapi import status
+
+from ..dependencies import fastapi_endpoint
+from ..routers import demo_router
+from ..schemas import DemoEndpointAppInfo, DemoExceptionType
+from .common import FastAPITransactionCase
+
+
+class FastAPIDemoCase(FastAPITransactionCase):
+    """The fastapi lib comes with a useful testclient that let's you
+    easily test your endpoints. Moreover, the dependency overrides functionality
+    allows you to provide specific implementation for part of the code to avoid
+    to rely on some tricky http stuff for example: authentication
+
+    This test class is an example on how you can test your own code
+    """
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        super().setUpClass()
+        cls.default_fastapi_router = demo_router
+        cls.default_fastapi_running_user = cls.env.ref("fastapi.my_demo_app_user")
+        cls.default_fastapi_authenticated_partner = cls.env["res.partner"].create(
+            {"name": "FastAPI Demo"}
+        )
+
+    def test_hello_world(self) -> None:
+        with self._create_test_client() as test_client:
+            response: Response = test_client.get("/demo/")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertDictEqual(response.json(), {"Hello": "World"})
+
+    def test_who_ami(self) -> None:
+        with self._create_test_client() as test_client:
+            response: Response = test_client.get("/demo/who_ami")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        partner = self.default_fastapi_authenticated_partner
+        self.assertDictEqual(
+            response.json(),
+            {
+                "name": partner.name,
+                "display_name": partner.display_name,
+            },
+        )
+
+    def test_endpoint_info(self) -> None:
+        demo_app = self.env.ref("fastapi.fastapi_endpoint_demo")
+        with self._create_test_client(
+            dependency_overrides={fastapi_endpoint: partial(lambda a: a, demo_app)}
+        ) as test_client:
+            response: Response = test_client.get("/demo/endpoint_app_info")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertDictEqual(
+            response.json(),
+            DemoEndpointAppInfo.model_validate(demo_app).model_dump(by_alias=True),
+        )
+
+    def test_exception_raised(self) -> None:
+        with self.assertRaisesRegex(UserError, "User Error"):
+            with self._create_test_client() as test_client:
+                test_client.get(
+                    "/demo/exception",
+                    params={
+                        "exception_type": DemoExceptionType.user_error.value,
+                        "error_message": "User Error",
+                    },
+                )
+
+        with self.assertRaisesRegex(NotImplementedError, "Bare Exception"):
+            with self._create_test_client() as test_client:
+                test_client.get(
+                    "/demo/exception",
+                    params={
+                        "exception_type": DemoExceptionType.bare_exception.value,
+                        "error_message": "Bare Exception",
+                    },
+                )
+
+    @mute_logger("odoo.addons.fastapi.tests.common")
+    def test_exception_not_raised(self) -> None:
+        with self._create_test_client(raise_server_exceptions=False) as test_client:
+            response: Response = test_client.get(
+                "/demo/exception",
+                params={
+                    "exception_type": DemoExceptionType.user_error.value,
+                    "error_message": "User Error",
+                },
+            )
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertDictEqual(response.json(), {"detail": "User Error"})
+
+        with self._create_test_client(raise_server_exceptions=False) as test_client:
+            response: Response = test_client.get(
+                "/demo/exception",
+                params={
+                    "exception_type": DemoExceptionType.bare_exception.value,
+                    "error_message": "Bare Exception",
+                },
+            )
+        self.assertEqual(response.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
+        self.assertDictEqual(response.json(), {"detail": "Internal Server Error"})