pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.2 KiB
79 lines
2.2 KiB
import warnings
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.exceptions import FastAPIDeprecationWarning
|
|
from fastapi.responses import ORJSONResponse, UJSONResponse # ty: ignore[deprecated]
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
from tests.utils import needs_orjson, needs_ujson
|
|
|
|
|
|
class Item(BaseModel):
|
|
name: str
|
|
price: float
|
|
|
|
|
|
# ORJSON
|
|
|
|
|
|
def _make_orjson_app() -> FastAPI:
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", FastAPIDeprecationWarning)
|
|
app = FastAPI(default_response_class=ORJSONResponse) # ty: ignore[deprecated]
|
|
|
|
@app.get("/items")
|
|
def get_items() -> Item:
|
|
return Item(name="widget", price=9.99)
|
|
|
|
return app
|
|
|
|
|
|
@needs_orjson
|
|
def test_orjson_response_returns_correct_data():
|
|
app = _make_orjson_app()
|
|
client = TestClient(app)
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", FastAPIDeprecationWarning)
|
|
response = client.get("/items")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"name": "widget", "price": 9.99}
|
|
|
|
|
|
@needs_orjson
|
|
def test_orjson_response_emits_deprecation_warning():
|
|
with pytest.warns(FastAPIDeprecationWarning, match="ORJSONResponse is deprecated"):
|
|
ORJSONResponse(content={"hello": "world"}) # ty: ignore[deprecated]
|
|
|
|
|
|
# UJSON
|
|
|
|
|
|
def _make_ujson_app() -> FastAPI:
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", FastAPIDeprecationWarning)
|
|
app = FastAPI(default_response_class=UJSONResponse) # ty: ignore[deprecated]
|
|
|
|
@app.get("/items")
|
|
def get_items() -> Item:
|
|
return Item(name="widget", price=9.99)
|
|
|
|
return app
|
|
|
|
|
|
@needs_ujson
|
|
def test_ujson_response_returns_correct_data():
|
|
app = _make_ujson_app()
|
|
client = TestClient(app)
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore", FastAPIDeprecationWarning)
|
|
response = client.get("/items")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"name": "widget", "price": 9.99}
|
|
|
|
|
|
@needs_ujson
|
|
def test_ujson_response_emits_deprecation_warning():
|
|
with pytest.warns(FastAPIDeprecationWarning, match="UJSONResponse is deprecated"):
|
|
UJSONResponse(content={"hello": "world"}) # ty: ignore[deprecated]
|
|
|