Coverage for tests / test_deprecated_responses.py: 100%

1import warnings 4 ctx1adbc

2 

3import pytest 4 ctx1adbc

4from fastapi import FastAPI 4 ctx1adbc

5from fastapi.exceptions import FastAPIDeprecationWarning 4 ctx1adbc

6from fastapi.responses import ORJSONResponse, UJSONResponse 4 ctx1adbc

7from fastapi.testclient import TestClient 4 ctx1adbc

8from pydantic import BaseModel 4 ctx1adbc

9 

10 

11class Item(BaseModel): 4 ctx1adbc

12 name: str 3 ctx1abc

13 price: float 3 ctx1abc

14 

15 

16# ORJSON 

17 

18 

19def _make_orjson_app() -> FastAPI: 4 ctx1adbc

20 with warnings.catch_warnings(): 4 ctx1efgh

21 warnings.simplefilter("ignore", FastAPIDeprecationWarning) 4 ctx1efgh

22 app = FastAPI(default_response_class=ORJSONResponse) 4 ctx1efgh

23 

24 @app.get("/items") 4 ctx1efgh

25 def get_items() -> Item: 4 ctx1efgh

26 return Item(name="widget", price=9.99) 4 ctx1efgh

27 

28 return app 4 ctx1efgh

29 

30 

31def test_orjson_response_returns_correct_data(): 4 ctx1adbc

32 app = _make_orjson_app() 4 ctx1efgh

33 client = TestClient(app) 4 ctx1efgh

34 with warnings.catch_warnings(): 4 ctx1efgh

35 warnings.simplefilter("ignore", FastAPIDeprecationWarning) 4 ctx1efgh

36 response = client.get("/items") 4 ctx1efgh

37 assert response.status_code == 200 4 ctx1efgh

38 assert response.json() == {"name": "widget", "price": 9.99} 4 ctx1efgh

39 

40 

41def test_orjson_response_emits_deprecation_warning(): 4 ctx1adbc

42 with pytest.warns(FastAPIDeprecationWarning, match="ORJSONResponse is deprecated"): 4 ctx1mnop

43 ORJSONResponse(content={"hello": "world"}) 4 ctx1mnop

44 

45 

46# UJSON 

47 

48 

49def _make_ujson_app() -> FastAPI: 4 ctx1adbc

50 with warnings.catch_warnings(): 4 ctx1ijkl

51 warnings.simplefilter("ignore", FastAPIDeprecationWarning) 4 ctx1ijkl

52 app = FastAPI(default_response_class=UJSONResponse) 4 ctx1ijkl

53 

54 @app.get("/items") 4 ctx1ijkl

55 def get_items() -> Item: 4 ctx1ijkl

56 return Item(name="widget", price=9.99) 4 ctx1ijkl

57 

58 return app 4 ctx1ijkl

59 

60 

61def test_ujson_response_returns_correct_data(): 4 ctx1adbc

62 app = _make_ujson_app() 4 ctx1ijkl

63 client = TestClient(app) 4 ctx1ijkl

64 with warnings.catch_warnings(): 4 ctx1ijkl

65 warnings.simplefilter("ignore", FastAPIDeprecationWarning) 4 ctx1ijkl

66 response = client.get("/items") 4 ctx1ijkl

67 assert response.status_code == 200 4 ctx1ijkl

68 assert response.json() == {"name": "widget", "price": 9.99} 4 ctx1ijkl

69 

70 

71def test_ujson_response_emits_deprecation_warning(): 4 ctx1adbc

72 with pytest.warns(FastAPIDeprecationWarning, match="UJSONResponse is deprecated"): 4 ctx1qrst

73 UJSONResponse(content={"hello": "world"}) 4 ctx1qrst