1 files changed, 37 insertions, 0 deletions
diff --git a/tests/test_server.py b/tests/test_server.py
new file mode 100644
index 0000000..4590049
--- /dev/null
+++ b/tests/test_server.py
@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from mcp import Client
+from mcp.types import ImageContent
+
+from mcp_sd15.server import create_server
+from mcp_sd15.service import ImageGenerationService
+
+from .test_service import FakeBackend
+
+
+async def test_exposes_exactly_three_tools_with_closed_enums(app_config) -> None:
+    server = create_server(ImageGenerationService(app_config, FakeBackend()))
+    async with Client(server) as client:
+        tools = (await client.list_tools()).tools
+    assert {tool.name for tool in tools} == {
+        "generate_image",
+        "generate_guided_image",
+        "inpaint_image",
+    }
+    guided = next(tool for tool in tools if tool.name == "generate_guided_image")
+    assert guided.input_schema["properties"]["control_mode"]["enum"] == [
+        "depth",
+        "openpose",
+        "openpose_full",
+    ]
+    assert "width" not in guided.input_schema["properties"]
+
+
+async def test_mcp_call_returns_png_image_content(app_config) -> None:
+    server = create_server(ImageGenerationService(app_config, FakeBackend()))
+    async with Client(server) as client:
+        result = await client.call_tool("generate_image", {"positive_prompt": "a cat"})
+    assert result.is_error is False
+    assert len(result.content) == 1
+    assert isinstance(result.content[0], ImageContent)
+    assert result.content[0].mime_type == "image/png"