committed by
GitHub
2 changed files with 78 additions and 6 deletions
@ -0,0 +1,72 @@ |
|||
from typing import Any |
|||
|
|||
import pytest |
|||
from fastapi import FastAPI |
|||
from fastapi.testclient import TestClient |
|||
from inline_snapshot import snapshot |
|||
from pydantic import BaseModel, ConfigDict, Field |
|||
|
|||
from tests.utils import needs_pydanticv2 |
|||
|
|||
|
|||
@pytest.fixture(name="client") |
|||
def get_client(): |
|||
app = FastAPI() |
|||
|
|||
class ModelWithRef(BaseModel): |
|||
ref: str = Field(validation_alias="$ref", serialization_alias="$ref") |
|||
model_config = ConfigDict(validate_by_alias=True, serialize_by_alias=True) |
|||
|
|||
@app.get("/", response_model=ModelWithRef) |
|||
async def read_root() -> Any: |
|||
return {"$ref": "some-ref"} |
|||
|
|||
client = TestClient(app) |
|||
return client |
|||
|
|||
|
|||
@needs_pydanticv2 |
|||
def test_get(client: TestClient): |
|||
response = client.get("/") |
|||
assert response.json() == {"$ref": "some-ref"} |
|||
|
|||
|
|||
@needs_pydanticv2 |
|||
def test_openapi_schema(client: TestClient): |
|||
response = client.get("openapi.json") |
|||
assert response.json() == snapshot( |
|||
{ |
|||
"openapi": "3.1.0", |
|||
"info": {"title": "FastAPI", "version": "0.1.0"}, |
|||
"paths": { |
|||
"/": { |
|||
"get": { |
|||
"summary": "Read Root", |
|||
"operationId": "read_root__get", |
|||
"responses": { |
|||
"200": { |
|||
"description": "Successful Response", |
|||
"content": { |
|||
"application/json": { |
|||
"schema": { |
|||
"$ref": "#/components/schemas/ModelWithRef" |
|||
} |
|||
} |
|||
}, |
|||
} |
|||
}, |
|||
} |
|||
} |
|||
}, |
|||
"components": { |
|||
"schemas": { |
|||
"ModelWithRef": { |
|||
"properties": {"$ref": {"type": "string", "title": "$Ref"}}, |
|||
"type": "object", |
|||
"required": ["$ref"], |
|||
"title": "ModelWithRef", |
|||
} |
|||
} |
|||
}, |
|||
} |
|||
) |
|||
Loading…
Reference in new issue