Coverage for tests / test_read_with_orm_mode.py: 100%

1from typing import Any 4 ctx1defg

2 

3from fastapi import FastAPI 4 ctx1defg

4from fastapi.testclient import TestClient 4 ctx1defg

5from pydantic import BaseModel, ConfigDict 4 ctx1defg

6 

7 

8def test_read_with_orm_mode() -> None: 4 ctx1defg

9 class PersonBase(BaseModel): 3 ctx1abc

10 name: str 3 ctx1abc

11 lastname: str 3 ctx1abc

12 

13 class Person(PersonBase): 3 ctx1abc

14 @property 3 ctx1abc

15 def full_name(self) -> str: 3 ctx1abc

16 return f"{self.name} {self.lastname}" 3 ctx1abc

17 

18 model_config = ConfigDict(from_attributes=True) 3 ctx1abc

19 

20 class PersonCreate(PersonBase): 3 ctx1abc

21 pass 3 ctx1abc

22 

23 class PersonRead(PersonBase): 3 ctx1abc

24 full_name: str 3 ctx1abc

25 

26 model_config = {"from_attributes": True} 3 ctx1abc

27 

28 app = FastAPI() 3 ctx1abc

29 

30 @app.post("/people/", response_model=PersonRead) 3 ctx1abc

31 def create_person(person: PersonCreate) -> Any: 3 ctx1abc

32 db_person = Person.model_validate(person) 3 ctx1abc

33 return db_person 3 ctx1abc

34 

35 client = TestClient(app) 3 ctx1abc

36 

37 person_data = {"name": "Dive", "lastname": "Wilson"} 3 ctx1abc

38 response = client.post("/people/", json=person_data) 3 ctx1abc

39 data = response.json() 3 ctx1abc

40 assert response.status_code == 200, response.text 3 ctx1abc

41 assert data["name"] == person_data["name"] 3 ctx1abc

42 assert data["lastname"] == person_data["lastname"] 3 ctx1abc

43 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 3 ctx1abc