Coverage for tests / test_read_with_orm_mode.py: 100%

1from typing import Any 4 ctx1efgh

2 

3from fastapi import FastAPI 4 ctx1efgh

4from fastapi.testclient import TestClient 4 ctx1efgh

5from pydantic import BaseModel, ConfigDict 4 ctx1efgh

6 

7 

8def test_read_with_orm_mode() -> None: 4 ctx1efgh

9 class PersonBase(BaseModel): 4 ctx1adbc

10 name: str 3 ctx1abc

11 lastname: str 3 ctx1abc

12 

13 class Person(PersonBase): 4 ctx1adbc

14 @property 4 ctx1adbc

15 def full_name(self) -> str: 4 ctx1adbc

16 return f"{self.name} {self.lastname}" 4 ctx1adbc

17 

18 model_config = ConfigDict(from_attributes=True) 4 ctx1adbc

19 

20 class PersonCreate(PersonBase): 4 ctx1adbc

21 pass 4 ctx1adbc

22 

23 class PersonRead(PersonBase): 4 ctx1adbc

24 full_name: str 3 ctx1abc

25 

26 model_config = {"from_attributes": True} 4 ctx1adbc

27 

28 app = FastAPI() 4 ctx1adbc

29 

30 @app.post("/people/", response_model=PersonRead) 4 ctx1adbc

31 def create_person(person: PersonCreate) -> Any: 4 ctx1adbc

32 db_person = Person.model_validate(person) 4 ctx1adbc

33 return db_person 4 ctx1adbc

34 

35 client = TestClient(app) 4 ctx1adbc

36 

37 person_data = {"name": "Dive", "lastname": "Wilson"} 4 ctx1adbc

38 response = client.post("/people/", json=person_data) 4 ctx1adbc

39 data = response.json() 4 ctx1adbc

40 assert response.status_code == 200, response.text 4 ctx1adbc

41 assert data["name"] == person_data["name"] 4 ctx1adbc

42 assert data["lastname"] == person_data["lastname"] 4 ctx1adbc

43 assert data["full_name"] == person_data["name"] + " " + person_data["lastname"] 4 ctx1adbc