mirror of https://github.com/tiangolo/fastapi.git
36 lines
726 B
Python
36 lines
726 B
Python
import json
|
|
from datetime import datetime, timezone
|
|
|
|
from fastapi import FastAPI
|
|
from pydantic import BaseModel
|
|
from starlette.testclient import TestClient
|
|
|
|
|
|
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.utcnow())
|
|
|
|
|
|
@app.get("/model", response_model=ModelWithDatetimeField)
|
|
def get_model():
|
|
return model
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
def test_dt():
|
|
with client:
|
|
response = client.get("/model")
|
|
assert json.loads(model.json()) == response.json()
|