Добавлен класс RequestManager

2026-07-16 19:53:53 +03:00
parent 63537f24a0
commit 3d1e6b47bd
10 changed files with 674 additions and 62 deletions
--- a/net/xeaf/rack/core/tests/core_test_case.py
+++ b/net/xeaf/rack/core/tests/core_test_case.py
@@ -9,6 +9,7 @@
 """

 import json
+from typing import Any

 from django.core.management import call_command
 from django.test import TestCase
@@ -16,8 +17,10 @@ from rest_framework.parsers import JSONParser
 from rest_framework.request import Request
 from rest_framework.test import APIRequestFactory

+from net.xeaf.rack.enums import HttpMethod
 from net.xeaf.rack.management.userstory import AccountTestData
 from net.xeaf.rack.management.userstory import SessionTestData
+from net.xeaf.rack.managers import SessionManager
 from net.xeaf.rack.models import AccountModel
 from net.xeaf.rack.models import SessionModel

@@ -38,12 +41,19 @@ class CoreTestCase(TestCase):
        call_command("rack_test_database")
        self.factory = APIRequestFactory()

-    def create_request(self, method, path, data=None, query_params=None, auth=None):
+    def create_request(self,
+                       method: HttpMethod,
+                       path: str,
+                       headers: dict[str, str] | None = None,
+                       data: dict[str, Any] | None = None,
+                       query_params: dict[str, str] | None = None,
+                       auth: SessionModel | None = None) -> Request:
        """
-        Создает DRF Request объект для тестирования
+        Создает объект запроса для тестирования

        :param method: HTTP метод (GET, POST, PUT, PATCH, DELETE)
        :param path: URL путь
+        :param headers: Заголовки запроса
        :param data: Данные для тела запроса
        :param query_params: Параметры запроса
        :param auth: Объект аутентификации (сессия)
@@ -51,69 +61,126 @@ class CoreTestCase(TestCase):
        :return: DRF Request объект
        """

-        method = method.upper()
-        data = data or {}
+        # Уточняем значения параметров
+        headers = self.__prepare_headers(headers or {})
        query_params = query_params or {}
+        data = data or {}

-        # Создаем WSGI request в зависимости от метода
-        if method == 'GET':
-            wsgi_request = self.factory.get(path, data=query_params)
-        elif method == 'POST':
-            wsgi_request = self.factory.post(path, data=data, format='json')
-        elif method == 'PUT':
-            wsgi_request = self.factory.put(path, data=data, format='json')
-        elif method == 'PATCH':
-            wsgi_request = self.factory.patch(path, data=data, format='json')
-        elif method == 'DELETE':
-            wsgi_request = self.factory.delete(path, data=data, format='json')
-        else:
+        # Создаем внутренний объект запроса
+        match method:
+            case HttpMethod.GET:
+                wsgi_request = self.factory.get(path, data=query_params, **headers)
+            case HttpMethod.POST:
+                wsgi_request = self.factory.post(path, data=data, format='json', **headers)
+            case HttpMethod.PUT:
+                wsgi_request = self.factory.put(path, data=data, format='json', **headers)
+            case HttpMethod.PATCH:
+                wsgi_request = self.factory.patch(path, data=data, format='json', **headers)
+            case HttpMethod.DELETE:
+                wsgi_request = self.factory.delete(path, **headers)
+            case _:
+                wsgi_request = self.factory.generic(
+                    method,
+                    path,
+                    content_type='application/json',
+                    data=json.dumps(data) if data else ''
+                )

-            wsgi_request = self.factory.generic(
-                method,
-                path,
-                content_type='application/json',
-                data=json.dumps(data) if data else ''
-            )
-
-        # Преобразуем в DRF Request
+        # Преобразуем к объекту запроса DRF
        drf_request = Request(wsgi_request, parsers=[JSONParser()])

-        # Добавляем аутентификацию если передана
-        if auth:
-            drf_request.auth = auth
-            if hasattr(auth, 'account'):
-                drf_request.user = auth.account
-
-        # Добавляем query_params для не-GET запросов
-        if query_params and method != 'GET':
-            # noinspection PyTypeChecker
-            drf_request.query_params._mutable = True
-            for key, value in query_params.items():
-                drf_request.query_params[key] = value
-            drf_request.query_params._mutable = False
+        if method != HttpMethod.GET:
+            self.__assign_query_params(drf_request, query_params)
+        self.__assign_auth(drf_request, auth)

        return drf_request

-    def create_authenticated_request(self, method, path, token, data=None, query_params=None):
+    def create_authenticated_request(self,
+                                     method: HttpMethod,
+                                     path: str,
+                                     headers: dict[str, str] | None = None,
+                                     data: dict[str, Any] | None = None,
+                                     query_params: dict[str, str] | None = None,
+                                     token: str | None = None) -> Request:
        """
-        Создает аутентифицированный DRF Request
+        Создает объект авторизованного запроса для тестирования

-        :param method: HTTP метод
+        :param method: HTTP метод (GET, POST, PUT, PATCH, DELETE)
        :param path: URL путь
-        :param token: Токен сессии
+        :param headers: Заголовки запроса
        :param data: Данные для тела запроса
        :param query_params: Параметры запроса
