Coverage for tests/test_datetime_custom_encoder.py: 100%

1from datetime import datetime, timezone 7 ctx1nopqrst

2 

3from fastapi import FastAPI 7 ctx1nopqrst

4from fastapi.testclient import TestClient 7 ctx1nopqrst

5from pydantic import BaseModel 7 ctx1nopqrst

6 

7from .utils import needs_pydanticv1, needs_pydanticv2 7 ctx1nopqrst

8 

9 

10@needs_pydanticv2 7 ctx1nopqrst

11def test_pydanticv2(): 7 ctx1nopqrst

12 from pydantic import field_serializer 7 ctx1abcdefg

13 

14 class ModelWithDatetimeField(BaseModel): 7 ctx1abcdefg

15 dt_field: datetime 7 ctx1abcdefg

16 

17 @field_serializer("dt_field") 7 ctx1abcdefg

18 def serialize_datetime(self, dt_field: datetime): 7 ctx1abcdefg

19 return dt_field.replace(microsecond=0, tzinfo=timezone.utc).isoformat() 7 ctx1abcdefg

20 

21 app = FastAPI() 7 ctx1abcdefg

22 model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8)) 7 ctx1abcdefg

23 

24 @app.get("/model", response_model=ModelWithDatetimeField) 7 ctx1abcdefg

25 def get_model(): 7 ctx1abcdefg

26 return model 7 ctx1abcdefg

27 

28 client = TestClient(app) 7 ctx1abcdefg

29 with client: 7 ctx1abcdefg

30 response = client.get("/model") 7 ctx1abcdefg

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

32 

33 

34# TODO: remove when deprecating Pydantic v1 

35@needs_pydanticv1 7 ctx1nopqrst

36def test_pydanticv1(): 7 ctx1nopqrst

37 class ModelWithDatetimeField(BaseModel): 6 ctx1hijklm

38 dt_field: datetime 6 ctx1hijklm

39 

40 class Config: 6 ctx1hijklm

41 json_encoders = { 6 ctx1hijklm

42 datetime: lambda dt: dt.replace( 

43 microsecond=0, tzinfo=timezone.utc 

44 ).isoformat() 

45 } 

46 

47 app = FastAPI() 6 ctx1hijklm

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

49 

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

51 def get_model(): 6 ctx1hijklm

52 return model 6 ctx1hijklm

53 

54 client = TestClient(app) 6 ctx1hijklm

55 with client: 6 ctx1hijklm

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

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