From f09447f16cbbb392443e60dc3adc05a81897cca8 Mon Sep 17 00:00:00 2001 From: Dariusz L Date: Sun, 2 Aug 2026 18:19:39 +0200 Subject: [PATCH] test: update download API patch points --- tests/test_refactoring_targets.py | 8 ++++++-- tests/test_route_registry.py | 4 ++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/test_refactoring_targets.py b/tests/test_refactoring_targets.py index 5929fd3..e4c6187 100644 --- a/tests/test_refactoring_targets.py +++ b/tests/test_refactoring_targets.py @@ -13,13 +13,17 @@ if parent_dir not in sys.path: # Import submodules dynamically to patch them before any higher level modules run setup_routes settings_mod = importlib.import_module("comfyui-model-resolver.core.settings") -downloader_mod = importlib.import_module("comfyui-model-resolver.core.downloader") +download_api_mod = importlib.import_module("comfyui-model-resolver.core.download.api") mock_load_settings = MagicMock(return_value={"aria2c_path": "default_path"}) mock_get_aria2_status = MagicMock(return_value={"status": "running"}) patch_load_settings = patch.object(settings_mod, "load_settings", mock_load_settings) -patch_get_aria2_status = patch.object(downloader_mod, "get_aria2_status", mock_get_aria2_status) +patch_get_aria2_status = patch.object( + download_api_mod, + "get_aria2_status", + mock_get_aria2_status, +) patch_load_settings.start() patch_get_aria2_status.start() diff --git a/tests/test_route_registry.py b/tests/test_route_registry.py index b341968..e41e5ce 100644 --- a/tests/test_route_registry.py +++ b/tests/test_route_registry.py @@ -139,8 +139,8 @@ def test_register_routes_skips_optional_download_routes_when_unavailable( route_environment, ): extension, routes = route_environment - downloader_module_name = f"{PACKAGE_NAME}.core.downloader" - monkeypatch.setitem(sys.modules, downloader_module_name, None) + download_api_module_name = f"{PACKAGE_NAME}.core.download.api" + monkeypatch.setitem(sys.modules, download_api_module_name, None) result = registry_module.register_routes(extension)