mirror of https://github.com/tiangolo/fastapi.git
105 lines
2.9 KiB
Python
105 lines
2.9 KiB
Python
import io
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
import pytest
|
|
from fastapi import FastAPI, File, Form, UploadFile
|
|
from fastapi.datastructures import Default
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
# TODO: remove when deprecating Pydantic v1
|
|
def test_upload_file_invalid():
|
|
with pytest.raises(ValueError):
|
|
UploadFile.validate("not a Starlette UploadFile")
|
|
|
|
|
|
def test_upload_file_invalid_pydantic_v2():
|
|
with pytest.raises(ValueError):
|
|
UploadFile._validate("not a Starlette UploadFile", {})
|
|
|
|
|
|
def test_default_placeholder_equals():
|
|
placeholder_1 = Default("a")
|
|
placeholder_2 = Default("a")
|
|
assert placeholder_1 == placeholder_2
|
|
assert placeholder_1.value == placeholder_2.value
|
|
|
|
|
|
def test_default_placeholder_bool():
|
|
placeholder_a = Default("a")
|
|
placeholder_b = Default("")
|
|
assert placeholder_a
|
|
assert not placeholder_b
|
|
|
|
|
|
@pytest.fixture(name="client", scope="module")
|
|
def get_client():
|
|
app = FastAPI()
|
|
client = TestClient(app)
|
|
return client
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def tmp_file(tmp_path_factory):
|
|
path = tmp_path_factory.getbasetemp() / "test.txt"
|
|
path.write_bytes(b"<file content>")
|
|
return path
|
|
|
|
|
|
def test_upload_file_is_closed(client: TestClient, tmp_file: Path):
|
|
app = client.app
|
|
|
|
testing_file_store: List[UploadFile] = []
|
|
|
|
@app.post("/uploadfile/")
|
|
def create_upload_file(file: UploadFile):
|
|
testing_file_store.append(file)
|
|
return {"filename": file.filename}
|
|
|
|
client = TestClient(app)
|
|
with tmp_file.open("rb") as file:
|
|
response = client.post("/uploadfile/", files={"file": file})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == {"filename": "test.txt"}
|
|
|
|
assert testing_file_store
|
|
assert testing_file_store[0].file.closed
|
|
|
|
|
|
def test_form_is_declared_before_file(client: TestClient, tmp_file: Path):
|
|
app = client.app
|
|
|
|
@app.post("/uploadfile_checksum/")
|
|
def create_upload_file_with_checksum(
|
|
checksum: str = Form(), file: UploadFile = File()
|
|
):
|
|
return {"checksum": checksum, "filename": file.filename}
|
|
|
|
client = TestClient(app)
|
|
with tmp_file.open("rb") as file:
|
|
response = client.post(
|
|
"/uploadfile_checksum/",
|
|
data={"checksum": "<file hash>"},
|
|
files={"file": file},
|
|
)
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == {"checksum": "<file hash>", "filename": "test.txt"}
|
|
|
|
|
|
# For UploadFile coverage, segments copied from Starlette tests
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_upload_file():
|
|
stream = io.BytesIO(b"data")
|
|
file = UploadFile(filename="file", file=stream, size=4)
|
|
assert await file.read() == b"data"
|
|
assert file.size == 4
|
|
await file.write(b" and more data!")
|
|
assert await file.read() == b""
|
|
assert file.size == 19
|
|
await file.seek(0)
|
|
assert await file.read() == b"data and more data!"
|
|
await file.close()
|