Coverage for tests/test_exception_handlers.py: 100%

1import pytest 5 ctx1abcde

2from fastapi import FastAPI, HTTPException 5 ctx1abcde

3from fastapi.exceptions import RequestValidationError 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5from starlette.responses import JSONResponse 5 ctx1abcde

6 

7 

8def http_exception_handler(request, exception): 5 ctx1abcde

9 return JSONResponse({"exception": "http-exception"}) 5 ctx1abcde

10 

11 

12def request_validation_exception_handler(request, exception): 5 ctx1abcde

13 return JSONResponse({"exception": "request-validation"}) 5 ctx1abcde

14 

15 

16def server_error_exception_handler(request, exception): 5 ctx1abcde

17 return JSONResponse(status_code=500, content={"exception": "server-error"}) 5 ctx1abcde

18 

19 

20app = FastAPI( 5 ctx1abcde

21 exception_handlers={ 

22 HTTPException: http_exception_handler, 

23 RequestValidationError: request_validation_exception_handler, 

24 Exception: server_error_exception_handler, 

25 } 

26) 

27 

28client = TestClient(app) 5 ctx1abcde

29 

30 

31@app.get("/http-exception") 5 ctx1abcde

32def route_with_http_exception(): 5 ctx1abcde

33 raise HTTPException(status_code=400) 5 ctx1abcde

34 

35 

36@app.get("/request-validation/{param}/") 5 ctx1abcde

37def route_with_request_validation_exception(param: int): 5 ctx1abcde

38 pass # pragma: no cover 

39 

40 

41@app.get("/server-error") 5 ctx1abcde

42def route_with_server_error(): 5 ctx1abcde

43 raise RuntimeError("Oops!") 5 ctx1abcde

44 

45 

46def test_override_http_exception(): 5 ctx1abcde

47 response = client.get("/http-exception") 5 ctx1abcde

48 assert response.status_code == 200 5 ctx1abcde

49 assert response.json() == {"exception": "http-exception"} 5 ctx1abcde

50 

51 

52def test_override_request_validation_exception(): 5 ctx1abcde

53 response = client.get("/request-validation/invalid") 5 ctx1abcde

54 assert response.status_code == 200 5 ctx1abcde

55 assert response.json() == {"exception": "request-validation"} 5 ctx1abcde

56 

57 

58def test_override_server_error_exception_raises(): 5 ctx1abcde

59 with pytest.raises(RuntimeError): 5 ctx1abcde

60 client.get("/server-error") 5 ctx1abcde

61 

62 

63def test_override_server_error_exception_response(): 5 ctx1abcde

64 client = TestClient(app, raise_server_exceptions=False) 5 ctx1abcde

65 response = client.get("/server-error") 5 ctx1abcde

66 assert response.status_code == 500 5 ctx1abcde

67 assert response.json() == {"exception": "server-error"} 5 ctx1abcde