mirror of
https://github.com/fastapi/fastapi.git
synced 2025-12-29 01:00:51 -05:00
29 lines
817 B
Python
29 lines
817 B
Python
from datetime import datetime, timezone
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
|
|
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"}
|