blob: 40d2b340701fe7d2de61a26f5f4b5abdd642c05d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
import zipfile
from pathlib import Path
from tempfile import TemporaryDirectory
from flask.testing import FlaskClient
def test_docs_upload(app, client: FlaskClient):
index_text = "This is a test."
with TemporaryDirectory() as tmpdir:
index_file = Path(tmpdir) / "index.html"
index_file.write_text(index_text)
zip_file = Path(tmpdir) / "docs.zip"
with zipfile.ZipFile(zip_file, "w") as z:
z.write(index_file, index_file.name)
with zip_file.open("rb") as z:
resp = client.post(f"/docs/upload?token={app.config['DOCS_AUTH_TOKEN']}", data={"data": z})
assert resp.status_code == 200
resp = client.get("/docs/index.html")
assert resp.status_code == 200
assert resp.data == index_text.encode()
|
