[FIX] fastapi: explicit load of demo data for tests

Author: lmignon

fastapi/models/fastapi_endpoint.py

@@ -11,6 +11,7 @@
 from starlette.routing import Mount
 
 from odoo import api, exceptions, fields, models, tools
+from odoo.tools import convert
 
 from fastapi import APIRouter, Depends, FastAPI
 
@@ -338,3 +339,19 @@ def _get_fastapi_app_middlewares(self) -> list[Middleware]:
     def _get_fastapi_app_dependencies(self) -> list[Depends]:
         """Return the dependencies to use for the fastapi app."""
         return [Depends(dependencies.accept_language)]
+
+    # test utility
+    @api.model
+    def has_demo_data(self):
+        return (
+            self.env.ref("fastapi.fastapi_endpoint_demo", raise_if_not_found=False)
+            is not None
+        )
+
+    def _load_demo_data(self):
+        if self.has_demo_data():
+            return
+        # Load demo data
+        convert.convert_file(
+            self.env, "fastapi", "demo/fastapi_endpoint_demo.xml", None, mode="init"
+        )

fastapi/tests/common.py

@@ -12,10 +12,10 @@
 
 from odoo.api import Environment
 from odoo.tests import tagged
-from odoo.tests.common import TransactionCase
 
 from odoo.addons.base.models.res_partner import ResPartner
 from odoo.addons.base.models.res_users import ResUsers
+from odoo.addons.base.tests.common import TransactionCaseWithUserDemo
 
 from fastapi import APIRouter, FastAPI
 from fastapi.testclient import TestClient
@@ -47,7 +47,7 @@ def default_exception_handler(request: Request, exc: Exception) -> Response:
 
 
 @tagged("post_install", "-at_install")
-class FastAPITransactionCase(TransactionCase):
+class FastAPITransactionCase(TransactionCaseWithUserDemo):
     """
     This class is a base class for FastAPI tests.
 

fastapi/tests/test_fastapi.py

@@ -19,6 +19,7 @@ class FastAPIHttpCase(HttpCase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
+        cls.env["fastapi.endpoint"]._load_demo_data()
         cls.fastapi_demo_app = cls.env.ref("fastapi.fastapi_endpoint_demo")
         cls.fastapi_multi_demo_app = cls.env.ref(
             "fastapi.fastapi_endpoint_multislash_demo"

fastapi/tests/test_fastapi_demo.py

@@ -32,6 +32,7 @@ class FastAPIDemoCase(FastAPITransactionCase):
     @classmethod
     def setUpClass(cls) -> None:
         super().setUpClass()
+        cls.env["fastapi.endpoint"]._load_demo_data()
         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(
@@ -67,7 +68,7 @@ def test_who_ami_http_basic(self) -> None:
         ) as test_client:
             response: Response = test_client.get("/demo/who_ami", auth=("demo", "demo"))
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-        partner = self.env.ref("base.partner_demo")
+        partner = self.partner_demo
         self.assertDictEqual(
             response.json(),
             {