OpenAI-compatible adapter for ComfyUI image generation and editing

tests/test_injector.py

@@ -0,0 +1,73 @@
+import pytest
+
+from app.config import DEFAULT_MAPPING
+from app.errors import InvalidRequest
+from app.injector import build_workflow, new_seed, parse_size
+from app.templates import load_all
+from app.config import Settings
+
+
+@pytest.fixture(scope="module")
+def templates():
+    return load_all(Settings(_env_file=None))
+
+
+def test_parse_size():
+    assert parse_size("1024x1024") == (1024, 1024)
+    assert parse_size("832X1216") == (832, 1216)
+
+
+@pytest.mark.parametrize("bad", ["1024", "ax1024", "1024x", "0x100", "1024x-2", "wide"])
+def test_parse_size_invalid(bad):
+    with pytest.raises(InvalidRequest):
+        parse_size(bad)
+
+
+def test_seed_random_differs():
+    assert new_seed(None) != new_seed(None)
+
+
+def test_seed_fixed_and_range():
+    assert new_seed(42) == 42
+    with pytest.raises(InvalidRequest):
+        new_seed(-1)
+
+
+def test_create_injection(templates):
+    create, _ = templates
+    mapping = DEFAULT_MAPPING["create"]
+    graph = build_workflow(create, mapping, prompt="a cat", seed=77, size="832x1216")
+    assert graph["83:27"]["inputs"]["text"] == "a cat"
+    assert graph["83:13"]["inputs"]["width"] == 832
+    assert graph["83:13"]["inputs"]["height"] == 1216
+    assert graph["83:3"]["inputs"]["seed"] == 77
+    # template untouched
+    assert create["83:27"]["inputs"]["text"] == "\n"
+    assert create["83:13"]["inputs"]["width"] == 1280
+
+
+def test_edit_injection(templates):
+    _, edit = templates
+    mapping = DEFAULT_MAPPING["edit"]
+    graph = build_workflow(edit, mapping, prompt="make it night", seed=5, image_name="adapter_x.png")
+    assert graph["115:111"]["inputs"]["prompt"] == "make it night"
+    assert graph["115:110"]["inputs"]["prompt"] == ""
+    assert graph["78"]["inputs"]["image"] == "adapter_x.png"
+    assert graph["115:3"]["inputs"]["seed"] == 5
+
+
+def test_edit_seed_varies_per_call(templates):
+    _, edit = templates
+    mapping = DEFAULT_MAPPING["edit"]
+    seeds = set()
+    for _ in range(3):
+        g = build_workflow(edit, mapping, prompt="p", seed=new_seed(None))
+        seeds.add(g["115:3"]["inputs"]["seed"])
+    assert len(seeds) == 3
+
+
+def test_unknown_mapping_node_raises(templates):
+    create, _ = templates
+    bad_mapping = {"prompt": {"node": "999", "field": "text"}}
+    with pytest.raises(InvalidRequest, match="999"):
+        build_workflow(create, bad_mapping, prompt="p", seed=1)