Coverage for tests / test_datetime_custom_encoder.py: 100%

1from datetime import datetime, timezone 4 ctx1efgh

2 

3from fastapi import FastAPI 4 ctx1efgh

4from fastapi.testclient import TestClient 4 ctx1efgh

5from pydantic import BaseModel 4 ctx1efgh

6 

7 

8def test_pydanticv2(): 4 ctx1efgh

9 from pydantic import field_serializer 4 ctx1adbc

10 

11 class ModelWithDatetimeField(BaseModel): 4 ctx1adbc

12 dt_field: datetime 3 ctx1abc

13 

14 @field_serializer("dt_field") 4 ctx1adbc

15 def serialize_datetime(self, dt_field: datetime): 4 ctx1adbc

16 return dt_field.replace(microsecond=0, tzinfo=timezone.utc).isoformat() 4 ctx1adbc

17 

18 app = FastAPI() 4 ctx1adbc

19 model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8)) 4 ctx1adbc

20 

21 @app.get("/model", response_model=ModelWithDatetimeField) 4 ctx1adbc

22 def get_model(): 4 ctx1adbc

23 return model 4 ctx1adbc

24 

25 client = TestClient(app) 4 ctx1adbc

26 with client: 4 ctx1adbc

27 response = client.get("/model") 4 ctx1adbc

28 assert response.json() == {"dt_field": "2019-01-01T08:00:00+00:00"} 4 ctx1adbc