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

add model extensions to test runner mimetype cache manually

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

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

@ -1,7 +1,8 @@
import pytest import pytest
import os import os
import tempfile import tempfile
from folder_paths import filter_files_content_types from folder_paths import filter_files_content_types, extension_mimetypes_cache
import mimetypes
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def file_extensions(): def file_extensions():
@ -12,6 +13,20 @@ 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):
@ -23,7 +38,7 @@ def mock_dir(file_extensions):
yield directory yield directory
def test_categorizes_all_correctly(mock_dir, file_extensions): def test_categorizes_all_correctly(mock_dir, file_extensions, mock_model_mimetypes):
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])