298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 12:23:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

use unittest.mock.patch

Browse Source
This commit is contained in:
bymyself 2025-04-11 09:08:03 -07:00
parent 02fe68121a
commit e526eab3c5
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests-unit/folder_paths_test/filter_by_content_types_test.py
View File

@ -2,7 +2,8 @@ import pytest
import os import os
import tempfile import tempfile
from folder_paths import filter_files_content_types, extension_mimetypes_cache from folder_paths import filter_files_content_types, extension_mimetypes_cache
import mimetypes from unittest.mock import patch
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def file_extensions(): def file_extensions():
@ -13,20 +14,6 @@ def file_extensions():
'model': ['gltf', 'glb', 'obj', 'mtl', 'fbx', 'stl'] 'model': ['gltf', 'glb', 'obj', 'mtl', 'fbx', 'stl']
} }
@pytest.fixture(scope="function")
def mock_model_mimetypes(file_extensions):
# Store original cache
original_cache = extension_mimetypes_cache.copy()
# Add model extensions to cache
for extension in file_extensions['model']:
extension_mimetypes_cache[extension] = 'model'
yield
# Restore original cache
extension_mimetypes_cache.clear()
extension_mimetypes_cache.update(original_cache)
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def mock_dir(file_extensions): def mock_dir(file_extensions):
@ -38,7 +25,18 @@ def mock_dir(file_extensions):
yield directory yield directory
def test_categorizes_all_correctly(mock_dir, file_extensions, mock_model_mimetypes): @pytest.fixture
def patched_mimetype_cache(file_extensions):
# Mock model file extensions since they may not be in the test-runner system's mimetype cache
new_cache = extension_mimetypes_cache.copy()
for extension in file_extensions["model"]:
new_cache[extension] = "model"
with patch("folder_paths.extension_mimetypes_cache", new_cache):
yield
def test_categorizes_all_correctly(mock_dir, file_extensions, patched_mimetype_cache):
files = os.listdir(mock_dir) files = os.listdir(mock_dir)
for content_type, extensions in file_extensions.items(): for content_type, extensions in file_extensions.items():
filtered_files = filter_files_content_types(files, [content_type]) filtered_files = filter_files_content_types(files, [content_type])
@ -46,7 +44,7 @@ def test_categorizes_all_correctly(mock_dir, file_extensions, mock_model_mimetyp
assert f"sample_{content_type}.{extension}" in filtered_files assert f"sample_{content_type}.{extension}" in filtered_files
def test_categorizes_all_uniquely(mock_dir, file_extensions): def test_categorizes_all_uniquely(mock_dir, file_extensions, patched_mimetype_cache):
files = os.listdir(mock_dir) files = os.listdir(mock_dir)
for content_type, extensions in file_extensions.items(): for content_type, extensions in file_extensions.items():
filtered_files = filter_files_content_types(files, [content_type]) filtered_files = filter_files_content_types(files, [content_type])