digital-garbageandClaude Sonnet 5.5
c0cdf9efd9
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
import json
|
|
|
|
from core import nodes_manager
|
|
|
|
|
|
def test_providers_reads_managers_node_map(tmp_path, monkeypatch):
|
|
manager = tmp_path / "ComfyUI-Manager"
|
|
manager.mkdir()
|
|
(manager / "extension-node-map.json").write_text(json.dumps({
|
|
"https://github.com/a/pack-a": [["NodeOne", "NodeTwo"], {"title": "A"}],
|
|
"https://github.com/b/pack-b": [["NodeTwo"], {}],
|
|
}))
|
|
monkeypatch.setattr(nodes_manager, "custom_nodes_root", lambda: tmp_path)
|
|
got = nodes_manager.providers(["NodeOne", "NodeTwo", "Nobody"])
|
|
assert got["manager"] is True
|
|
assert got["providers"] == {"NodeOne": "https://github.com/a/pack-a", "NodeTwo": "https://github.com/a/pack-a", "Nobody": None}
|
|
|
|
|
|
def test_without_manager_nothing_is_known(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(nodes_manager, "custom_nodes_root", lambda: tmp_path)
|
|
assert nodes_manager.providers(["X"]) == {"providers": {"X": None}, "manager": False}
|
|
(tmp_path / "ComfyUI-Manager").mkdir()
|
|
(tmp_path / "ComfyUI-Manager" / "extension-node-map.json").write_text("not json")
|
|
assert nodes_manager.providers(["X"]) == {"providers": {"X": None}, "manager": False}
|
|
|
|
|
|
def test_manager_folder_case_does_not_matter(tmp_path, monkeypatch):
|
|
(tmp_path / "comfyui-manager").mkdir()
|
|
(tmp_path / "comfyui-manager" / "extension-node-map.json").write_text(json.dumps({"https://github.com/a/p": [["N"], {}]}))
|
|
monkeypatch.setattr(nodes_manager, "custom_nodes_root", lambda: tmp_path)
|
|
assert nodes_manager.providers(["N"])["providers"]["N"] == "https://github.com/a/p"
|