Aleksandr Meshchriakov
a91ed1f1ae
Some checks failed
CI/CD Pipeline / Code Quality Checks (push) Failing after 3m10s
CI/CD Pipeline / Run Tests (push) Successful in 3m35s
CI/CD Pipeline / Telegram Notify Success (push) Has been skipped
CI/CD Pipeline / Code Quality Checks (pull_request) Failing after 2m26s
CI/CD Pipeline / Run Tests (pull_request) Successful in 2m46s
CI/CD Pipeline / Telegram Notify Success (pull_request) Has been skipped
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
"""Tests for core middleware"""
|
|
|
|
import logging
|
|
from io import StringIO
|
|
|
|
from apps.core.middleware import (
|
|
RequestIDMiddleware,
|
|
RequestLoggingMiddleware,
|
|
get_request_id,
|
|
)
|
|
from django.http import HttpResponse
|
|
from django.test import RequestFactory
|
|
from django.urls import reverse
|
|
from rest_framework.test import APITestCase
|
|
|
|
|
|
class RequestIDMiddlewareTest(APITestCase):
|
|
"""Tests for RequestIDMiddleware"""
|
|
|
|
def test_request_id_generated(self):
|
|
"""Test that request ID is generated and returned in response header"""
|
|
url = reverse("core:health")
|
|
response = self.client.get(url)
|
|
|
|
self.assertIn("X-Request-ID", response)
|
|
self.assertIsNotNone(response["X-Request-ID"])
|
|
# UUID format check (36 chars with hyphens)
|
|
self.assertEqual(len(response["X-Request-ID"]), 36)
|
|
|
|
def test_request_id_passed_through(self):
|
|
"""Test that provided X-Request-ID is passed through"""
|
|
url = reverse("core:health")
|
|
custom_id = "custom-request-id-12345"
|
|
response = self.client.get(url, HTTP_X_REQUEST_ID=custom_id)
|
|
|
|
self.assertEqual(response["X-Request-ID"], custom_id)
|
|
|
|
def test_different_requests_different_ids(self):
|
|
"""Test that different requests get different IDs"""
|
|
url = reverse("core:health")
|
|
response1 = self.client.get(url)
|
|
response2 = self.client.get(url)
|
|
|
|
self.assertNotEqual(response1["X-Request-ID"], response2["X-Request-ID"])
|
|
|
|
|
|
class RequestLoggingMiddlewareTest(APITestCase):
|
|
def setUp(self):
|
|
self.factory = RequestFactory()
|
|
self.logger = logging.getLogger("apps.core.middleware")
|
|
self.logger.setLevel(logging.INFO)
|
|
self.stream = StringIO()
|
|
handler = logging.StreamHandler(self.stream)
|
|
self.logger.handlers = []
|
|
self.logger.addHandler(handler)
|
|
|
|
def test_process_request_and_response_logs(self):
|
|
middleware = RequestLoggingMiddleware(lambda req: HttpResponse(status=200))
|
|
request = self.factory.get("/health/")
|
|
request.request_id = "req-123"
|
|
middleware.process_request(request)
|
|
response = middleware.process_response(request, HttpResponse(status=200))
|
|
self.assertEqual(response.status_code, 200)
|
|
output = self.stream.getvalue()
|
|
self.assertIn("Started", output)
|
|
self.assertIn("200", output)
|
|
|
|
def test_request_id_middleware_exception(self):
|
|
middleware = RequestIDMiddleware(lambda req: None)
|
|
request = self.factory.get("/health/")
|
|
middleware.process_request(request)
|
|
middleware.process_exception(request, RuntimeError("boom"))
|
|
response = middleware.process_response(request, HttpResponse(status=200))
|
|
self.assertIn("X-Request-ID", response)
|
|
self.assertIsNone(get_request_id())
|