test: add astra integration test (#2189)

* add first astra integ test framework

* use fixtures

* remove old tests from merge

* Add correct sender type

* chore: Update unit test command in GitHub workflow

---------

Co-authored-by: ogabrielluiz <gabriel@langflow.org>

tests/unit/test_files.py

@@ -0,0 +1,104 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from langflow.services.deps import get_storage_service
+from langflow.services.storage.service import StorageService
+
+
+@pytest.fixture
+def mock_storage_service():
+    # Create a mock instance of StorageService
+    service = MagicMock(spec=StorageService)
+    # Setup mock behaviors for the service methods as needed
+    service.save_file.return_value = None
+    service.get_file.return_value = b"file content"  # Binary content for files
+    service.list_files.return_value = ["file1.txt", "file2.jpg"]
+    service.delete_file.return_value = None
+    return service
+
+
+def test_upload_file(client, mock_storage_service, created_api_key, flow):
+    headers = {"x-api-key": created_api_key.api_key}
+    # Replace the actual storage service with the mock
+    client.app.dependency_overrides[get_storage_service] = lambda: mock_storage_service
+
+    response = client.post(
+        f"api/v1/files/upload/{flow.id}",
+        files={"file": ("test.txt", b"test content")},
+        headers=headers,
+    )
+    assert response.status_code == 201
+    assert response.json() == {
+        "flowId": str(flow.id),
+        "file_path": f"{flow.id}/test.txt",
+    }
+
+
+def test_download_file(client, mock_storage_service, created_api_key, flow):
+    headers = {"x-api-key": created_api_key.api_key}
+    client.app.dependency_overrides[get_storage_service] = lambda: mock_storage_service
+
+    response = client.get(f"api/v1/files/download/{flow.id}/test.txt", headers=headers)
+    assert response.status_code == 200
+    assert response.content == b"file content"
+
+
+def test_list_files(client, mock_storage_service, created_api_key, flow):
+    headers = {"x-api-key": created_api_key.api_key}
+    client.app.dependency_overrides[get_storage_service] = lambda: mock_storage_service
+
+    response = client.get(f"api/v1/files/list/{flow.id}", headers=headers)
+    assert response.status_code == 200
+    assert response.json() == {"files": ["file1.txt", "file2.jpg"]}
+
+
+def test_delete_file(client, mock_storage_service, created_api_key, flow):
+    headers = {"x-api-key": created_api_key.api_key}
+    client.app.dependency_overrides[get_storage_service] = lambda: mock_storage_service
+
+    response = client.delete(f"api/v1/files/delete/{flow.id}/test.txt", headers=headers)
+    assert response.status_code == 200
+    assert response.json() == {"message": "File test.txt deleted successfully"}
+
+
+def test_file_operations(client, created_api_key, flow):
+    headers = {"x-api-key": created_api_key.api_key}
+    flow_id = flow.id
+    file_name = "test.txt"
+    file_content = b"Hello, world!"
+
+    # Step 1: Upload the file
+    response = client.post(
+        f"api/v1/files/upload/{flow_id}",
+        files={"file": (file_name, file_content)},
+        headers=headers,
+    )
+    assert response.status_code == 201
+    assert response.json() == {
+        "flowId": str(flow_id),
+        "file_path": f"{flow_id}/{file_name}",
+    }
+
+    # Step 2: List files in the folder
+    response = client.get(f"api/v1/files/list/{flow_id}", headers=headers)
+    assert response.status_code == 200
+    assert file_name in response.json()["files"]
+
+    # Step 3: Download the file and verify its content
+
+    response = client.get(f"api/v1/files/download/{flow_id}/{file_name}", headers=headers)
+    assert response.status_code == 200
+    assert response.content == file_content
+    # the headers are application/octet-stream
+    assert response.headers["content-type"] == "application/octet-stream"
+    # mime_type is inside media_type
+
+    # Step 4: Delete the file
+    response = client.delete(f"api/v1/files/delete/{flow_id}/{file_name}", headers=headers)
+    assert response.status_code == 200
+    assert response.json() == {"message": f"File {file_name} deleted successfully"}
+
+    # Verify that the file is indeed deleted
+    response = client.get(f"api/v1/files/list/{flow_id}", headers=headers)
+    assert file_name not in response.json()["files"]