Coverage for tests/test_response_model_data_filter.py: 100%

1from typing import List 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5from pydantic import BaseModel 5 ctx1abcde

6 

7app = FastAPI() 5 ctx1abcde

8 

9 

10class UserBase(BaseModel): 5 ctx1abcde

11 email: str 5 ctx1abcde

12 

13 

14class UserCreate(UserBase): 5 ctx1abcde

15 password: str 5 ctx1abcde

16 

17 

18class UserDB(UserBase): 5 ctx1abcde

19 hashed_password: str 5 ctx1abcde

20 

21 

22class PetDB(BaseModel): 5 ctx1abcde

23 name: str 5 ctx1abcde

24 owner: UserDB 5 ctx1abcde

25 

26 

27class PetOut(BaseModel): 5 ctx1abcde

28 name: str 5 ctx1abcde

29 owner: UserBase 5 ctx1abcde

30 

31 

32@app.post("/users/", response_model=UserBase) 5 ctx1abcde

33async def create_user(user: UserCreate): 5 ctx1abcde

34 return user 5 ctx1abcde

35 

36 

37@app.get("/pets/{pet_id}", response_model=PetOut) 5 ctx1abcde

38async def read_pet(pet_id: int): 5 ctx1abcde

39 user = UserDB( 5 ctx1abcde

40 email="johndoe@example.com", 

41 hashed_password="secrethashed", 

42 ) 

43 pet = PetDB(name="Nibbler", owner=user) 5 ctx1abcde

44 return pet 5 ctx1abcde

45 

46 

47@app.get("/pets/", response_model=List[PetOut]) 5 ctx1abcde

48async def read_pets(): 5 ctx1abcde

49 user = UserDB( 5 ctx1abcde

50 email="johndoe@example.com", 

51 hashed_password="secrethashed", 

52 ) 

53 pet1 = PetDB(name="Nibbler", owner=user) 5 ctx1abcde

54 pet2 = PetDB(name="Zoidberg", owner=user) 5 ctx1abcde

55 return [pet1, pet2] 5 ctx1abcde

56 

57 

58client = TestClient(app) 5 ctx1abcde

59 

60 

61def test_filter_top_level_model(): 5 ctx1abcde

62 response = client.post( 5 ctx1abcde

63 "/users", json={"email": "johndoe@example.com", "password": "secret"} 

64 ) 

65 assert response.json() == {"email": "johndoe@example.com"} 5 ctx1abcde

66 

67 

68def test_filter_second_level_model(): 5 ctx1abcde

69 response = client.get("/pets/1") 5 ctx1abcde

70 assert response.json() == { 5 ctx1abcde

71 "name": "Nibbler", 

72 "owner": {"email": "johndoe@example.com"}, 

73 } 

74 

75 

76def test_list_of_models(): 5 ctx1abcde

77 response = client.get("/pets/") 5 ctx1abcde

78 assert response.json() == [ 5 ctx1abcde

79 {"name": "Nibbler", "owner": {"email": "johndoe@example.com"}}, 

80 {"name": "Zoidberg", "owner": {"email": "johndoe@example.com"}}, 

81 ]