Browse Source
Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Co-authored-by: Motov Yurii <[email protected]>pull/14816/head
committed by
GitHub
2 changed files with 125 additions and 1 deletions
@ -0,0 +1,123 @@ |
|||||
|
""" |
||||
|
Regression test: Ensure app-level responses with Union models and content/examples |
||||
|
don't accumulate duplicate $ref entries in anyOf arrays. |
||||
|
See https://github.com/fastapi/fastapi/pull/14463 |
||||
|
""" |
||||
|
|
||||
|
from typing import Union |
||||
|
|
||||
|
from fastapi import FastAPI |
||||
|
from fastapi.testclient import TestClient |
||||
|
from pydantic import BaseModel |
||||
|
|
||||
|
|
||||
|
class ModelA(BaseModel): |
||||
|
a: str |
||||
|
|
||||
|
|
||||
|
class ModelB(BaseModel): |
||||
|
b: str |
||||
|
|
||||
|
|
||||
|
app = FastAPI( |
||||
|
responses={ |
||||
|
500: { |
||||
|
"model": Union[ModelA, ModelB], |
||||
|
"content": {"application/json": {"examples": {"Case A": {"value": "a"}}}}, |
||||
|
} |
||||
|
} |
||||
|
) |
||||
|
|
||||
|
|
||||
|
@app.get("/route1") |
||||
|
async def route1(): |
||||
|
pass # pragma: no cover |
||||
|
|
||||
|
|
||||
|
@app.get("/route2") |
||||
|
async def route2(): |
||||
|
pass # pragma: no cover |
||||
|
|
||||
|
|
||||
|
client = TestClient(app) |
||||
|
|
||||
|
|
||||
|
def test_openapi_schema(): |
||||
|
response = client.get("/openapi.json") |
||||
|
assert response.status_code == 200, response.text |
||||
|
assert response.json() == { |
||||
|
"openapi": "3.1.0", |
||||
|
"info": {"title": "FastAPI", "version": "0.1.0"}, |
||||
|
"paths": { |
||||
|
"/route1": { |
||||
|
"get": { |
||||
|
"summary": "Route1", |
||||
|
"operationId": "route1_route1_get", |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": {"application/json": {"schema": {}}}, |
||||
|
}, |
||||
|
"500": { |
||||
|
"description": "Internal Server Error", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"anyOf": [ |
||||
|
{"$ref": "#/components/schemas/ModelA"}, |
||||
|
{"$ref": "#/components/schemas/ModelB"}, |
||||
|
], |
||||
|
"title": "Response 500 Route1 Route1 Get", |
||||
|
}, |
||||
|
"examples": {"Case A": {"value": "a"}}, |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
"/route2": { |
||||
|
"get": { |
||||
|
"summary": "Route2", |
||||
|
"operationId": "route2_route2_get", |
||||
|
"responses": { |
||||
|
"200": { |
||||
|
"description": "Successful Response", |
||||
|
"content": {"application/json": {"schema": {}}}, |
||||
|
}, |
||||
|
"500": { |
||||
|
"description": "Internal Server Error", |
||||
|
"content": { |
||||
|
"application/json": { |
||||
|
"schema": { |
||||
|
"anyOf": [ |
||||
|
{"$ref": "#/components/schemas/ModelA"}, |
||||
|
{"$ref": "#/components/schemas/ModelB"}, |
||||
|
], |
||||
|
"title": "Response 500 Route2 Route2 Get", |
||||
|
}, |
||||
|
"examples": {"Case A": {"value": "a"}}, |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
}, |
||||
|
"components": { |
||||
|
"schemas": { |
||||
|
"ModelA": { |
||||
|
"properties": {"a": {"type": "string", "title": "A"}}, |
||||
|
"type": "object", |
||||
|
"required": ["a"], |
||||
|
"title": "ModelA", |
||||
|
}, |
||||
|
"ModelB": { |
||||
|
"properties": {"b": {"type": "string", "title": "B"}}, |
||||
|
"type": "object", |
||||
|
"required": ["b"], |
||||
|
"title": "ModelB", |
||||
|
}, |
||||
|
} |
||||
|
}, |
||||
|
} |
||||
Loading…
Reference in new issue