mirror of https://github.com/tiangolo/fastapi.git
Add test
This commit is contained in:
parent
18599a7d00
commit
39cb6471f7
|
|
@ -1,5 +1,3 @@
|
|||
from typing import Optional
|
||||
|
||||
from fastapi import FastAPI, File
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
|
|
@ -7,7 +5,7 @@ app = FastAPI()
|
|||
|
||||
|
||||
@app.post("/files")
|
||||
async def upload_files(files: Optional[list[bytes]] = File(None)):
|
||||
async def upload_files(files: list[bytes] | None = File(None)):
|
||||
if files is None:
|
||||
return {"files_count": 0}
|
||||
return {"files_count": len(files), "sizes": [len(f) for f in files]}
|
||||
|
|
@ -28,3 +26,13 @@ def test_optional_bytes_list_no_files():
|
|||
response = client.post("/files")
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"files_count": 0}
|
||||
|
||||
|
||||
def test_optional_bytes_list_send_empty_str():
|
||||
client = TestClient(app)
|
||||
response = client.post(
|
||||
"/files",
|
||||
files=[("files", b"")],
|
||||
)
|
||||
assert response.status_code == 200
|
||||
assert response.json() == {"files_count": 1, "sizes": [0]}
|
||||
|
|
|
|||
Loading…
Reference in New Issue