From b3b7b9a3bfdbbd3faa965ff6dd779aea4073e2b6 Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Sat, 10 Aug 2024 13:25:48 -0700 Subject: [PATCH] Add output and user to valid directory test. --- tests-unit/server/routes/internal_routes_test.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests-unit/server/routes/internal_routes_test.py b/tests-unit/server/routes/internal_routes_test.py index c82fd5a53..2d2b43bd6 100644 --- a/tests-unit/server/routes/internal_routes_test.py +++ b/tests-unit/server/routes/internal_routes_test.py @@ -32,6 +32,12 @@ async def test_list_files_valid_directory(aiohttp_client_factory, internal_route assert len(data['files']) == 2 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 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"))