mirror of
https://github.com/fastapi/fastapi.git
synced 2026-01-22 12:58:11 -05:00
3397d4d69a
* Implemented response_model_exclude_defaults and response_model_exclude_none to be compatible pydantic options. * 🚚 Rename and invert include_none to exclude_none to keep in sync with Pydantic Co-authored-by: Lukas Voegtle <lukas.voegtle@sick.de> Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
96 lines
1.9 KiB
Python
96 lines
1.9 KiB
Python
from typing import Optional
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class SubModel(BaseModel):
|
|
a: Optional[str] = "foo"
|
|
|
|
|
|
class Model(BaseModel):
|
|
x: Optional[int]
|
|
sub: SubModel
|
|
|
|
|
|
class ModelSubclass(Model):
|
|
y: int
|
|
z: int = 0
|
|
w: int = None
|
|
|
|
|
|
class ModelDefaults(BaseModel):
|
|
w: Optional[str] = None
|
|
x: Optional[str] = None
|
|
y: str = "y"
|
|
z: str = "z"
|
|
|
|
|
|
@app.get("/", response_model=Model, response_model_exclude_unset=True)
|
|
def get() -> ModelSubclass:
|
|
return ModelSubclass(sub={}, y=1, z=0)
|
|
|
|
|
|
@app.get(
|
|
"/exclude_unset", response_model=ModelDefaults, response_model_exclude_unset=True
|
|
)
|
|
def get() -> ModelDefaults:
|
|
return ModelDefaults(x=None, y="y")
|
|
|
|
|
|
@app.get(
|
|
"/exclude_defaults",
|
|
response_model=ModelDefaults,
|
|
response_model_exclude_defaults=True,
|
|
)
|
|
def get() -> ModelDefaults:
|
|
return ModelDefaults(x=None, y="y")
|
|
|
|
|
|
@app.get(
|
|
"/exclude_none", response_model=ModelDefaults, response_model_exclude_none=True
|
|
)
|
|
def get() -> ModelDefaults:
|
|
return ModelDefaults(x=None, y="y")
|
|
|
|
|
|
@app.get(
|
|
"/exclude_unset_none",
|
|
response_model=ModelDefaults,
|
|
response_model_exclude_unset=True,
|
|
response_model_exclude_none=True,
|
|
)
|
|
def get() -> ModelDefaults:
|
|
return ModelDefaults(x=None, y="y")
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
def test_return_defaults():
|
|
response = client.get("/")
|
|
assert response.json() == {"sub": {}}
|
|
|
|
|
|
def test_return_exclude_unset():
|
|
response = client.get("/exclude_unset")
|
|
assert response.json() == {"x": None, "y": "y"}
|
|
|
|
|
|
def test_return_exclude_defaults():
|
|
response = client.get("/exclude_defaults")
|
|
assert response.json() == {}
|
|
|
|
|
|
def test_return_exclude_none():
|
|
response = client.get("/exclude_none")
|
|
assert response.json() == {"y": "y", "z": "z"}
|
|
|
|
|
|
def test_return_exclude_unset_none():
|
|
response = client.get("/exclude_unset_none")
|
|
assert response.json() == {"y": "y"}
|