58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
|
from datetime import datetime, timezone
|
||
|
|
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from .utils import needs_pydanticv1, needs_pydanticv2
|
||
|
|
|
||
|
|
|
||
|
|
@needs_pydanticv2
|
||
|
|
def test_pydanticv2():
|
||
|
|
from pydantic import field_serializer
|
||
|
|
|
||
|
|
class ModelWithDatetimeField(BaseModel):
|
||
|
|
dt_field: datetime
|
||
|
|
|
||
|
|
@field_serializer("dt_field")
|
||
|
|
def serialize_datetime(self, dt_field: datetime):
|
||
|
|
return dt_field.replace(microsecond=0, tzinfo=timezone.utc).isoformat()
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8))
|
||
|
|
|
||
|
|
@app.get("/model", response_model=ModelWithDatetimeField)
|
||
|
|
def get_model():
|
||
|
|
return model
|
||
|
|
|
||
|
|
client = TestClient(app)
|
||
|
|
with client:
|
||
|
|
response = client.get("/model")
|
||
|
|
assert response.json() == {"dt_field": "2019-01-01T08:00:00+00:00"}
|
||
|
|
|
||
|
|
|
||
|
|
# TODO: remove when deprecating Pydantic v1
|
||
|
|
@needs_pydanticv1
|
||
|
|
def test_pydanticv1():
|
||
|
|
class ModelWithDatetimeField(BaseModel):
|
||
|
|
dt_field: datetime
|
||
|
|
|
||
|
|
class Config:
|
||
|
|
json_encoders = {
|
||
|
|
datetime: lambda dt: dt.replace(
|
||
|
|
microsecond=0, tzinfo=timezone.utc
|
||
|
|
).isoformat()
|
||
|
|
}
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
model = ModelWithDatetimeField(dt_field=datetime(2019, 1, 1, 8))
|
||
|
|
|
||
|
|
@app.get("/model", response_model=ModelWithDatetimeField)
|
||
|
|
def get_model():
|
||
|
|
return model
|
||
|
|
|
||
|
|
client = TestClient(app)
|
||
|
|
with client:
|
||
|
|
response = client.get("/model")
|
||
|
|
assert response.json() == {"dt_field": "2019-01-01T08:00:00+00:00"}
|