jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add mime type validation for file download

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-27 00:11:00 -03:00
commit bcd0151e7b
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_files.py
View file

@ -3,6 +3,7 @@ from unittest.mock import MagicMock
import pytest import pytest
from langflow.services.deps import get_storage_service from langflow.services.deps import get_storage_service
from langflow.services.storage.service import StorageService from langflow.services.storage.service import StorageService
from langflow.services.storage.utils import build_content_type_from_extension
@pytest.fixture @pytest.fixture
@ -66,9 +67,11 @@ def test_file_operations(client):
assert file_name in response.json()["files"] assert file_name in response.json()["files"]
# Step 3: Download the file and verify its content # Step 3: Download the file and verify its content
mime_type = build_content_type_from_extension(file_name.split(".")[-1])
response = client.get(f"api/v1/files/download/{flow_id}/{file_name}") response = client.get(f"api/v1/files/download/{flow_id}/{file_name}")
assert response.status_code == 200 assert response.status_code == 200
assert response.content == file_content assert response.content == file_content
assert mime_type in response.headers["content-type"]
# Step 4: Delete the file # Step 4: Delete the file
response = client.delete(f"api/v1/files/delete/{flow_id}/{file_name}") response = client.delete(f"api/v1/files/delete/{flow_id}/{file_name}")