mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
18 lines
551 B
Python
18 lines
551 B
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import AsyncGenerator
|
|
|
|
from sqlalchemy.ext.asyncio.session import AsyncSession
|
|
|
|
from backend.database.db import create_async_engine_and_session, create_database_url
|
|
|
|
TEST_SQLALCHEMY_DATABASE_URL = create_database_url(unittest=True)
|
|
|
|
_, async_test_db_session = create_async_engine_and_session(TEST_SQLALCHEMY_DATABASE_URL)
|
|
|
|
|
|
async def override_get_db() -> AsyncGenerator[AsyncSession, None]:
|
|
"""session 生成器"""
|
|
async with async_test_db_session() as session:
|
|
yield session
|