feat(FN-004): merge fusion/fn-004

apps/api/tests/conftest.py

@@ -0,0 +1,84 @@
+from collections.abc import AsyncGenerator, Generator
+from typing import Any
+
+import pytest
+from httpx import ASGITransport, AsyncClient
+from sqlalchemy.ext.asyncio import (
+    AsyncEngine,
+    AsyncSession,
+    async_sessionmaker,
+    create_async_engine,
+)
+from sqlalchemy.pool import NullPool
+
+from app.config import settings
+from app.db import get_db_session
+from app.main import app
+from app.models import Base
+
+TEST_DATABASE_URL = settings.database_url.replace("/headquarter", "/headquarter_test")
+if TEST_DATABASE_URL.startswith("postgresql://"):
+    TEST_DATABASE_URL = TEST_DATABASE_URL.replace("postgresql://", "postgresql+asyncpg://", 1)
+
+
+@pytest.fixture(scope="session")
+def event_loop() -> Generator[Any, None, None]:
+    import asyncio
+    loop = asyncio.get_event_loop_policy().new_event_loop()
+    yield loop
+    loop.close()
+
+
+@pytest.fixture(scope="session")
+async def db_engine() -> AsyncGenerator[AsyncEngine, None]:
+    engine = create_async_engine(TEST_DATABASE_URL, echo=False, poolclass=NullPool)
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+    yield engine
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.drop_all)
+    await engine.dispose()
+
+
+@pytest.fixture
+async def db_session(
+    db_engine: AsyncEngine,
+) -> AsyncGenerator[async_sessionmaker[AsyncSession], None]:
+    async with db_engine.connect() as connection:
+        trans = await connection.begin_nested()
+        testing_session_local = async_sessionmaker(
+            connection, class_=AsyncSession, expire_on_commit=False
+        )
+
+        async def override_get_db() -> AsyncGenerator[AsyncSession, None]:
+            async with testing_session_local() as session:
+                yield session
+
+        app.dependency_overrides[get_db_session] = override_get_db
+        original_db_url = settings.database_url
+        settings.database_url = TEST_DATABASE_URL
+        yield testing_session_local
+        settings.database_url = original_db_url
+        app.dependency_overrides.pop(get_db_session, None)
+        await trans.rollback()
+
+
+@pytest.fixture
+async def client(
+    db_session: async_sessionmaker[AsyncSession],
+) -> AsyncGenerator[AsyncClient, None]:
+    async with AsyncClient(transport=ASGITransport(app=app), base_url="http://test") as ac:
+        yield ac
+
+
+@pytest.fixture
+async def auth_client(
+    client: AsyncClient,
+) -> AsyncGenerator[AsyncClient, None]:
+    original_debug = settings.debug
+    original_bypass = settings.auth_dev_bypass
+    settings.debug = True
+    settings.auth_dev_bypass = True
+    yield client
+    settings.debug = original_debug
+    settings.auth_dev_bypass = original_bypass