wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 06:40:48 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix frontend manager test patch

Browse Source
This commit is contained in:
doctorpangloss 2024-08-29 18:23:26 -07:00
parent 52230c24f2
commit d2acf8a93f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unit/app_test/frontend_manager_test.py
View File

@ -87,15 +87,15 @@ def test_init_frontend_invalid_provider():
@pytest.fixture
def mock_os_functions():
with patch('app.frontend_management.os.makedirs') as mock_makedirs, \
patch('app.frontend_management.os.listdir') as mock_listdir, \
patch('app.frontend_management.os.rmdir') as mock_rmdir:
with patch('comfy.app.frontend_management.os.makedirs') as mock_makedirs, \
patch('comfy.app.frontend_management.os.listdir') as mock_listdir, \
patch('comfy.app.frontend_management.os.rmdir') as mock_rmdir:
mock_listdir.return_value = [] # Simulate empty directory
yield mock_makedirs, mock_listdir, mock_rmdir
@pytest.fixture
def mock_download():
with patch('app.frontend_management.download_release_asset_zip') as mock:
with patch('comfy.app.frontend_management.download_release_asset_zip') as mock:
mock.side_effect = Exception("Download failed") # Simulate download failure
yield mock