[MCP-266] Add Gumloop extension tests and fix CI (#12)

dvlpjrs · web-flow · commit 4c1d01c249c9 · 2025-12-09T13:01:57.000-08:00
* [MCP-266] Add tests for Gumloop server extensions

* Fix uv workspace config for gumloop-mcp package name
diff --git a/examples/servers/simple-auth/pyproject.toml b/examples/servers/simple-auth/pyproject.toml
@@ -10,7 +10,7 @@ dependencies = [
     "anyio>=4.5",
     "click>=8.2.0",
     "httpx>=0.27",
-    "mcp",
+    "gumloop-mcp",
     "pydantic>=2.0",
     "pydantic-settings>=2.5.2",
     "sse-starlette>=1.6.1",
diff --git a/examples/servers/simple-pagination/pyproject.toml b/examples/servers/simple-pagination/pyproject.toml
@@ -9,7 +9,7 @@ maintainers = [
     { name = "David Soria Parra", email = "davidsp@anthropic.com" },
     { name = "Justin Spahr-Summers", email = "justin@anthropic.com" },
 ]
-keywords = ["mcp", "llm", "automation", "pagination", "cursor"]
+keywords = ["gumloop-mcp", "llm", "automation", "pagination", "cursor"]
 license = { text = "MIT" }
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -18,7 +18,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.10",
 ]
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp"]
 
 [project.scripts]
 mcp-simple-pagination = "mcp_simple_pagination.server:main"
diff --git a/examples/servers/simple-prompt/pyproject.toml b/examples/servers/simple-prompt/pyproject.toml
@@ -9,7 +9,7 @@ maintainers = [
     { name = "David Soria Parra", email = "davidsp@anthropic.com" },
     { name = "Justin Spahr-Summers", email = "justin@anthropic.com" },
 ]
-keywords = ["mcp", "llm", "automation", "web", "fetch"]
+keywords = ["gumloop-mcp", "llm", "automation", "web", "fetch"]
 license = { text = "MIT" }
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -18,7 +18,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.10",
 ]
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp"]
 
 [project.scripts]
 mcp-simple-prompt = "mcp_simple_prompt.server:main"
diff --git a/examples/servers/simple-resource/pyproject.toml b/examples/servers/simple-resource/pyproject.toml
@@ -9,7 +9,7 @@ maintainers = [
     { name = "David Soria Parra", email = "davidsp@anthropic.com" },
     { name = "Justin Spahr-Summers", email = "justin@anthropic.com" },
 ]
-keywords = ["mcp", "llm", "automation", "web", "fetch"]
+keywords = ["gumloop-mcp", "llm", "automation", "web", "fetch"]
 license = { text = "MIT" }
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -18,7 +18,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.10",
 ]
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp"]
 
 [project.scripts]
 mcp-simple-resource = "mcp_simple_resource.server:main"
diff --git a/examples/servers/simple-streamablehttp-stateless/pyproject.toml b/examples/servers/simple-streamablehttp-stateless/pyproject.toml
@@ -5,9 +5,9 @@ description = "A simple MCP server exposing a StreamableHttp transport in statel
 readme = "README.md"
 requires-python = ">=3.10"
 authors = [{ name = "Anthropic, PBC." }]
-keywords = ["mcp", "llm", "automation", "web", "fetch", "http", "streamable", "stateless"]
+keywords = ["gumloop-mcp", "llm", "automation", "web", "fetch", "http", "streamable", "stateless"]
 license = { text = "MIT" }
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp", "starlette", "uvicorn"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp", "starlette", "uvicorn"]
 
 [project.scripts]
 mcp-simple-streamablehttp-stateless = "mcp_simple_streamablehttp_stateless.server:main"
diff --git a/examples/servers/simple-streamablehttp/pyproject.toml b/examples/servers/simple-streamablehttp/pyproject.toml
@@ -5,9 +5,9 @@ description = "A simple MCP server exposing a StreamableHttp transport for testi
 readme = "README.md"
 requires-python = ">=3.10"
 authors = [{ name = "Anthropic, PBC." }]
-keywords = ["mcp", "llm", "automation", "web", "fetch", "http", "streamable"]
+keywords = ["gumloop-mcp", "llm", "automation", "web", "fetch", "http", "streamable"]
 license = { text = "MIT" }
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp", "starlette", "uvicorn"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp", "starlette", "uvicorn"]
 
 [project.scripts]
 mcp-simple-streamablehttp = "mcp_simple_streamablehttp.server:main"
