Browse Source

add tests for using `Form` and `File` in the same endpoint

pull/11194/head
Thomas LEVEIL 5 years ago
committed by Thomas LEVEIL
parent
commit
7e92b2b46d
  1. 92
      tests/test_file_and_form_order_issue_9116.py

92
tests/test_file_and_form_order_issue_9116.py

@ -0,0 +1,92 @@
"""
See https://github.com/tiangolo/fastapi/discussions/9116
Showcases regression introduced in commit ab2b86f
"""
import pathlib
from typing import List
import pytest
from fastapi import FastAPI, File, Form
from fastapi.testclient import TestClient
from typing_extensions import Annotated
app = FastAPI()
@app.post("/file_before_form")
def file_before_form(
file: bytes = File(...),
city: str = Form(...),
):
return {"file_content": file, "city": city}
@app.post("/file_after_form")
def file_after_form(
city: str = Form(...),
file: bytes = File(...),
):
return {"file_content": file, "city": city}
@app.post("/file_list_before_form")
def file_list_before_form(
files: Annotated[List[bytes], File()],
city: Annotated[str, Form(...)],
):
return {"file_contents": files, "city": city}
@app.post("/file_list_after_form")
def file_list_after_form(
city: Annotated[str, Form(...)],
files: Annotated[List[bytes], File()],
):
return {"file_contents": files, "city": city}
client = TestClient(app)
@pytest.fixture
def tmp_file_1(tmp_path) -> pathlib.Path:
f = tmp_path / "example1.txt"
f.write_text("foo")
return f
@pytest.fixture
def tmp_file_2(tmp_path) -> pathlib.Path:
f = tmp_path / "example2.txt"
f.write_text("bar")
return f
@pytest.mark.parametrize("endpoint_path", ("/file_before_form", "/file_after_form"))
def test_file_form_order(endpoint_path: str, tmp_file_1):
response = client.post(
url=endpoint_path,
data={"city": "Thimphou"},
files={"file": (tmp_file_1.name, tmp_file_1.read_bytes())},
)
assert response.status_code == 200, response.text
assert response.json() == {"file_content": "foo", "city": "Thimphou"}
@pytest.mark.parametrize(
"endpoint_path", ("/file_list_before_form", "/file_list_after_form")
)
def test_file_list_form_order(endpoint_path: str, tmp_file_1, tmp_file_2):
response = client.post(
url=endpoint_path,
data={"city": "Thimphou"},
files=(
("files", (tmp_file_1.name, tmp_file_1.read_bytes())),
("files", (tmp_file_2.name, tmp_file_2.read_bytes())),
),
)
assert response.status_code == 200, response.text
assert response.json() == {"file_contents": ["foo", "bar"], "city": "Thimphou"}
Loading…
Cancel
Save