Coverage for tests/test_response_model_data_filter_no_inheritance.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 UserCreate(BaseModel): 5 ctx1abcde

11 email: str 5 ctx1abcde

12 password: str 5 ctx1abcde

13 

14 

15class UserDB(BaseModel): 5 ctx1abcde

16 email: str 5 ctx1abcde

17 hashed_password: str 5 ctx1abcde

18 

19 

20class User(BaseModel): 5 ctx1abcde

21 email: str 5 ctx1abcde

22 

23 

24class PetDB(BaseModel): 5 ctx1abcde

25 name: str 5 ctx1abcde

26 owner: UserDB 5 ctx1abcde

27 

28 

29class PetOut(BaseModel): 5 ctx1abcde

30 name: str 5 ctx1abcde

31 owner: User 5 ctx1abcde

32 

33 

34@app.post("/users/", response_model=User) 5 ctx1abcde

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

36 return user 5 ctx1abcde

37 

38 

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

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

41 user = UserDB( 5 ctx1abcde

42 email="johndoe@example.com", 

43 hashed_password="secrethashed", 

44 ) 

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

46 return pet 5 ctx1abcde

47 

48 

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

50async def read_pets(): 5 ctx1abcde

51 user = UserDB( 5 ctx1abcde

52 email="johndoe@example.com", 

53 hashed_password="secrethashed", 

54 ) 

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

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

57 return [pet1, pet2] 5 ctx1abcde

58 

59 

60client = TestClient(app) 5 ctx1abcde

61 

62 

63def test_filter_top_level_model(): 5 ctx1abcde

64 response = client.post( 5 ctx1abcde

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

66 ) 

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

68 

69 

70def test_filter_second_level_model(): 5 ctx1abcde

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

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

73 "name": "Nibbler", 

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

75 } 

76 

77 

78def test_list_of_models(): 5 ctx1abcde

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

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

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

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

83 ]