64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
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()
|