Coverage for tests / test_skip_defaults.py: 100%

1from typing import Optional 4 ctx1abcd

2 

3from fastapi import FastAPI 4 ctx1abcd

4from fastapi.testclient import TestClient 4 ctx1abcd

5from pydantic import BaseModel 4 ctx1abcd

6 

7app = FastAPI() 4 ctx1abcd

8 

9 

10class SubModel(BaseModel): 4 ctx1abcd

11 a: Optional[str] = "foo" 4 ctx1abcd

12 

13 

14class Model(BaseModel): 4 ctx1abcd

15 x: Optional[int] = None 4 ctx1abcd

16 sub: SubModel 4 ctx1abcd

17 

18 

19class ModelSubclass(Model): 4 ctx1abcd

20 y: int 4 ctx1abcd

21 z: int = 0 4 ctx1abcd

22 w: Optional[int] = None 4 ctx1abcd

23 

24 

25class ModelDefaults(BaseModel): 4 ctx1abcd

26 w: Optional[str] = None 4 ctx1abcd

27 x: Optional[str] = None 4 ctx1abcd

28 y: str = "y" 4 ctx1abcd

29 z: str = "z" 4 ctx1abcd

30 

31 

32@app.get("/", response_model=Model, response_model_exclude_unset=True) 4 ctx1abcd

33def get_root() -> ModelSubclass: 4 ctx1abcd

34 return ModelSubclass(sub={}, y=1, z=0) 3 ctx1efg

35 

36 

37@app.get( 4 ctx1abcd

38 "/exclude_unset", response_model=ModelDefaults, response_model_exclude_unset=True 

39) 

40def get_exclude_unset() -> ModelDefaults: 4 ctx1abcd

41 return ModelDefaults(x=None, y="y") 3 ctx1hij

42 

43 

44@app.get( 4 ctx1abcd

45 "/exclude_defaults", 

46 response_model=ModelDefaults, 

47 response_model_exclude_defaults=True, 

48) 

49def get_exclude_defaults() -> ModelDefaults: 4 ctx1abcd

50 return ModelDefaults(x=None, y="y") 3 ctx1klm

51 

52 

53@app.get( 4 ctx1abcd

54 "/exclude_none", response_model=ModelDefaults, response_model_exclude_none=True 

55) 

56def get_exclude_none() -> ModelDefaults: 4 ctx1abcd

57 return ModelDefaults(x=None, y="y") 3 ctx1nop

58 

59 

60@app.get( 4 ctx1abcd

61 "/exclude_unset_none", 

62 response_model=ModelDefaults, 

63 response_model_exclude_unset=True, 

64 response_model_exclude_none=True, 

65) 

66def get_exclude_unset_none() -> ModelDefaults: 4 ctx1abcd

67 return ModelDefaults(x=None, y="y") 3 ctx1qrs

68 

69 

70client = TestClient(app) 4 ctx1abcd

71 

72 

73def test_return_defaults(): 4 ctx1abcd

74 response = client.get("/") 3 ctx1efg

75 assert response.json() == {"sub": {}} 3 ctx1efg

76 

77 

78def test_return_exclude_unset(): 4 ctx1abcd

79 response = client.get("/exclude_unset") 3 ctx1hij

80 assert response.json() == {"x": None, "y": "y"} 3 ctx1hij

81 

82 

83def test_return_exclude_defaults(): 4 ctx1abcd

84 response = client.get("/exclude_defaults") 3 ctx1klm

85 assert response.json() == {} 3 ctx1klm

86 

87 

88def test_return_exclude_none(): 4 ctx1abcd

89 response = client.get("/exclude_none") 3 ctx1nop

90 assert response.json() == {"y": "y", "z": "z"} 3 ctx1nop

91 

92 

93def test_return_exclude_unset_none(): 4 ctx1abcd

94 response = client.get("/exclude_unset_none") 3 ctx1qrs

95 assert response.json() == {"y": "y"} 3 ctx1qrs