xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 20:47:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add output and user to valid directory test.

Browse Source
This commit is contained in:
Robin Huang 2024-08-10 13:25:48 -07:00
parent 9f6196c1cf
commit b3b7b9a3bf
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests-unit/server/routes/internal_routes_test.py
View File

@ -32,6 +32,12 @@ async def test_list_files_valid_directory(aiohttp_client_factory, internal_route
assert len(data['files']) == 2 assert len(data['files']) == 2
assert data['files'] == mock_file_list assert data['files'] == mock_file_list
# Check other valid directories
resp = await client.get('/files?directory=user')
assert resp.status == 200
resp = await client.get('/files?directory=output')
assert resp.status == 200
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_list_files_invalid_directory(aiohttp_client_factory, internal_routes): async def test_list_files_invalid_directory(aiohttp_client_factory, internal_routes):
internal_routes.file_service.list_files = MagicMock(side_effect=ValueError("Invalid directory key")) internal_routes.file_service.list_files = MagicMock(side_effect=ValueError("Invalid directory key"))