Coverage for tests/test_read_with_orm_mode.py: 100%

1from typing import Any 7 ctx1nopqrst

2 

3from fastapi import FastAPI 7 ctx1nopqrst

4from fastapi.testclient import TestClient 7 ctx1nopqrst

5from pydantic import BaseModel, ConfigDict 7 ctx1nopqrst

6 

7from .utils import needs_pydanticv1, needs_pydanticv2 7 ctx1nopqrst

8 

9 

10@needs_pydanticv2 7 ctx1nopqrst

11def test_read_with_orm_mode() -> None: 7 ctx1nopqrst

12 class PersonBase(BaseModel): 7 ctx1ghijklm

13 name: str 7 ctx1ghijklm

14 lastname: str 7 ctx1ghijklm

15 

16 class Person(PersonBase): 7 ctx1ghijklm

17 @property 7 ctx1ghijklm

18 def full_name(self) -> str: 7 ctx1ghijklm

19 return f"{self.name} {self.lastname}" 7 ctx1ghijklm

20 

21 model_config = ConfigDict(from_attributes=True) 7 ctx1ghijklm

22 

23 class PersonCreate(PersonBase): 7 ctx1ghijklm

24 pass 7 ctx1ghijklm

25 

26 class PersonRead(PersonBase): 7 ctx1ghijklm

27 full_name: str 7 ctx1ghijklm

28 

29 model_config = {"from_attributes": True} 7 ctx1ghijklm

30 

31 app = FastAPI() 7 ctx1ghijklm

32 

33 @app.post("/people/", response_model=PersonRead) 7 ctx1ghijklm

34 def create_person(person: PersonCreate) -> Any: 7 ctx1ghijklm

35 db_person = Person.model_validate(person) 7 ctx1ghijklm

36 return db_person 7 ctx1ghijklm

37 

38 client = TestClient(app) 7 ctx1ghijklm

39 

40 person_data = {"name": "Dive", "lastname": "Wilson"} 7 ctx1ghijklm

41 response = client.post("/people/", json=person_data) 7 ctx1ghijklm

42 data = response.json() 7 ctx1ghijklm

43 assert response.status_code == 200, response.text 7 ctx1ghijklm

44 assert data["name"] == person_data["name"] 7 ctx1ghijklm

45 assert data["lastname"] == person_data["lastname"] 7 ctx1ghijklm

46 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 7 ctx1ghijklm

47 

48 

49@needs_pydanticv1 7 ctx1nopqrst

50def test_read_with_orm_mode_pv1() -> None: 7 ctx1nopqrst

51 class PersonBase(BaseModel): 6 ctx1abcdef

52 name: str 6 ctx1abcdef

53 lastname: str 6 ctx1abcdef

54 

55 class Person(PersonBase): 6 ctx1abcdef

56 @property 6 ctx1abcdef

57 def full_name(self) -> str: 6 ctx1abcdef

58 return f"{self.name} {self.lastname}" 6 ctx1abcdef

59 

60 class Config: 6 ctx1abcdef

61 orm_mode = True 6 ctx1abcdef

62 read_with_orm_mode = True 6 ctx1abcdef

63 

64 class PersonCreate(PersonBase): 6 ctx1abcdef

65 pass 6 ctx1abcdef

66 

67 class PersonRead(PersonBase): 6 ctx1abcdef

68 full_name: str 6 ctx1abcdef

69 

70 class Config: 6 ctx1abcdef

71 orm_mode = True 6 ctx1abcdef

72 

73 app = FastAPI() 6 ctx1abcdef

74 

75 @app.post("/people/", response_model=PersonRead) 6 ctx1abcdef

76 def create_person(person: PersonCreate) -> Any: 6 ctx1abcdef

77 db_person = Person.from_orm(person) 6 ctx1abcdef

78 return db_person 6 ctx1abcdef

79 

80 client = TestClient(app) 6 ctx1abcdef

81 

82 person_data = {"name": "Dive", "lastname": "Wilson"} 6 ctx1abcdef

83 response = client.post("/people/", json=person_data) 6 ctx1abcdef

84 data = response.json() 6 ctx1abcdef

85 assert response.status_code == 200, response.text 6 ctx1abcdef

86 assert data["name"] == person_data["name"] 6 ctx1abcdef

87 assert data["lastname"] == person_data["lastname"] 6 ctx1abcdef

88 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 6 ctx1abcdef