fix: table reflection with async db url - DIA-51295 (#82)

Author: hadrien

fastapi_sqla/_pytest_plugin.py

@@ -178,7 +178,15 @@ async def patch_new_engine(async_sqlalchemy_url, async_sqla_connection, request)
                 yield new_engine
 
     @fixture
-    async def async_session(async_sqla_connection, sqla_reflection, patch_new_engine):
+    async def async_sqla_reflection(sqla_modules, async_sqla_connection):
+        from fastapi_sqla import Base
+
+        await async_sqla_connection.run_sync(lambda conn: Base.prepare(conn.engine))
+
+    @fixture
+    async def async_session(
+        async_sqla_connection, async_sqla_reflection, patch_new_engine
+    ):
         from fastapi_sqla.asyncio_support import _AsyncSession
 
         session = _AsyncSession(bind=async_sqla_connection)

fastapi_sqla/asyncio_support.py

@@ -45,7 +45,9 @@ async def startup():
         )
         raise
 
-    Base.prepare(engine.engine)
+    async with engine.connect() as connection:
+        await connection.run_sync(lambda conn: Base.prepare(conn.engine))
+
     _AsyncSession.configure(bind=engine, expire_on_commit=False)
     logger.info("startup", async_engine=engine)
 

tests/test_middleware.py

@@ -89,7 +89,7 @@ async def a_middleware(request, call_next):
 
 
 @fixture
-async def client(app):
+async def client(app, mock_middleware):
 
     async with LifespanManager(app):
         transport = httpx.ASGITransport(app=app, raise_app_exceptions=False)

tests/test_startup.py

@@ -126,6 +126,6 @@ async def test_async_startup_with_aws_rds_iam_enabled(
 
     await startup()
 
-    boto_client_mock.generate_db_auth_token.assert_called_once_with(
+    boto_client_mock.generate_db_auth_token.assert_called_with(
         DBHostname=db_host, Port=5432, DBUsername=db_user
     )