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.
77 lines
2.0 KiB
77 lines
2.0 KiB
from fastapi import FastAPI
|
|
from fastapi.params import Body, Cookie, Form, Header, Query
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel, Field
|
|
from typing_extensions import Annotated
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class DataModel(BaseModel):
|
|
foo: str = Field(alias="bar", default="default_foo")
|
|
bar: str = Field(alias="foo", default="default_bar")
|
|
|
|
|
|
@app.post("/param/body")
|
|
def post_param_body(data: Annotated[DataModel, Body()]):
|
|
return data
|
|
|
|
|
|
@app.post("/param/form")
|
|
def post_param_form(data: Annotated[DataModel, Form()]):
|
|
return data
|
|
|
|
|
|
@app.post("/param/query")
|
|
def post_param_query(data: Annotated[DataModel, Query()]):
|
|
return data
|
|
|
|
|
|
@app.post("/param/cookies")
|
|
def post_param_cookies(data: Annotated[DataModel, Cookie()]):
|
|
return data
|
|
|
|
|
|
@app.post("/param/headers")
|
|
def post_param_headers(data: Annotated[DataModel, Header()]):
|
|
return data
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
expected_json = {
|
|
"bar": "default_foo",
|
|
"foo": "set_foo",
|
|
}
|
|
|
|
|
|
def test_param_body_with_alias():
|
|
response = client.post("/param/body", json={"foo": "set_foo"})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == expected_json
|
|
|
|
|
|
def test_param_form_with_alias():
|
|
response = client.post("/param/form", data={"foo": "set_foo"})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == expected_json
|
|
|
|
|
|
def test_param_query_with_alias():
|
|
response = client.post("/param/query", params={"foo": "set_foo"})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == expected_json
|
|
|
|
|
|
def test_param_headers_with_alias():
|
|
response = client.post("/param/headers", headers={"foo": "set_foo"})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == expected_json
|
|
|
|
|
|
def test_param_cookies_with_alias():
|
|
with client as c:
|
|
c.cookies.set("foo", "set_foo")
|
|
response = c.post("/param/cookies")
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == expected_json
|
|
|