happyz
/
fastapi
mirror of https://github.com/tiangolo/fastapi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

🎨 [pre-commit.ci] Auto format from pre-commit.com hooks

This commit is contained in:
pre-commit-ci[bot] 2024-03-01 22:41:12 +00:00 committed by Vinícius Aguiar
parent 932659f49b
commit 0cb111e4b8
2 changed files with 43 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
fastapi/openapi/utils.py
View File

@ -347,12 +347,14 @@ def get_openapi_path(
response_schema = {}
route_responses = dict((str(k), v) for k, v in route.responses.items())
if status_code not in route_responses or not route_responses.get(status_code).get('superimpose'):
if status_code not in route_responses or not route_responses.get(
status_code
).get("superimpose"):
operation.setdefault("responses", {}).setdefault(
status_code, {}
).setdefault("content", {}).setdefault(route_response_media_type, {})[
"schema"
] = response_schema
).setdefault("content", {}).setdefault(
route_response_media_type, {}
)["schema"] = response_schema
if route.responses:
operation_responses = operation.setdefault("responses", {})

110
tests/test_additional_responses_default_media_type.py
View File

@ -3,11 +3,10 @@
over the default route's response class's media_type.
"""
from pydantic import BaseModel
from starlette.status import HTTP_200_OK, HTTP_500_INTERNAL_SERVER_ERROR
from fastapi import FastAPI
from fastapi.testclient import TestClient
from pydantic import BaseModel
from starlette.status import HTTP_200_OK, HTTP_500_INTERNAL_SERVER_ERROR
app = FastAPI()
client = TestClient(app)
@ -22,7 +21,7 @@ class Error(BaseModel):
"/a",
responses={
HTTP_200_OK: {"superimpose": True, "content": {"text/event-stream": {}}},
HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error}
HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error},
},
)
def a():
@ -33,14 +32,19 @@ def a():
"/b",
responses={
str(HTTP_200_OK): {"content": {"text/event-stream": {}}},
HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error}
}
HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error},
},
)
def b():
pass # pragma: no cover
@app.get("/c", responses={HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error}})
@app.get(
"/c",
responses={
HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Error", "model": Error}
},
)
def c():
pass # pragma: no cover
@ -56,10 +60,7 @@ def test_openapi_schema():
assert response.status_code == 200, response.text
assert response.json() == {
"openapi": "3.1.0",
"info": {
"title": "FastAPI",
"version": "0.1.0"
},
"info": {"title": "FastAPI", "version": "0.1.0"},
"paths": {
"/a": {
"get": {
@ -68,23 +69,17 @@ def test_openapi_schema():
"responses": {
"200": {
"description": "Successful Response",
"content": {
"text/event-stream": {
}
}
"content": {"text/event-stream": {}},
},
"500": {
"description": "Error",
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/Error"
}
"schema": {"$ref": "#/components/schemas/Error"}
}
}
}
}
},
},
},
}
},
"/b": {
@ -95,27 +90,19 @@ def test_openapi_schema():
"200": {
"description": "Successful Response",
"content": {
"application/json": {
"schema": {
}
},
"text/event-stream": {
}
}
"application/json": {"schema": {}},
"text/event-stream": {},
},
},
"500": {
"description": "Error",
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/Error"
}
"schema": {"$ref": "#/components/schemas/Error"}
}
}
}
}
},
},
},
}
},
"/c": {
@ -125,25 +112,17 @@ def test_openapi_schema():
"responses": {
"200": {
"description": "Successful Response",
"content": {
"application/json": {
"schema": {
}
}
}
"content": {"application/json": {"schema": {}}},
},
"500": {
"description": "Error",
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/Error"
}
"schema": {"$ref": "#/components/schemas/Error"}
}
}
}
}
},
},
},
}
},
"/d": {
@ -153,38 +132,23 @@ def test_openapi_schema():
"responses": {
"200": {
"description": "Successful Response",
"content": {
"application/json": {
"schema": {
}
}
}
"content": {"application/json": {"schema": {}}},
}
}
},
}
}
},
},
"components": {
"schemas": {
"Error": {
"properties": {
"status": {
"type": "string",
"title": "Status"
},
"title": {
"type": "string",
"title": "Title"
}
"status": {"type": "string", "title": "Status"},
"title": {"type": "string", "title": "Title"},
},
"type": "object",
"required": [
"status",
"title"
],
"title": "Error"
"required": ["status", "title"],
"title": "Error",
}
}
}
},
}