68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
"""
|
|
Конфигурация для тестов
|
|
"""
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import StaticPool
|
|
from starlette.applications import Starlette
|
|
from starlette.middleware import Middleware
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.testclient import TestClient
|
|
|
|
# Используем in-memory SQLite для тестов
|
|
TEST_DB_URL = "sqlite:///:memory:"
|
|
|
|
|
|
class DatabaseMiddleware(BaseHTTPMiddleware):
|
|
"""Middleware для внедрения сессии БД"""
|
|
|
|
def __init__(self, app, session_maker):
|
|
super().__init__(app)
|
|
self.session_maker = session_maker
|
|
|
|
async def dispatch(self, request, call_next):
|
|
session = self.session_maker()
|
|
request.state.db = session
|
|
try:
|
|
response = await call_next(request)
|
|
finally:
|
|
session.close()
|
|
return response
|
|
|
|
|
|
def create_test_app():
|
|
"""Create a test Starlette application."""
|
|
from services.db import Base
|
|
|
|
# Создаем движок и таблицы
|
|
engine = create_engine(
|
|
TEST_DB_URL,
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool,
|
|
echo=False,
|
|
)
|
|
Base.metadata.drop_all(bind=engine)
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
# Создаем фабрику сессий
|
|
SessionLocal = sessionmaker(bind=engine)
|
|
|
|
# Создаем middleware для сессий
|
|
middleware = [Middleware(DatabaseMiddleware, session_maker=SessionLocal)]
|
|
|
|
# Создаем тестовое приложение
|
|
app = Starlette(
|
|
debug=True,
|
|
middleware=middleware,
|
|
routes=[], # Здесь можно добавить тестовые маршруты если нужно
|
|
)
|
|
|
|
return app, SessionLocal
|
|
|
|
|
|
def get_test_client():
|
|
"""Get a test client with initialized database."""
|
|
app, SessionLocal = create_test_app()
|
|
return TestClient(app), SessionLocal
|