98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
|
# mypy: ignore-errors
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from typing import Any, Optional
|
||
|
|
|
||
|
|
from openapi_spec_validator import validate
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from openapi_pydantic.compat import PYDANTIC_V2
|
||
|
|
from openapi_pydantic.v3.v3_0 import (
|
||
|
|
Components,
|
||
|
|
DataType,
|
||
|
|
Example,
|
||
|
|
Header,
|
||
|
|
Info,
|
||
|
|
MediaType,
|
||
|
|
OpenAPI,
|
||
|
|
Operation,
|
||
|
|
PathItem,
|
||
|
|
RequestBody,
|
||
|
|
Response,
|
||
|
|
Schema,
|
||
|
|
)
|
||
|
|
from openapi_pydantic.v3.v3_0.util import (
|
||
|
|
PydanticSchema,
|
||
|
|
construct_open_api_with_schema_class,
|
||
|
|
)
|
||
|
|
|
||
|
|
if sys.version_info < (3, 9):
|
||
|
|
from typing_extensions import Literal
|
||
|
|
else:
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
|
||
|
|
def test_basic_schema() -> None:
|
||
|
|
class SampleModel(BaseModel):
|
||
|
|
required: bool
|
||
|
|
optional: Optional[bool] = None
|
||
|
|
one_literal_choice: Literal["only_choice"]
|
||
|
|
multiple_literal_choices: Literal["choice1", "choice2"]
|
||
|
|
|
||
|
|
part_api = construct_sample_api(SampleModel)
|
||
|
|
|
||
|
|
api = construct_open_api_with_schema_class(part_api)
|
||
|
|
assert api.components is not None
|
||
|
|
assert api.components.schemas is not None
|
||
|
|
|
||
|
|
if PYDANTIC_V2:
|
||
|
|
json_api: Any = api.model_dump(mode="json", by_alias=True, exclude_none=True)
|
||
|
|
else:
|
||
|
|
json_api: Any = api.dict(by_alias=True, exclude_none=True)
|
||
|
|
validate(json_api)
|
||
|
|
|
||
|
|
|
||
|
|
def construct_sample_api(SampleModel) -> OpenAPI:
|
||
|
|
class SampleRequest(SampleModel):
|
||
|
|
model_config = {"json_schema_mode": "validation"}
|
||
|
|
|
||
|
|
class SampleResponse(SampleModel):
|
||
|
|
model_config = {"json_schema_mode": "serialization"}
|
||
|
|
|
||
|
|
return OpenAPI(
|
||
|
|
info=Info(
|
||
|
|
title="Sample API",
|
||
|
|
version="v0.0.1",
|
||
|
|
),
|
||
|
|
paths={
|
||
|
|
"/callme": PathItem(
|
||
|
|
post=Operation(
|
||
|
|
requestBody=RequestBody(
|
||
|
|
content={
|
||
|
|
"application/json": MediaType(
|
||
|
|
schema=PydanticSchema(schema_class=SampleRequest)
|
||
|
|
)
|
||
|
|
}
|
||
|
|
),
|
||
|
|
responses={
|
||
|
|
"200": Response(
|
||
|
|
description="resp",
|
||
|
|
headers={
|
||
|
|
"WWW-Authenticate": Header(
|
||
|
|
description="Indicate how to authenticate",
|
||
|
|
schema=Schema(type=DataType.STRING),
|
||
|
|
)
|
||
|
|
},
|
||
|
|
content={
|
||
|
|
"application/json": MediaType(
|
||
|
|
schema=PydanticSchema(schema_class=SampleResponse)
|
||
|
|
)
|
||
|
|
},
|
||
|
|
)
|
||
|
|
},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
},
|
||
|
|
components=Components(examples={"thing-example": Example(value="thing1")}),
|
||
|
|
)
|