run-llama/create-llama
1
0
Fork 0
mirror of https://github.com/run-llama/create-llama.git synced 2026-07-19 23:13:36 -04:00
Code Issues 39 Packages Projects Releases 10 Wiki Activity

fix ruff

Browse Source
This commit is contained in:
leehuwuj
2025-03-25 13:58:02 +07:00
parent 90b3e86249
commit a13dd8a901
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
llama-index-server/tests/services/test_file_service.py
+15
View File
@@ -76,7 +76,12 @@ class TestFileService:
# Assert
expected_path = os.path.join("test_dir", f"test_{test_uuid}.txt")
mock_makedirs.assert_called_once_with(
os.path.dirname(expected_path), exist_ok=True
)
mock_file_open.assert_called_once_with(expected_path, "wb")
mock_file_open().write.assert_called_once_with(b"Hello World")
assert result.path == expected_path
assert result.type == "txt"
@patch("uuid.uuid4")
@@ -98,6 +103,11 @@ class TestFileService:
# Assert
expected_path = os.path.join("test_dir", f"test_file__{test_uuid}.txt")
mock_makedirs.assert_called_once_with(
os.path.dirname(expected_path), exist_ok=True
)
mock_file_open.assert_called_once_with(expected_path, "wb")
assert result.path == expected_path
assert result.name == f"test_file__{test_uuid}.txt"
@patch("uuid.uuid4")
@@ -143,6 +153,11 @@ class TestFileService:
# Assert
expected_path = os.path.join("test_dir", f"test_{test_uuid}.txt")
mock_makedirs.assert_called_once_with(
os.path.dirname(expected_path), exist_ok=True
)
mock_file_open.assert_called_once_with(expected_path, "wb")
assert result.path == expected_path
expected_url = os.path.join(
"https://custom-url.com/files", "test_dir", f"test_{test_uuid}.txt"
)