Coverage for tests / test_datetime_custom_encoder.py: 100%

1from datetime import datetime, timezone 4 ctx1defg

2 

3from fastapi import FastAPI 4 ctx1defg

4from fastapi.testclient import TestClient 4 ctx1defg

5from pydantic import BaseModel 4 ctx1defg

6 

7 

8def test_pydanticv2(): 4 ctx1defg

9 from pydantic import field_serializer 3 ctx1abc

10 

11 class ModelWithDatetimeField(BaseModel): 3 ctx1abc

12 dt_field: datetime 3 ctx1abc

13 

14 @field_serializer("dt_field") 3 ctx1abc

15 def serialize_datetime(self, dt_field: datetime): 3 ctx1abc

16 return dt_field.replace(microsecond=0, tzinfo=timezone.utc).isoformat() 3 ctx1abc

17 

18 app = FastAPI() 3 ctx1abc

19 model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8)) 3 ctx1abc

20 

21 @app.get("/model", response_model=ModelWithDatetimeField) 3 ctx1abc

22 def get_model(): 3 ctx1abc

23 return model 3 ctx1abc

24 

25 client = TestClient(app) 3 ctx1abc

26 with client: 3 ctx1abc

27 response = client.get("/model") 3 ctx1abc

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