Coverage for tests/test_read_with_orm_mode.py: 100%

1from typing import Any 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4from fastapi.testclient import TestClient 5 ctx1abcde

5from pydantic import BaseModel, ConfigDict 5 ctx1abcde

6 

7from .utils import needs_pydanticv1, needs_pydanticv2 5 ctx1abcde

8 

9 

10@needs_pydanticv2 5 ctx1abcde

11def test_read_with_orm_mode() -> None: 5 ctx1abcde

12 class PersonBase(BaseModel): 5 ctx1abcde

13 name: str 5 ctx1abcde

14 lastname: str 5 ctx1abcde

15 

16 class Person(PersonBase): 5 ctx1abcde

17 @property 5 ctx1abcde

18 def full_name(self) -> str: 5 ctx1abcde

19 return f"{self.name} {self.lastname}" 5 ctx1abcde

20 

21 model_config = ConfigDict(from_attributes=True) 5 ctx1abcde

22 

23 class PersonCreate(PersonBase): 5 ctx1abcde

24 pass 5 ctx1abcde

25 

26 class PersonRead(PersonBase): 5 ctx1abcde

27 full_name: str 5 ctx1abcde

28 

29 model_config = {"from_attributes": True} 5 ctx1abcde

30 

31 app = FastAPI() 5 ctx1abcde

32 

33 @app.post("/people/", response_model=PersonRead) 5 ctx1abcde

34 def create_person(person: PersonCreate) -> Any: 5 ctx1abcde

35 db_person = Person.model_validate(person) 5 ctx1abcde

36 return db_person 5 ctx1abcde

37 

38 client = TestClient(app) 5 ctx1abcde

39 

40 person_data = {"name": "Dive", "lastname": "Wilson"} 5 ctx1abcde

41 response = client.post("/people/", json=person_data) 5 ctx1abcde

42 data = response.json() 5 ctx1abcde

43 assert response.status_code == 200, response.text 5 ctx1abcde

44 assert data["name"] == person_data["name"] 5 ctx1abcde

45 assert data["lastname"] == person_data["lastname"] 5 ctx1abcde

46 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 5 ctx1abcde

47 

48 

49@needs_pydanticv1 5 ctx1abcde

50def test_read_with_orm_mode_pv1() -> None: 5 ctx1abcde

51 class PersonBase(BaseModel): 5 ctx1abcde

52 name: str 5 ctx1abcde

53 lastname: str 5 ctx1abcde

54 

55 class Person(PersonBase): 5 ctx1abcde

56 @property 5 ctx1abcde

57 def full_name(self) -> str: 5 ctx1abcde

58 return f"{self.name} {self.lastname}" 5 ctx1abcde

59 

60 class Config: 5 ctx1abcde

61 orm_mode = True 5 ctx1abcde

62 read_with_orm_mode = True 5 ctx1abcde

63 

64 class PersonCreate(PersonBase): 5 ctx1abcde

65 pass 5 ctx1abcde

66 

67 class PersonRead(PersonBase): 5 ctx1abcde

68 full_name: str 5 ctx1abcde

69 

70 class Config: 5 ctx1abcde

71 orm_mode = True 5 ctx1abcde

72 

73 app = FastAPI() 5 ctx1abcde

74 

75 @app.post("/people/", response_model=PersonRead) 5 ctx1abcde

76 def create_person(person: PersonCreate) -> Any: 5 ctx1abcde

77 db_person = Person.from_orm(person) 5 ctx1abcde

78 return db_person 5 ctx1abcde

79 

80 client = TestClient(app) 5 ctx1abcde

81 

82 person_data = {"name": "Dive", "lastname": "Wilson"} 5 ctx1abcde

83 response = client.post("/people/", json=person_data) 5 ctx1abcde

84 data = response.json() 5 ctx1abcde

85 assert response.status_code == 200, response.text 5 ctx1abcde

86 assert data["name"] == person_data["name"] 5 ctx1abcde

87 assert data["lastname"] == person_data["lastname"] 5 ctx1abcde

88 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 5 ctx1abcde