microblog.pub/tests/conftest.py

53 lines
1.5 KiB
Python

from typing import Generator
import pytest
from fastapi.testclient import TestClient
from app.database import Base
from app.database import SessionLocal
from app.database import async_engine
from app.database import async_session
from app.database import engine
from app.main import app
@pytest.fixture
async def async_db_session():
async with async_session() as session:
async with async_engine.begin() as conn:
await conn.run_sync(Base.metadata.create_all)
yield session
async with async_engine.begin() as conn:
await conn.run_sync(Base.metadata.drop_all)
@pytest.fixture
def db() -> Generator:
Base.metadata.create_all(bind=engine)
try:
yield SessionLocal()
finally:
try:
Base.metadata.drop_all(bind=engine)
except Exception:
# XXX: for some reason, the teardown occasionally fails because of this
pass
@pytest.fixture
def exclude_fastapi_middleware():
"""Workaround for https://github.com/encode/starlette/issues/472"""
user_middleware = app.user_middleware.copy()
app.user_middleware = []
app.middleware_stack = app.build_middleware_stack()
yield
app.user_middleware = user_middleware
app.middleware_stack = app.build_middleware_stack()
@pytest.fixture
def client(db, exclude_fastapi_middleware) -> Generator:
# app.dependency_overrides[get_db] = _get_db_for_testing
with TestClient(app) as c:
yield c