-        :return: DRF Request с аутентификацией
+        :param token: Токен сессии авторизации
+
+        :return: DRF Request объект
        """

-        session = SessionModel.objects.get(token=token)
-        return self.create_request(method, path, data, query_params, auth=session)
+        # Получаем сессию
+        if token is None:
+            session = None
+        else:
+            session = SessionManager.find_by_token(token)
+
+        # Возвращаем объект запроса
+        return self.create_request(method=method, path=path, headers=headers, data=data, query_params=query_params, auth=session)
+
+    @classmethod
+    def __prepare_headers(cls, headers: dict[str, str]) -> dict[str, str]:
+        """
+        Подготавливаем объект заголовков для передачи в фабрику
+
+        :param headers: Заголовки запроса
+
+        :return: Объект заголовков для фабрики
+        """
+
+        wsgi_headers = {}
+        for key, value in headers.items():
+            if key.startswith('HTTP_'):
+                wsgi_headers[key] = value
+            else:
+                norm_key = key.upper().replace('-', '_')
+                if norm_key in ('CONTENT_TYPE', 'CONTENT_LENGTH'):
+                    wsgi_headers[norm_key.lower()] = value
+                else:
+                    wsgi_headers[f'HTTP_{norm_key}'] = value
+
+        return wsgi_headers
+
+    @classmethod
+    def __assign_query_params(cls, request: Request, query_params: dict[str, Any]) -> None:
+        """
+        Присваивает параметры запроса в DRF Request объект
+
+        :param request: DRF Request объект
+        :param query_params: Параметры запроса
+        """
+
+        # noinspection PyTypeChecker
+        request.query_params._mutable = True
+        for key, value in query_params.items():
+            request.query_params[key] = value
+        request.query_params._mutable = False
+
+    @classmethod
+    def __assign_auth(cls, request: Request, auth: SessionModel | None) -> None:
+        """
+        Присваивает аутентификацию в DRF Request объект
+
+        :param request: Объект запроса
+        :param auth: Объект сессии пользователя
+        """
+
+        if auth:
+            request.auth = auth
+            if hasattr(auth, 'account'):
+                request.user = auth.account

    @classmethod
    def get_i_ivanov_account(cls) -> AccountModel:
        """
        Возвращает аккаунт Ивана И. Иванова
-        
+
        :return: Аккаунт Ивана И. Иванова
        """

@@ -123,30 +190,39 @@ class CoreTestCase(TestCase):
    def get_i_ivanov_session(cls) -> SessionModel:
        """
        Возвращает сессию Ивана И. Иванова
-        
+
        :return: Сессия Ивана И. Иванова
        """

        return SessionTestData.get_i_ivanov_session()

-    def create_i_ivanov_request(self, method, path, data=None, query_params=None):
+    def create_i_ivanov_request(self, method: HttpMethod, path: str, headers: dict[str, str] | None = None, data: dict[str, Any] | None = None, query_params: dict[str, Any] | None = None) -> Request:
        """
        Создает запрос от имени Ивана И. Иванова
+
+        :param method: HTTP метод (GET, POST, PUT, PATCH, DELETE)
+        :param path: URL путь
+        :param headers: Заголовки запроса
+        :param data: Данные для тела запроса
+        :param query_params: Параметры запроса
+
+        :return: Объект запроса с аутентификацией Ивана И. Иванова
        """

        return self.create_authenticated_request(
-            method,
-            path,
-            SessionTestData.I_IVANOV_SESSION_TOKEN,
-            data,
-            query_params
+            method=method,
+            path=path,
+            headers=headers,
+            data=data,
+            query_params=query_params,
+            token=SessionTestData.I_IVANOV_SESSION_TOKEN,
        )

    @classmethod
    def get_p_petrov_account(cls) -> AccountModel:
        """
        Возвращает аккаунт Петра П. Петрова
-        
+
        :return: Аккаунт Петра П. Петрова
        """
        return AccountTestData.get_p_petrov_account()
@@ -155,21 +231,30 @@ class CoreTestCase(TestCase):
    def get_p_petrov_session(cls) -> SessionModel:
        """
        Возвращает сессию Петра П. Петрова
-        
+
        :return: Сессия Петра П. Петрова
        """

        return SessionTestData.get_p_petrov_session()

-    def create_p_petrov_request(self, method, path, data=None, query_params=None):
+    def create_p_petrov_request(self, method: HttpMethod, path: str, headers: dict[str, str] | None = None, data: dict[str, Any] | None = None, query_params: dict[str, Any] | None = None) -> Request:
        """
        Создает запрос от имени Петра П. Петрова
+
+        :param method: HTTP метод (GET, POST, PUT, PATCH, DELETE)
+        :param path: URL путь
+        :param headers: Заголовки запроса
+        :param data: Данные для тела запроса
+        :param query_params: Параметры запроса
+
+        :return: Объект запроса с аутентификацией Петра П. Петрова
        """

        return self.create_authenticated_request(
-            method,
-            path,
-            SessionTestData.P_PETROV_SESSION_TOKEN,
-            data,
-            query_params
+            method=method,
+            path=path,
+            headers=headers,
+            data=data,
+            query_params=query_params,
+            token=SessionTestData.P_PETROV_SESSION_TOKEN
        )