Improve topic sorting: add popular sorting by publications and authors count

tests/test_config.py

@@ -8,8 +8,28 @@ from sqlalchemy.pool import StaticPool
 from starlette.applications import Starlette
 from starlette.middleware import Middleware
 from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.routing import Route
 from starlette.testclient import TestClient
 
+# Импортируем все модели чтобы SQLAlchemy знал о них
+from auth.orm import (  # noqa: F401
+    Author,
+    AuthorBookmark,
+    AuthorFollower,
+    AuthorRating,
+    AuthorRole,
+    Permission,
+    Role,
+    RolePermission,
+)
+from orm.collection import ShoutCollection  # noqa: F401
+from orm.community import Community, CommunityAuthor, CommunityFollower  # noqa: F401
+from orm.draft import Draft, DraftAuthor, DraftTopic  # noqa: F401
+from orm.invite import Invite  # noqa: F401
+from orm.notification import Notification  # noqa: F401
+from orm.shout import Shout, ShoutReactionsFollower, ShoutTopic  # noqa: F401
+from orm.topic import Topic, TopicFollower  # noqa: F401
+
 # Используем in-memory SQLite для тестов
 TEST_DB_URL = "sqlite:///:memory:"
 
@@ -33,7 +53,14 @@ class DatabaseMiddleware(BaseHTTPMiddleware):
 
 def create_test_app():
     """Create a test Starlette application."""
+    from importlib import import_module
+
+    from ariadne import load_schema_from_path, make_executable_schema
+    from ariadne.asgi import GraphQL
+    from starlette.responses import JSONResponse
+
     from services.db import Base
+    from services.schema import resolvers
 
     # Создаем движок и таблицы
     engine = create_engine(
@@ -46,22 +73,60 @@ def create_test_app():
     Base.metadata.create_all(bind=engine)
 
     # Создаем фабрику сессий
-    SessionLocal = sessionmaker(bind=engine)
+    session_local = sessionmaker(bind=engine)
+
+    # Импортируем резолверы для GraphQL
+    import_module("resolvers")
+
+    # Создаем схему GraphQL
+    schema = make_executable_schema(load_schema_from_path("schema/"), list(resolvers))
+
+    # Создаем кастомный GraphQL класс для тестов
+    class TestGraphQL(GraphQL):
+        async def get_context_for_request(self, request, data):
+            """Переопределяем контекст для тестов"""
+            context = {
+                "request": None,  # Устанавливаем None для активации тестового режима
+                "author": None,
+                "roles": [],
+            }
+
+            # Для тестов, если есть заголовок авторизации, создаем мок пользователя
+            auth_header = request.headers.get("authorization")
+            if auth_header and auth_header.startswith("Bearer "):
+                # Простая мок авторизация для тестов - создаем пользователя с ID 1
+                context["author"] = {"id": 1, "name": "Test User"}
+                context["roles"] = ["reader", "author"]
+
+            return context
+
+    # Создаем GraphQL приложение с кастомным классом
+    graphql_app = TestGraphQL(schema, debug=True)
+
+    async def graphql_handler(request):
+        """Простой GraphQL обработчик для тестов"""
+        try:
+            return await graphql_app.handle_request(request)
+        except Exception as e:
+            return JSONResponse({"error": str(e)}, status_code=500)
 
     # Создаем middleware для сессий
-    middleware = [Middleware(DatabaseMiddleware, session_maker=SessionLocal)]
+    middleware = [Middleware(DatabaseMiddleware, session_maker=session_local)]
 
-    # Создаем тестовое приложение
+    # Создаем тестовое приложение с GraphQL маршрутом
     app = Starlette(
         debug=True,
         middleware=middleware,
-        routes=[],  # Здесь можно добавить тестовые маршруты если нужно
+        routes=[
+            Route("/", graphql_handler, methods=["GET", "POST"]),  # Основной GraphQL эндпоинт
+            Route("/graphql", graphql_handler, methods=["GET", "POST"]),  # Альтернативный путь
+        ],
     )
 
-    return app, SessionLocal
+    return app, session_local
 
 
 def get_test_client():
     """Get a test client with initialized database."""
-    app, SessionLocal = create_test_app()
-    return TestClient(app), SessionLocal
+    app, session_local = create_test_app()
+    return TestClient(app), session_local