test: update download API patch points

This commit is contained in:
Dariusz L
2026-08-02 18:19:39 +02:00
parent 42ec5aa048
commit f09447f16c
2 changed files with 8 additions and 4 deletions
+6 -2
View File
@@ -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()
+2 -2
View File
@@ -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)