OpenAI-compatible adapter for ComfyUI image generation and editing

tests/test_templates.py

@@ -0,0 +1,57 @@
+import pytest
+
+from app.config import Settings
+from app.templates import TemplateError, check_mapping, find_broken_links, load_all
+
+
+def test_find_broken_links_detects_missing_node():
+    graph = {
+        "1": {"class_type": "X", "inputs": {"m": ["9", 0]}},
+    }
+    errors = find_broken_links(graph)
+    assert len(errors) == 1
+    assert "missing node '9'" in errors[0]
+    assert "node 1" in errors[0]
+
+
+def test_find_broken_links_ok_on_valid_graph():
+    graph = {
+        "1": {"class_type": "X", "inputs": {"m": ["2", 0]}},
+        "2": {"class_type": "Y", "inputs": {}},
+    }
+    assert find_broken_links(graph) == []
+
+
+def test_check_mapping_missing_node():
+    graph = {"1": {"class_type": "X", "inputs": {"text": ""}}}
+    mapping = {"create": {"prompt": {"node": "404", "field": "text"}}}
+    errors = check_mapping(graph, mapping["create"], "create")
+    assert any("not found" in e for e in errors)
+
+
+def test_check_mapping_non_scalar_field():
+    graph = {"1": {"class_type": "X", "inputs": {"text": ["2", 0]}}}
+    mapping = {"create": {"prompt": {"node": "1", "field": "text"}}}
+    errors = check_mapping(graph, mapping["create"], "create")
+    assert any("no scalar input" in e for e in errors)
+
+
+def test_shipped_templates_load_cleanly():
+    s = Settings(_env_file=None)
+    create, edit = load_all(s)
+    assert create["83:28"]["class_type"] == "UNETLoader"
+    assert edit["115:89"]["class_type"] == "LoraLoaderModelOnly"
+
+
+def test_broken_template_blocks_startup(tmp_path):
+    bad = tmp_path / "bad.json"
+    bad.write_text('{"1": {"class_type": "X", "inputs": {"m": ["9", 0]}}}', encoding="utf-8")
+    good = tmp_path / "good.json"
+    good.write_text(
+        '{"1": {"class_type": "X", "inputs": {"m": ["2", 0]}}, "2": {"class_type": "Y", "inputs": {}}}',
+        encoding="utf-8",
+    )
+    s = Settings(_env_file=None, create_template_path=bad, edit_template_path=good)
+    with pytest.raises(TemplateError) as exc:
+        load_all(s)
+    assert "missing node '9'" in str(exc.value)