async def db_session() -> AsyncGenerator[AsyncSession, None]:
"""Create test database session that will then be reverted after test run"""
async with engine.begin() as connection:
await connection.run_sync(Base.metadata.drop_all)
await connection.run_sync(Base.metadata.create_all)
async with async_session(bind=connection) as session:
yield session
await session.flush()
await session.rollback()
async def get_db() -> AsyncGenerator:
"""
Зависимость для сесси к бд
"""
try:
db = async_session()
yield db
finally:
await db.close()
async def db() -> Generator:
async with async_session() as session:
yield session
async def init() -> None:
async with async_session() as db:
await init_db(db)