fix: make files be saved in unique path (#8965)

Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>

src/backend/tests/unit/api/v2/test_files.py

@@ -256,3 +256,268 @@ async def test_upload_list_delete_and_validate_files(files_client, files_created
     assert file2["name"] in file_names
     assert file2["id"] in file_ids
     assert len(files) == 1
+
+
+async def test_upload_files_with_same_name_creates_unique_names(files_client, files_created_api_key):
+    """Test that uploading files with the same name creates unique filenames."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload first file
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("duplicate.txt", b"content1")},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+    assert file1["name"] == "duplicate"
+
+    # Upload second file with same name
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("duplicate.txt", b"content2")},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+    assert file2["name"] == "duplicate (1)"
+
+    # Upload third file with same name
+    response3 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("duplicate.txt", b"content3")},
+        headers=headers,
+    )
+    assert response3.status_code == 201
+    file3 = response3.json()
+    assert file3["name"] == "duplicate (2)"
+
+    # Verify all files can be downloaded with their unique content
+    download1 = await files_client.get(f"api/v2/files/{file1['id']}", headers=headers)
+    assert download1.status_code == 200
+    assert download1.content == b"content1"
+
+    download2 = await files_client.get(f"api/v2/files/{file2['id']}", headers=headers)
+    assert download2.status_code == 200
+    assert download2.content == b"content2"
+
+    download3 = await files_client.get(f"api/v2/files/{file3['id']}", headers=headers)
+    assert download3.status_code == 200
+    assert download3.content == b"content3"
+
+    # List files and verify all three are present with unique names
+    response = await files_client.get("api/v2/files", headers=headers)
+    assert response.status_code == 200
+    files = response.json()
+    file_names = [f["name"] for f in files]
+    assert "duplicate" in file_names
+    assert "duplicate (1)" in file_names
+    assert "duplicate (2)" in file_names
+    assert len(files) == 3
+
+
+async def test_upload_files_without_extension_creates_unique_names(files_client, files_created_api_key):
+    """Test that uploading files without extensions also creates unique filenames."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload first file without extension
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("noextension", b"content1")},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+    assert file1["name"] == "noextension"
+
+    # Upload second file with same name
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("noextension", b"content2")},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+    assert file2["name"] == "noextension (1)"
+
+    # Verify both files can be downloaded
+    download1 = await files_client.get(f"api/v2/files/{file1['id']}", headers=headers)
+    assert download1.status_code == 200
+    assert download1.content == b"content1"
+
+    download2 = await files_client.get(f"api/v2/files/{file2['id']}", headers=headers)
+    assert download2.status_code == 200
+    assert download2.content == b"content2"
+
+
+async def test_upload_files_with_different_extensions_same_name(files_client, files_created_api_key):
+    """Test that files with same root name but different extensions create unique names."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload file with .txt extension
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("document.txt", b"text content")},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+    assert file1["name"] == "document"
+
+    # Upload file with .md extension and same root name
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("document.md", b"markdown content")},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+    assert file2["name"] == "document (1)"
+
+    # Upload another .txt file with same root name
+    response3 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("document.txt", b"more text content")},
+        headers=headers,
+    )
+    assert response3.status_code == 201
+    file3 = response3.json()
+    assert file3["name"] == "document (2)"
+
+
+async def test_mcp_servers_file_replacement(files_client, files_created_api_key):
+    """Test that _mcp_servers file gets replaced instead of creating unique names."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload first _mcp_servers file
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("_mcp_servers.json", b'{"servers": ["server1"]}')},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+    assert file1["name"] == "_mcp_servers"
+
+    # Upload second _mcp_servers file - should replace the first one
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("_mcp_servers.json", b'{"servers": ["server2"]}')},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+    assert file2["name"] == "_mcp_servers"
+
+    # Note: _mcp_servers files are filtered out from the regular file list
+    # This is expected behavior since they're managed separately
+    response = await files_client.get("api/v2/files", headers=headers)
+    assert response.status_code == 200
+    files = response.json()
+    mcp_files = [f for f in files if f["name"] == "_mcp_servers"]
+    assert len(mcp_files) == 0  # MCP servers files are filtered out from regular list
+
+    # Verify the second file can be downloaded with the updated content
+    download2 = await files_client.get(f"api/v2/files/{file2['id']}", headers=headers)
+    assert download2.status_code == 200
+    assert download2.content == b'{"servers": ["server2"]}'
+
+    # Verify the first file no longer exists (should return 404)
+    download1 = await files_client.get(f"api/v2/files/{file1['id']}", headers=headers)
+    assert download1.status_code == 404
+
+    # Verify the file IDs are different (new file replaced old one)
+    assert file1["id"] != file2["id"]
+
+
+async def test_unique_filename_counter_handles_gaps(files_client, files_created_api_key):
+    """Test that the unique filename counter properly handles gaps in sequence."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload original file
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("gaptest.txt", b"content1")},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+    assert file1["name"] == "gaptest"
+
+    # Upload second file (should be gaptest (1))
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("gaptest.txt", b"content2")},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+    assert file2["name"] == "gaptest (1)"
+
+    # Upload third file (should be gaptest (2))
+    response3 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("gaptest.txt", b"content3")},
+        headers=headers,
+    )
+    assert response3.status_code == 201
+    file3 = response3.json()
+    assert file3["name"] == "gaptest (2)"
+
+    # Delete the middle file (gaptest (1))
+    delete_response = await files_client.delete(f"api/v2/files/{file2['id']}", headers=headers)
+    assert delete_response.status_code == 200
+
+    # Upload another file - should be gaptest (3), not filling the gap
+    response4 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("gaptest.txt", b"content4")},
+        headers=headers,
+    )
+    assert response4.status_code == 201
+    file4 = response4.json()
+    assert file4["name"] == "gaptest (3)"
+
+    # Verify final state
+    response = await files_client.get("api/v2/files", headers=headers)
+    assert response.status_code == 200
+    files = response.json()
+    file_names = [f["name"] for f in files]
+    assert "gaptest" in file_names
+    assert "gaptest (1)" not in file_names  # deleted
+    assert "gaptest (2)" in file_names
+    assert "gaptest (3)" in file_names
+    assert len([name for name in file_names if name.startswith("gaptest")]) == 3
+
+
+async def test_unique_filename_path_storage(files_client, files_created_api_key):
+    """Test that files with unique names are stored with unique paths."""
+    headers = {"x-api-key": files_created_api_key.api_key}
+
+    # Upload two files with same name
+    response1 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("pathtest.txt", b"path content 1")},
+        headers=headers,
+    )
+    assert response1.status_code == 201
+    file1 = response1.json()
+
+    response2 = await files_client.post(
+        "api/v2/files",
+        files={"file": ("pathtest.txt", b"path content 2")},
+        headers=headers,
+    )
+    assert response2.status_code == 201
+    file2 = response2.json()
+
+    # Verify both files have different paths and can be downloaded independently
+    assert file1["path"] != file2["path"]
+
+    download1 = await files_client.get(f"api/v2/files/{file1['id']}", headers=headers)
+    assert download1.status_code == 200
+    assert download1.content == b"path content 1"
+
+    download2 = await files_client.get(f"api/v2/files/{file2['id']}", headers=headers)
+    assert download2.status_code == 200
+    assert download2.content == b"path content 2"