pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-25 12:25:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update test_s3_endpoints.py

Browse Source
This commit is contained in:
Eric Hare
2026-05-21 14:28:23 -07:00
parent a5eeaa47d3
commit 879a45224f
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/backend/tests/unit/api/test_s3_endpoints.py
View File

@ -59,10 +59,12 @@ class TestS3FileEndpoints:
mock_file = MagicMock()
mock_file.path = "user_123/subfolder/document.pdf"
mock_file.name = "document"
mock_file.user_id = "user_123"
with (
patch("langflow.api.v2.files.fetch_file_object", return_value=mock_file),
patch("langflow.api.v2.files.CurrentActiveUser", return_value=mock_user),
patch("langflow.api.v2.files.ensure_file_permission", new=AsyncMock()),
):
from langflow.api.v2.files import download_file
@ -90,10 +92,12 @@ class TestS3FileEndpoints:
mock_file = MagicMock()
mock_file.path = "user_123/document.pdf"
mock_file.name = "document"
mock_file.user_id = "user_123"
with (
patch("langflow.api.v2.files.fetch_file_object", return_value=mock_file),
patch("langflow.api.v2.files.CurrentActiveUser", return_value=mock_user),
patch("langflow.api.v2.files.ensure_file_permission", new=AsyncMock()),
):
from fastapi.responses import StreamingResponse
from langflow.api.v2.files import download_file
@ -125,10 +129,12 @@ class TestS3FileEndpoints:
mock_file = MagicMock()
mock_file.path = "user_123/document.txt"
mock_file.name = "document"
mock_file.user_id = "user_123"
with (
patch("langflow.api.v2.files.fetch_file_object", return_value=mock_file),
patch("langflow.api.v2.files.CurrentActiveUser", return_value=mock_user),
patch("langflow.api.v2.files.ensure_file_permission", new=AsyncMock()),
):
from langflow.api.v2.files import download_file
@ -157,6 +163,7 @@ class TestS3FileEndpoints:
mock_file = MagicMock()
mock_file.path = "user_123/folder/document.pdf"
mock_file.name = "document"
mock_file.user_id = "user_123"
mock_session = MagicMock()
mock_session.delete = AsyncMock()
@ -164,6 +171,7 @@ class TestS3FileEndpoints:
with (
patch("langflow.api.v2.files.fetch_file_object", return_value=mock_file),
patch("langflow.api.v2.files.CurrentActiveUser", return_value=mock_user),
patch("langflow.api.v2.files.ensure_file_permission", new=AsyncMock()),
):
from langflow.api.v2.files import delete_file
@ -196,10 +204,12 @@ class TestS3FileEndpoints:
mock_file = MagicMock()
mock_file.path = "user_123/missing.pdf"
mock_file.name = "missing"
mock_file.user_id = "user_123"
with (
patch("langflow.api.v2.files.fetch_file_object", return_value=mock_file),
patch("langflow.api.v2.files.CurrentActiveUser", return_value=mock_user),
patch("langflow.api.v2.files.ensure_file_permission", new=AsyncMock()),
):
from langflow.api.v2.files import download_file