diff --git a/examples/servers/simple-tool/pyproject.toml b/examples/servers/simple-tool/pyproject.toml
@@ -9,7 +9,7 @@ maintainers = [
     { name = "David Soria Parra", email = "davidsp@anthropic.com" },
     { name = "Justin Spahr-Summers", email = "justin@anthropic.com" },
 ]
-keywords = ["mcp", "llm", "automation", "web", "fetch"]
+keywords = ["gumloop-mcp", "llm", "automation", "web", "fetch"]
 license = { text = "MIT" }
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -18,7 +18,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.10",
 ]
-dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "mcp"]
+dependencies = ["anyio>=4.5", "click>=8.2.0", "httpx>=0.27", "gumloop-mcp"]
 
 [project.scripts]
 mcp-simple-tool = "mcp_simple_tool.server:main"
diff --git a/examples/snippets/pyproject.toml b/examples/snippets/pyproject.toml
@@ -4,7 +4,7 @@ version = "0.1.0"
 description = "MCP Example Snippets"
 requires-python = ">=3.10"
 dependencies = [
-    "mcp",
+    "gumloop-mcp",
 ]
 
 [build-system]
diff --git a/pyproject.toml b/pyproject.toml
@@ -126,7 +126,7 @@ max-statements = 102  # Default is 50
 members = ["examples/servers/*", "examples/snippets"]
 
 [tool.uv.sources]
-mcp = { workspace = true }
+gumloop-mcp = { workspace = true }
 
 [tool.pytest.ini_options]
 xfail_strict = true
