From 89b3539dd6f96086f635716fb47f6a5fa3805010 Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" Date: Mon, 23 Sep 2024 19:00:18 +0900 Subject: [PATCH] avoid 'mock_directory' use temp dir to avoid accidental fs pollution from tests --- tests-unit/prompt_server_test/download_models_test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests-unit/prompt_server_test/download_models_test.py b/tests-unit/prompt_server_test/download_models_test.py index bd41745fa..128dfeb9a 100644 --- a/tests-unit/prompt_server_test/download_models_test.py +++ b/tests-unit/prompt_server_test/download_models_test.py @@ -109,19 +109,18 @@ async def test_download_model_success(temp_dir): mock_make_request.assert_called_once_with('http://example.com/model.sft') @pytest.mark.asyncio -async def test_download_model_url_request_failure(): +async def test_download_model_url_request_failure(temp_dir): # Mock dependencies mock_response = AsyncMock(spec=ClientResponse) mock_response.status = 404 # Simulate a "Not Found" error mock_get = AsyncMock(return_value=mock_response) mock_progress_callback = AsyncMock() - fake_paths = {'checkpoints': (['mock_directory'], folder_paths.supported_pt_extensions)} + fake_paths = {'checkpoints': ([temp_dir], folder_paths.supported_pt_extensions)} # Mock the create_model_path function with patch('model_filemanager.create_model_path', return_value='/mock/path/model.safetensors'), \ patch('model_filemanager.check_file_exists', return_value=None), \ - patch('folder_paths.get_folder_paths', return_value=['mock_directory']), \ patch('folder_paths.folder_names_and_paths', fake_paths): # Call the function result = await download_model( @@ -129,7 +128,7 @@ async def test_download_model_url_request_failure(): 'model.safetensors', 'http://example.com/model.safetensors', 'checkpoints', - 'mock_directory', + temp_dir, mock_progress_callback )