feat: add foreign key constraints and metrics for blockchain node

apps/pool-hub/tests/conftest.py

@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+import os
+import sys
+from pathlib import Path
+
+import pytest
+import pytest_asyncio
+from redis.asyncio import Redis
+from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker, create_async_engine
+
+BASE_DIR = Path(__file__).resolve().parents[2]
+POOLHUB_SRC = BASE_DIR / "pool-hub" / "src"
+if str(POOLHUB_SRC) not in sys.path:
+    sys.path.insert(0, str(POOLHUB_SRC))
+
+from poolhub.models import Base
+
+
+def _get_required_env(name: str) -> str:
+    value = os.getenv(name)
+    if not value:
+        pytest.skip(f"Set {name} to run Pool Hub integration tests")
+    return value
+
+
+@pytest_asyncio.fixture()
+async def db_engine() -> AsyncEngine:
+    dsn = _get_required_env("POOLHUB_TEST_POSTGRES_DSN")
+    engine = create_async_engine(dsn, pool_pre_ping=True)
+
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.drop_all)
+        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_asyncio.fixture
+async def db_session(db_engine: AsyncEngine) -> AsyncSession:
+    session_factory = async_sessionmaker(db_engine, expire_on_commit=False, autoflush=False)
+    async with session_factory() as session:
+        yield session
+        await session.rollback()
+
+
+@pytest_asyncio.fixture()
+async def redis_client() -> Redis:
+    redis_url = _get_required_env("POOLHUB_TEST_REDIS_URL")
+    client = Redis.from_url(redis_url, encoding="utf-8", decode_responses=True)
+    await client.flushdb()
+    yield client
+    await client.flushdb()
+    await client.close()
+
+
+@pytest_asyncio.fixture(autouse=True)
+async def _clear_redis(redis_client: Redis) -> None:
+    await redis_client.flushdb()