diff --git a/tests/server/lowlevel/test_aggregate_results.py b/tests/server/lowlevel/test_aggregate_results.py
@@ -0,0 +1,56 @@
+"""Tests for Gumloop aggregate_tool_call_results config."""
+
+import json
+
+import pytest
+
+from mcp.server import Server
+from mcp.shared.memory import create_connected_server_and_client_session
+from mcp.types import TextContent, Tool
+
+
+@pytest.mark.anyio
+async def test_aggregate_tool_call_results_enabled():
+    """When aggregate_tool_call_results=True, multiple content items combine into single JSON."""
+    server = Server("test")
+    server.config = {"aggregate_tool_call_results": True}
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="multi", description="Multi", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        return [TextContent(type="text", text="Result 1"), TextContent(type="text", text="Result 2")]
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("multi", {})
+
+    assert result.isError is False
+    assert len(result.content) == 1
+    parsed = json.loads(result.content[0].text)
+    assert len(parsed) == 2
+    assert parsed[0]["text"] == "Result 1"
+    assert parsed[1]["text"] == "Result 2"
+
+
+@pytest.mark.anyio
+async def test_aggregate_tool_call_results_disabled():
+    """When aggregate_tool_call_results not set, multiple content items remain separate."""
+    server = Server("test")
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="multi", description="Multi", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        return [TextContent(type="text", text="Result 1"), TextContent(type="text", text="Result 2")]
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("multi", {})
+
+    assert result.isError is False
+    assert len(result.content) == 2
+    assert result.content[0].text == "Result 1"
+    assert result.content[1].text == "Result 2"
diff --git a/tests/server/lowlevel/test_auth_error.py b/tests/server/lowlevel/test_auth_error.py
@@ -0,0 +1,54 @@
+"""Tests for Gumloop AuthError exception handling."""
+
+import json
+
+import pytest
+
+from mcp.server import Server
+from mcp.shared.exceptions import AuthError
+from mcp.shared.memory import create_connected_server_and_client_session
+from mcp.types import AuthErrorData, TextContent, Tool
+
+
+@pytest.mark.anyio
+async def test_auth_error_returns_structured_data():
+    """AuthError raised in tool should return isError=False with serialized AuthErrorData."""
+    server = Server("test")
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="auth", description="Auth", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        raise AuthError(AuthErrorData(error="credentials_not_found", message="Auth required", service="svc", error_status=401))
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("auth", {})
+
+    assert result.isError is False
+    parsed = json.loads(result.content[0].text)
+    assert parsed["error"] == "credentials_not_found"
+    assert parsed["message"] == "Auth required"
+    assert parsed["service"] == "svc"
+    assert parsed["error_status"] == 401
+
+
+@pytest.mark.anyio
+async def test_regular_exception_returns_is_error_true():
+    """Regular exceptions should return isError=True."""
+    server = Server("test")
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="error", description="Error", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        raise ValueError("Something went wrong")
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("error", {})
+
+    assert result.isError is True
+    assert "Something went wrong" in result.content[0].text
diff --git a/tests/server/lowlevel/test_external_client.py b/tests/server/lowlevel/test_external_client.py
@@ -0,0 +1,77 @@
+"""Tests for Gumloop external_client config."""
+
+import pytest
+
+from mcp.server import Server
+from mcp.shared.memory import create_connected_server_and_client_session
+from mcp.types import TextContent, Tool
+
+
+@pytest.mark.anyio
+async def test_external_client_removes_custom_fields():
+    """external_client=True should remove outputSchema, requiredScopes, creditCost from tools."""
+    server = Server("test")
+    server.config = {"external_client": True}
+
+    @server.list_tools()
+    async def list_tools():
+        tool = Tool(name="tool", description="Tool", inputSchema={"type": "object", "properties": {}})
+        object.__setattr__(tool, "outputSchema", {"type": "object"})
+        object.__setattr__(tool, "requiredScopes", ["read"])
+        object.__setattr__(tool, "creditCost", 10)
+        return [tool]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        return [TextContent(type="text", text="ok")]
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.list_tools()
+
+    assert len(result.tools) == 1
+    assert result.tools[0].outputSchema is None
+    assert not hasattr(result.tools[0], "requiredScopes")
+    assert not hasattr(result.tools[0], "creditCost")
+
+
+@pytest.mark.anyio
+async def test_external_client_empty_result_default_message():
+    """external_client=True with gummie_id and empty result returns default JSON."""
+    server = Server("test")
+    server.config = {"external_client": True, "gummie_id": "123"}
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="empty", description="Empty", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        return []
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("empty", {})
+
+    assert result.isError is False
+    assert len(result.content) == 1
+    assert result.content[0].text == '{"message": "No result found"}'
+
+
+@pytest.mark.anyio
+async def test_external_client_without_gummie_id_no_default():
+    """external_client=True without gummie_id should NOT add default message."""
+    server = Server("test")
+    server.config = {"external_client": True}
+
+    @server.list_tools()
+    async def list_tools():
+        return [Tool(name="empty", description="Empty", inputSchema={"type": "object", "properties": {}})]
+
+    @server.call_tool()
+    async def call_tool(name: str, arguments: dict):
+        return []
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.call_tool("empty", {})
+
+    assert result.isError is False
+    assert len(result.content) == 0
diff --git a/tests/server/lowlevel/test_tool_deprecation.py b/tests/server/lowlevel/test_tool_deprecation.py
@@ -0,0 +1,55 @@
+"""Tests for Gumloop tool deprecation filtering."""
+
+import pytest
+
+from mcp.server import Server
+from mcp.shared.memory import create_connected_server_and_client_session
+from mcp.types import Tool
+
+
+@pytest.mark.anyio
+async def test_deprecated_tools_filtered():
+    """Tools with is_deprecated=True should not appear in list_tools."""
+    server = Server("test")
+
+    @server.list_tools()
+    async def list_tools():
+        active = Tool(name="active", description="Active", inputSchema={"type": "object", "properties": {}})
+        deprecated = Tool(name="deprecated", description="Old", inputSchema={"type": "object", "properties": {}})
+        object.__setattr__(deprecated, "is_deprecated", True)
+        return [active, deprecated]
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.list_tools()
+
+    assert len(result.tools) == 1
+    assert result.tools[0].name == "active"
+
+
+@pytest.mark.anyio
+async def test_deprecated_properties_removed():
+    """Properties with is_deprecated=True in inputSchema should be removed."""
+    server = Server("test")
+
+    @server.list_tools()
+    async def list_tools():
+        return [
+            Tool(
+                name="tool",
+                description="Tool",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "active_param": {"type": "string"},
+                        "old_param": {"type": "string", "is_deprecated": True},
+                    },
+                },
+            )
+        ]
+
+    async with create_connected_server_and_client_session(server) as client:
+        result = await client.list_tools()
+
+    assert len(result.tools) == 1
+    assert "active_param" in result.tools[0].inputSchema["properties"]
+    assert "old_param" not in result.tools[0].inputSchema["properties"]
diff --git a/tests/server/lowlevel/test_tool_restrictions.py b/tests/server/lowlevel/test_tool_restrictions.py
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ version = "0.1.0"`
`4`	`4`	`description = "MCP Example Snippets"`
`5`	`5`	`requires-python = ">=3.10"`
`6`	`6`	`dependencies = [`
`7`		`- "mcp",`
	`7`	`+ "gumloop-mcp",`
`8`	`8`	`]`
`9`	`9`
`10`	`10`	`[build-system]`