zip_path

Author: MaxNumerique

tests/test_utils_functions.py

@@ -6,7 +6,6 @@
 import flask
 import shutil
 import uuid
-import glob
 import zipfile
 import io
 
@@ -224,11 +223,10 @@ def test_send_file_multiple_returns_zip(client, tmp_path):
             response = utils_functions.send_file(app.config["UPLOAD_FOLDER"], file_paths, "bundle")
             assert response.status_code == 200
             assert response.mimetype == "application/zip"
-            assert response.headers.get("new-file-name") == "bundle.zip"
-
-            response.direct_passthrough = False
-            zip_bytes = response.get_data()
-            with zipfile.ZipFile(io.BytesIO(zip_bytes), "r") as zip_file:
+            new_file_name = response.headers.get("new-file-name")
+            assert new_file_name == "bundle.zip"
+            zip_path = os.path.join(app.config["UPLOAD_FOLDER"], new_file_name)
+            with zipfile.ZipFile(zip_path, "r") as zip_file:
                 zip_entries = zip_file.namelist()
                 assert "tmp_send_file_1.txt" in zip_entries
                 assert "tmp_send_file_2.txt" in zip_entries
@@ -245,9 +243,10 @@ def test_send_file_single_returns_octet_binary(client, tmp_path):
             response = utils_functions.send_file(app.config["UPLOAD_FOLDER"], [str(file_path)], "tmp_send_file_1.txt")
             assert response.status_code == 200
             assert response.mimetype == "application/octet-binary"
-            assert response.headers.get("new-file-name") == "tmp_send_file_1.txt"
-
-            response.direct_passthrough = False
-            file_bytes = response.get_data()
+            new_file_name = response.headers.get("new-file-name")
+            assert new_file_name == "tmp_send_file_1.txt"
+            zip_path = os.path.join(app.config["UPLOAD_FOLDER"], new_file_name)
+            with open(zip_path, "rb") as f:
+                file_bytes = f.read()
             assert file_bytes == b"hello 1"
             response.close()