lqs
d20566fe07
Some checks failed
Test / lint (push) Has been cancelled
Test / test (pydantic-v1, 3.10) (push) Has been cancelled
Test / test (pydantic-v1, 3.11) (push) Has been cancelled
Test / test (pydantic-v1, 3.12) (push) Has been cancelled
Test / test (pydantic-v1, 3.8) (push) Has been cancelled
Test / test (pydantic-v1, 3.9) (push) Has been cancelled
Test / test (pydantic-v2, 3.10) (push) Has been cancelled
Test / test (pydantic-v2, 3.11) (push) Has been cancelled
Test / test (pydantic-v2, 3.12) (push) Has been cancelled
Test / test (pydantic-v2, 3.8) (push) Has been cancelled
Test / test (pydantic-v2, 3.9) (push) Has been cancelled
Test / coverage-combine (push) Has been cancelled
Test / check (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Label Approved / label-approved (push) Has been cancelled
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"}
|