Coverage for tests/test_read_with_orm_mode.py: 100%

1from typing import Any 6 ctx1mnopqr

2 

3from fastapi import FastAPI 6 ctx1mnopqr

4from fastapi.testclient import TestClient 6 ctx1mnopqr

5from pydantic import BaseModel, ConfigDict 6 ctx1mnopqr

6 

7from .utils import needs_pydanticv1, needs_pydanticv2 6 ctx1mnopqr

8 

9 

10@needs_pydanticv2 6 ctx1mnopqr

11def test_read_with_orm_mode() -> None: 6 ctx1mnopqr

12 class PersonBase(BaseModel): 6 ctx1ghijkl

13 name: str 6 ctx1ghijkl

14 lastname: str 6 ctx1ghijkl

15 

16 class Person(PersonBase): 6 ctx1ghijkl

17 @property 6 ctx1ghijkl

18 def full_name(self) -> str: 6 ctx1ghijkl

19 return f"{self.name} {self.lastname}" 6 ctx1ghijkl

20 

21 model_config = ConfigDict(from_attributes=True) 6 ctx1ghijkl

22 

23 class PersonCreate(PersonBase): 6 ctx1ghijkl

24 pass 6 ctx1ghijkl

25 

26 class PersonRead(PersonBase): 6 ctx1ghijkl

27 full_name: str 6 ctx1ghijkl

28 

29 model_config = {"from_attributes": True} 6 ctx1ghijkl

30 

31 app = FastAPI() 6 ctx1ghijkl

32 

33 @app.post("/people/", response_model=PersonRead) 6 ctx1ghijkl

34 def create_person(person: PersonCreate) -> Any: 6 ctx1ghijkl

35 db_person = Person.model_validate(person) 6 ctx1ghijkl

36 return db_person 6 ctx1ghijkl

37 

38 client = TestClient(app) 6 ctx1ghijkl

39 

40 person_data = {"name": "Dive", "lastname": "Wilson"} 6 ctx1ghijkl

41 response = client.post("/people/", json=person_data) 6 ctx1ghijkl

42 data = response.json() 6 ctx1ghijkl

43 assert response.status_code == 200, response.text 6 ctx1ghijkl

44 assert data["name"] == person_data["name"] 6 ctx1ghijkl

45 assert data["lastname"] == person_data["lastname"] 6 ctx1ghijkl

46 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 6 ctx1ghijkl

47 

48 

49@needs_pydanticv1 6 ctx1mnopqr

50def test_read_with_orm_mode_pv1() -> None: 6 ctx1mnopqr

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