🐛 Fix: Refactor app startup and engine initialization in test setup

Author: pcrespov

services/api-server/tests/unit/_with_db/test_repository_api_keys.py

@@ -3,41 +3,55 @@
 # pylint: disable=unused-variable
 # pylint: disable=too-many-arguments
 
-from collections.abc import AsyncGenerator, Callable
+from collections.abc import AsyncGenerator, AsyncIterator, Callable
 
-import httpx
 import pytest
+from asgi_lifespan import LifespanManager
+from fastapi import FastAPI
 from fastapi.security import HTTPBasicCredentials
 from models_library.api_schemas_api_server.api_keys import ApiKeyInDB
 from pydantic import PositiveInt
 from simcore_service_api_server.api.dependencies.authentication import (
     get_current_identity,
 )
+from simcore_service_api_server.clients.postgres import get_engine
 from simcore_service_api_server.repository.api_keys import ApiKeysRepository
 from simcore_service_api_server.repository.users import UsersRepository
 from sqlalchemy.ext.asyncio import AsyncEngine
 
+MAX_TIME_FOR_APP_TO_STARTUP = 10
+MAX_TIME_FOR_APP_TO_SHUTDOWN = 10
+
 
 @pytest.fixture
-def async_engine_after_app_started(
-    client: httpx.AsyncClient,  # ensures app context is available
-    async_engine: AsyncEngine,
-) -> AsyncEngine:
-    return async_engine
+async def app_started(app: FastAPI, is_pdb_enabled: bool) -> AsyncIterator[FastAPI]:
+    # LifespanManager will trigger app's startup&shutown event handlers
+    async with LifespanManager(
+        app,
+        startup_timeout=None if is_pdb_enabled else MAX_TIME_FOR_APP_TO_STARTUP,
+        shutdown_timeout=None if is_pdb_enabled else MAX_TIME_FOR_APP_TO_SHUTDOWN,
+    ):
+        yield app
+
+
+@pytest.fixture
+async def async_engine(app_started: FastAPI) -> AsyncEngine:
+    # Overrides
+    return get_engine(app_started)
 
 
 @pytest.fixture
 def api_key_repo(
-    async_engine_after_app_started: AsyncEngine,
+    async_engine: AsyncEngine,
 ) -> ApiKeysRepository:
-    return ApiKeysRepository(db_engine=async_engine_after_app_started)
+    return ApiKeysRepository(db_engine=async_engine)
 
 
 @pytest.fixture
 def users_repo(
-    async_engine_after_app_started: AsyncEngine,
+    async_engine: AsyncEngine,
 ) -> UsersRepository:
-    return UsersRepository(db_engine=async_engine_after_app_started)
+    return UsersRepository(db_engine=async_engine)
 
 
 async def test_get_user_with_valid_credentials(