Browse Source

Adding tests from similiar pr,

Co-authored-by: dlax <denis@laxalde.org>
pull/13639/head
Lukas Rajala 1 week ago
parent
commit
3cd4f27278
  1. 19
      tests/test_custom_schema_fields.py

19
tests/test_custom_schema_fields.py

@ -1,7 +1,13 @@
from typing import Optional
from fastapi import FastAPI
from fastapi._compat import PYDANTIC_V2
from fastapi.testclient import TestClient
from pydantic import BaseModel
from typing_extensions import Annotated
if PYDANTIC_V2:
from pydantic import WithJsonSchema
app = FastAPI()
@ -10,12 +16,17 @@ class Item(BaseModel):
name: str
if PYDANTIC_V2:
description: Annotated[
Optional[str], WithJsonSchema({"type": ["string", "null"]})
] = None
model_config = {
"json_schema_extra": {
"x-something-internal": {"level": 4},
}
}
else:
description: Optional[str] = None # type: ignore[no-redef]
class Config:
schema_extra = {
@ -42,7 +53,11 @@ item_schema = {
"name": {
"title": "Name",
"type": "string",
}
},
"description": {
"title": "Description",
"type": ["string", "null"] if PYDANTIC_V2 else "string",
},
},
}
@ -57,4 +72,4 @@ def test_response():
# For coverage
response = client.get("/foo")
assert response.status_code == 200, response.text
assert response.json() == {"name": "Foo item"}
assert response.json() == {"name": "Foo item", "description": None}

Loading…
Cancel
Save