Spaces:
Sleeping
Sleeping
File size: 2,642 Bytes
2aa2b79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
"""
Tests for error handling functionality.
"""
import pytest
from unittest.mock import Mock, patch
from fastapi import FastAPI, Request
from app.core.errors import init_exception_handlers
class TestErrorHandlers:
"""Test error handling functionality."""
def test_init_exception_handlers(self):
"""Test that exception handlers are initialized."""
app = FastAPI()
init_exception_handlers(app)
# Verify exception handler was registered
assert Exception in app.exception_handlers
@pytest.mark.asyncio
async def test_unhandled_exception_handler(self):
"""Test unhandled exception handler."""
app = FastAPI()
init_exception_handlers(app)
# Create a mock request with request_id
request = Mock(spec=Request)
request.state.request_id = "test-request-id"
# Create a test exception
test_exception = Exception("Test error")
# Get the exception handler
handler = app.exception_handlers[Exception]
# Test the handler
response = await handler(request, test_exception)
# Verify response
assert response.status_code == 500
assert response.headers["content-type"] == "application/json"
# Verify response content
import json
content = json.loads(response.body.decode())
assert content["detail"] == "Internal server error"
assert content["code"] == "INTERNAL_ERROR"
assert content["request_id"] == "test-request-id"
@pytest.mark.asyncio
async def test_unhandled_exception_handler_no_request_id(self):
"""Test unhandled exception handler without request ID."""
app = FastAPI()
init_exception_handlers(app)
# Create a mock request without request_id
request = Mock(spec=Request)
request.state = Mock()
del request.state.request_id # Remove request_id
# Create a test exception
test_exception = Exception("Test error")
# Get the exception handler
handler = app.exception_handlers[Exception]
# Test the handler
response = await handler(request, test_exception)
# Verify response
assert response.status_code == 500
# Verify response content
import json
content = json.loads(response.body.decode())
assert content["detail"] == "Internal server error"
assert content["code"] == "INTERNAL_ERROR"
assert content["request_id"] is None
|