Implementation of SEP 973 - Additional metadata + icons support (#1357)

tests/issues/test_1338_icons_and_metadata.py

@@ -0,0 +1,126 @@
+"""Test icon and metadata support (SEP-973)."""
+
+import pytest
+
+from mcp.server.fastmcp import FastMCP
+from mcp.types import Icon
+
+pytestmark = pytest.mark.anyio
+
+
+async def test_icons_and_website_url():
+    """Test that icons and websiteUrl are properly returned in API calls."""
+
+    # Create test icon
+    test_icon = Icon(
+        src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==",
+        mimeType="image/png",
+        sizes="1x1",
+    )
+
+    # Create server with website URL and icon
+    mcp = FastMCP("TestServer", website_url="https://example.com", icons=[test_icon])
+
+    # Create tool with icon
+    @mcp.tool(icons=[test_icon])
+    def test_tool(message: str) -> str:
+        """A test tool with an icon."""
+        return message
+
+    # Create resource with icon
+    @mcp.resource("test://resource", icons=[test_icon])
+    def test_resource() -> str:
+        """A test resource with an icon."""
+        return "test content"
+
+    # Create prompt with icon
+    @mcp.prompt("test_prompt", icons=[test_icon])
+    def test_prompt(text: str) -> str:
+        """A test prompt with an icon."""
+        return text
+
+    # Test server metadata includes websiteUrl and icons
+    assert mcp.name == "TestServer"
+    assert mcp.website_url == "https://example.com"
+    assert mcp.icons is not None
+    assert len(mcp.icons) == 1
+    assert mcp.icons[0].src == test_icon.src
+    assert mcp.icons[0].mimeType == test_icon.mimeType
+    assert mcp.icons[0].sizes == test_icon.sizes
+
+    # Test tool includes icon
+    tools = await mcp.list_tools()
+    assert len(tools) == 1
+    tool = tools[0]
+    assert tool.name == "test_tool"
+    assert tool.icons is not None
+    assert len(tool.icons) == 1
+    assert tool.icons[0].src == test_icon.src
+
+    # Test resource includes icon
+    resources = await mcp.list_resources()
+    assert len(resources) == 1
+    resource = resources[0]
+    assert str(resource.uri) == "test://resource"
+    assert resource.icons is not None
+    assert len(resource.icons) == 1
+    assert resource.icons[0].src == test_icon.src
+
+    # Test prompt includes icon
+    prompts = await mcp.list_prompts()
+    assert len(prompts) == 1
+    prompt = prompts[0]
+    assert prompt.name == "test_prompt"
+    assert prompt.icons is not None
+    assert len(prompt.icons) == 1
+    assert prompt.icons[0].src == test_icon.src
+
+
+async def test_multiple_icons():
+    """Test that multiple icons can be added to tools, resources, and prompts."""
+
+    # Create multiple test icons
+    icon1 = Icon(src="data:image/png;base64,icon1", mimeType="image/png", sizes="16x16")
+    icon2 = Icon(src="data:image/png;base64,icon2", mimeType="image/png", sizes="32x32")
+    icon3 = Icon(src="data:image/png;base64,icon3", mimeType="image/png", sizes="64x64")
+
+    mcp = FastMCP("MultiIconServer")
+
+    # Create tool with multiple icons
+    @mcp.tool(icons=[icon1, icon2, icon3])
+    def multi_icon_tool() -> str:
+        """A tool with multiple icons."""
+        return "success"
+
+    # Test tool has all icons
+    tools = await mcp.list_tools()
+    assert len(tools) == 1
+    tool = tools[0]
+    assert tool.icons is not None
+    assert len(tool.icons) == 3
+    assert tool.icons[0].sizes == "16x16"
+    assert tool.icons[1].sizes == "32x32"
+    assert tool.icons[2].sizes == "64x64"
+
+
+async def test_no_icons_or_website():
+    """Test that server works without icons or websiteUrl."""
+
+    mcp = FastMCP("BasicServer")
+
+    @mcp.tool()
+    def basic_tool() -> str:
+        """A basic tool without icons."""
+        return "success"
+
+    # Test server metadata has no websiteUrl or icons
+    assert mcp.name == "BasicServer"
+    assert mcp.website_url is None
+    assert mcp.icons is None
+
+    # Test tool has no icons
+    tools = await mcp.list_tools()
+    assert len(tools) == 1
+    tool = tools[0]
+    assert tool.name == "basic_tool"
+    assert tool.icons is None