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
61 lines
1.2 KiB
Python
61 lines
1.2 KiB
Python
from fastapi import FastAPI
|
|
from fastapi._compat import PYDANTIC_V2
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class Item(BaseModel):
|
|
name: str
|
|
|
|
if PYDANTIC_V2:
|
|
model_config = {
|
|
"json_schema_extra": {
|
|
"x-something-internal": {"level": 4},
|
|
}
|
|
}
|
|
else:
|
|
|
|
class Config:
|
|
schema_extra = {
|
|
"x-something-internal": {"level": 4},
|
|
}
|
|
|
|
|
|
@app.get("/foo", response_model=Item)
|
|
def foo():
|
|
return {"name": "Foo item"}
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
item_schema = {
|
|
"title": "Item",
|
|
"required": ["name"],
|
|
"type": "object",
|
|
"x-something-internal": {
|
|
"level": 4,
|
|
},
|
|
"properties": {
|
|
"name": {
|
|
"title": "Name",
|
|
"type": "string",
|
|
}
|
|
},
|
|
}
|
|
|
|
|
|
def test_custom_response_schema():
|
|
response = client.get("/openapi.json")
|
|
assert response.status_code == 200, response.text
|
|
assert response.json()["components"]["schemas"]["Item"] == item_schema
|
|
|
|
|
|
def test_response():
|
|
# For coverage
|
|
response = client.get("/foo")
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == {"name": "Foo item"}
|