Shaman
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
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)
|