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.
99 lines
2.7 KiB
99 lines
2.7 KiB
from typing import Annotated
|
|
|
|
import pytest
|
|
from fastapi import Depends, FastAPI, Path
|
|
from fastapi.param_functions import Query
|
|
from fastapi.testclient import TestClient
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
def test_no_annotated_defaults():
|
|
with pytest.raises(
|
|
AssertionError, match="Path parameters cannot have a default value"
|
|
):
|
|
|
|
@app.get("/items/{item_id}/")
|
|
async def get_item(item_id: Annotated[int, Path(default=1)]):
|
|
pass # pragma: nocover
|
|
|
|
with pytest.raises(
|
|
AssertionError,
|
|
match=(
|
|
"`Query` default value cannot be set in `Annotated` for 'item_id'. Set the"
|
|
" default value with `=` instead."
|
|
),
|
|
):
|
|
|
|
@app.get("/")
|
|
async def get(item_id: Annotated[int, Query(default=1)]):
|
|
pass # pragma: nocover
|
|
|
|
|
|
def test_multiple_annotations():
|
|
async def dep():
|
|
pass # pragma: nocover
|
|
|
|
@app.get("/multi-query")
|
|
async def get(foo: Annotated[int, Query(gt=2), Query(lt=10)]):
|
|
return foo
|
|
|
|
with pytest.raises(
|
|
AssertionError,
|
|
match=(
|
|
"Cannot specify `Depends` in `Annotated` and default value"
|
|
" together for 'foo'"
|
|
),
|
|
):
|
|
|
|
@app.get("/")
|
|
async def get2(foo: Annotated[int, Depends(dep)] = Depends(dep)):
|
|
pass # pragma: nocover
|
|
|
|
with pytest.raises(
|
|
AssertionError,
|
|
match=(
|
|
"Cannot specify a FastAPI annotation in `Annotated` and `Depends` as a"
|
|
" default value together for 'foo'"
|
|
),
|
|
):
|
|
|
|
@app.get("/")
|
|
async def get3(foo: Annotated[int, Query(min_length=1)] = Depends(dep)):
|
|
pass # pragma: nocover
|
|
|
|
client = TestClient(app)
|
|
response = client.get("/multi-query", params={"foo": "5"})
|
|
assert response.status_code == 200
|
|
assert response.json() == 5
|
|
|
|
response = client.get("/multi-query", params={"foo": "123"})
|
|
assert response.status_code == 422
|
|
|
|
response = client.get("/multi-query", params={"foo": "1"})
|
|
assert response.status_code == 422
|
|
|
|
|
|
def test_query_edge_cases():
|
|
app = FastAPI()
|
|
|
|
@app.get("/edge-query")
|
|
async def read_edge(q: Annotated[str | None, Query()] = None):
|
|
return {"q": q}
|
|
|
|
client = TestClient(app)
|
|
|
|
# Empty string
|
|
response = client.get("/edge-query?q=")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"q": ""}
|
|
|
|
# Special characters
|
|
response = client.get("/edge-query", params={"q": "@fastapi#123"})
|
|
assert response.status_code == 200
|
|
assert response.json() == {"q": "@fastapi#123"}
|
|
|
|
# Unicode input
|
|
response = client.get("/edge-query?q=नमस्ते")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"q": "नमस्ते"}
|
|
|