Get service provider from FastAPI

Author: AndreuCodina

pyproject.toml

@@ -10,8 +10,8 @@ dependencies = [
     "greenlet>=3.3.1",
     "pydantic>=2.12.5",
     "pydantic-settings[azure-key-vault]>=2.12.0",
-    "sqlmodel>=0.0.31",
-    "wirio>=0.8.0",
+    "sqlmodel>=0.0.32",
+    "wirio>=0.9.0",
 ]
 
 [dependency-groups]
@@ -21,7 +21,7 @@ dev = [
     "pytest-asyncio>=1.3.0",
     "pytest-cov>=7.0.0",
     "pytest-mock>=3.15.1",
-    "ruff>=0.14.14",
+    "ruff>=0.15.0",
     "ty>=0.0.14",
 ]
 

src/python_template/api/service_collection_extensions.py

@@ -18,10 +18,11 @@
 def add_observability(
     services: ServiceCollection, application_settings: ApplicationSettings
 ) -> None:
+    logging.basicConfig(level=application_settings.logging_level)
+
     def inject_logging() -> Logger:
         return logging.getLogger(__name__)
 
-    logging.basicConfig(level=application_settings.logging_level)
     services.add_singleton(inject_logging)
 
     if ApplicationEnvironment.get_current() != ApplicationEnvironment.LOCAL:
@@ -36,18 +37,20 @@ def add_sqlmodel(services: ServiceCollection) -> None:
     def inject_async_engine(application_settings: ApplicationSettings) -> AsyncEngine:
         return create_async_engine(application_settings.postgresql_connection_string)
 
+    services.add_singleton(inject_async_engine)
+
     def inject_async_sessionmaker(
         async_engine: AsyncEngine,
     ) -> async_sessionmaker[AsyncSession]:
         return async_sessionmaker(
             async_engine, class_=AsyncSession, expire_on_commit=False
         )
 
+    services.add_singleton(inject_async_sessionmaker)
+
     def inject_async_session(
         async_sessionmaker: async_sessionmaker[AsyncSession],
     ) -> AsyncSession:
         return async_sessionmaker()
 
-    services.add_singleton(inject_async_engine)
-    services.add_singleton(inject_async_sessionmaker)
     services.add_transient(inject_async_session)

tests/integration/api/conftest.py

@@ -1,12 +1,19 @@
-from collections.abc import AsyncGenerator
+from collections.abc import Generator
 
 import pytest
+from fastapi.testclient import TestClient
+from wirio.integrations.fastapi import get_service_provider
 from wirio.service_provider import ServiceProvider
 
-from python_template.api.main import services
+from python_template.api.main import app
+
+
+@pytest.fixture(autouse=True)
+def test_client() -> Generator[None]:
+    with TestClient(app):
+        yield
 
 
 @pytest.fixture
-async def service_provider() -> AsyncGenerator[ServiceProvider]:
-    async with services.build_service_provider() as service_provider:
-        yield service_provider
+def service_provider() -> ServiceProvider:
+    return get_service_provider(app)