Coverage for tests/test_datetime_custom_encoder.py: 100%

1from datetime import datetime, timezone 6 ctx1mnopqr

2 

3from fastapi import FastAPI 6 ctx1mnopqr

4from fastapi.testclient import TestClient 6 ctx1mnopqr

5from pydantic import BaseModel 6 ctx1mnopqr

6 

7from .utils import needs_pydanticv1, needs_pydanticv2 6 ctx1mnopqr

8 

9 

10@needs_pydanticv2 6 ctx1mnopqr

11def test_pydanticv2(): 6 ctx1mnopqr

12 from pydantic import field_serializer 6 ctx1abcdef

13 

14 class ModelWithDatetimeField(BaseModel): 6 ctx1abcdef

15 dt_field: datetime 6 ctx1abcdef

16 

17 @field_serializer("dt_field") 6 ctx1abcdef

18 def serialize_datetime(self, dt_field: datetime): 6 ctx1abcdef

19 return dt_field.replace(microsecond=0, tzinfo=timezone.utc).isoformat() 6 ctx1abcdef

20 

21 app = FastAPI() 6 ctx1abcdef

22 model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8)) 6 ctx1abcdef

23 

24 @app.get("/model", response_model=ModelWithDatetimeField) 6 ctx1abcdef

25 def get_model(): 6 ctx1abcdef

26 return model 6 ctx1abcdef

27 

28 client = TestClient(app) 6 ctx1abcdef

29 with client: 6 ctx1abcdef

30 response = client.get("/model") 6 ctx1abcdef

31 assert response.json() == {"dt_field": "2019-01-01T08:00:00+00:00"} 6 ctx1abcdef

32 

33 

34# TODO: remove when deprecating Pydantic v1 

35@needs_pydanticv1 6 ctx1mnopqr

36def test_pydanticv1(): 6 ctx1mnopqr

37 class ModelWithDatetimeField(BaseModel): 6 ctx1ghijkl

38 dt_field: datetime 6 ctx1ghijkl

39 

40 class Config: 6 ctx1ghijkl

41 json_encoders = { 6 ctx1ghijkl

42 datetime: lambda dt: dt.replace( 

43 microsecond=0, tzinfo=timezone.utc 

44 ).isoformat() 

45 } 

46 

47 app = FastAPI() 6 ctx1ghijkl

48 model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8)) 6 ctx1ghijkl

49 

50 @app.get("/model", response_model=ModelWithDatetimeField) 6 ctx1ghijkl

51 def get_model(): 6 ctx1ghijkl

52 return model 6 ctx1ghijkl

53 

54 client = TestClient(app) 6 ctx1ghijkl

55 with client: 6 ctx1ghijkl

56 response = client.get("/model") 6 ctx1ghijkl

57 assert response.json() == {"dt_field": "2019-01-01T08:00:00+00:00"} 6 ctx1ghijkl