OpenAI-compatible adapter for ComfyUI image generation and editing

tests/test_errors.py

@@ -0,0 +1,29 @@
+from app.errors import ComfyUIError, InvalidRequest, Unauthorized, error_body
+from app.schemas import ImageGenerationRequest
+
+import pytest
+
+
+def test_error_body_canonical_shape():
+    body = error_body("boom", "server_error", "comfyui_timeout")
+    assert body == {"error": {"message": "boom", "type": "server_error", "param": None, "code": "comfyui_timeout"}}
+
+
+def test_exception_status_mapping():
+    assert InvalidRequest("x").status == 400
+    assert Unauthorized().status == 401
+    assert ComfyUIError("x").status == 502
+
+
+def test_generation_request_defaults():
+    r = ImageGenerationRequest(prompt="p")
+    assert r.size == "1024x1024"
+    assert r.n == 1
+    assert r.seed is None
+    assert r.model is None
+
+
+@pytest.mark.parametrize("bad", ["", "abc", "1024", "1024x", "0x10", "10x0"])
+def test_size_validation(bad):
+    with pytest.raises(ValueError):
+        ImageGenerationRequest(prompt="p", size=bad)