token-storage-refactored

tests/conftest.py

@@ -1,9 +1,81 @@
 import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.pool import StaticPool
 
+from services.db import Base
 from services.redis import redis
 from tests.test_config import get_test_client
 
 
+@pytest.fixture(scope="session")
+def test_engine():
+    """
+    Создает тестовый engine для всей сессии тестирования.
+    Использует in-memory SQLite для быстрых тестов.
+    """
+    engine = create_engine(
+        "sqlite:///:memory:", echo=False, poolclass=StaticPool, connect_args={"check_same_thread": False}
+    )
+
+    # Создаем все таблицы
+    Base.metadata.create_all(engine)
+
+    yield engine
+
+    # Cleanup после всех тестов
+    Base.metadata.drop_all(engine)
+
+
+@pytest.fixture(scope="session")
+def test_session_factory(test_engine):
+    """
+    Создает фабрику сессий для тестирования.
+    """
+    return sessionmaker(bind=test_engine, expire_on_commit=False)
+
+
+@pytest.fixture
+def db_session(test_session_factory):
+    """
+    Создает новую сессию БД для каждого теста.
+    Простая реализация без вложенных транзакций.
+    """
+    session = test_session_factory()
+    yield session
+
+    # Очищаем все данные после теста
+    try:
+        for table in reversed(Base.metadata.sorted_tables):
+            session.execute(table.delete())
+        session.commit()
+    except Exception:
+        session.rollback()
+    finally:
+        session.close()
+
+
+@pytest.fixture
+def db_session_commit(test_session_factory):
+    """
+    Создает сессию БД с реальными commit'ами для интеграционных тестов.
+    Используется когда нужно тестировать реальные транзакции.
+    """
+    session = test_session_factory()
+
+    yield session
+
+    # Очищаем все данные после теста
+    try:
+        for table in reversed(Base.metadata.sorted_tables):
+            session.execute(table.delete())
+        session.commit()
+    except Exception:
+        session.rollback()
+    finally:
+        session.close()
+
+
 @pytest.fixture(scope="session")
 def test_app():
     """Create a test client and session factory."""
@@ -11,18 +83,6 @@ def test_app():
     return client, session_local
 
 
-@pytest.fixture
-def db_session(test_app):
-    """Create a new database session for a test."""
-    _, session_local = test_app
-    session = session_local()
-
-    yield session
-
-    session.rollback()
-    session.close()
-
-
 @pytest.fixture
 def test_client(test_app):
     """Get the test client."""
@@ -33,8 +93,43 @@ def test_client(test_app):
 @pytest.fixture
 async def redis_client():
     """Create a test Redis client."""
-    await redis.connect()
-    await redis.flushall()  # Очищаем Redis перед каждым тестом
-    yield redis
-    await redis.flushall()  # Очищаем после теста
-    await redis.disconnect()
+    try:
+        await redis.connect()
+        await redis.execute("FLUSHALL")  # Очищаем Redis перед каждым тестом
+        yield redis
+        await redis.execute("FLUSHALL")  # Очищаем после теста
+    finally:
+        try:
+            await redis.disconnect()
+        except Exception:
+            pass
+
+
+@pytest.fixture
+def oauth_db_session(test_session_factory):
+    """
+    Fixture для dependency injection OAuth модуля с тестовой БД.
+    Настраивает OAuth модуль на использование тестовой сессии.
+    """
+    # Импортируем OAuth модуль и настраиваем dependency injection
+    from auth import oauth
+
+    # Сохраняем оригинальную фабрику через SessionManager
+    original_factory = oauth.session_manager._factory
+
+    # Устанавливаем тестовую фабрику
+    oauth.set_session_factory(lambda: test_session_factory())
+
+    session = test_session_factory()
+    yield session
+
+    # Очищаем данные и восстанавливаем оригинальную фабрику
+    try:
+        for table in reversed(Base.metadata.sorted_tables):
+            session.execute(table.delete())
+        session.commit()
+    except Exception:
+        session.rollback()
+    finally:
+        session.close()
+        oauth.session_manager.set_factory(original_factory)