refactor: replace lowlevel Server decorators with on_* constructor kwargs (#1985)

tests/server/mcpserver/auth/test_auth_integration.py

@@ -21,7 +21,8 @@ from mcp.server.auth.provider import (
     RefreshToken,
     construct_redirect_uri,
 )
-from mcp.server.auth.routes import ClientRegistrationOptions, RevocationOptions, create_auth_routes
+from mcp.server.auth.routes import create_auth_routes
+from mcp.server.auth.settings import ClientRegistrationOptions, RevocationOptions
 from mcp.shared.auth import OAuthClientInformationFull, OAuthToken
 
 

tests/server/mcpserver/prompts/test_base.py

@@ -2,8 +2,8 @@ from typing import Any
 
 import pytest
 
-from mcp.server.mcpserver.prompts.base import AssistantMessage, Message, Prompt, TextContent, UserMessage
-from mcp.types import EmbeddedResource, TextResourceContents
+from mcp.server.mcpserver.prompts.base import AssistantMessage, Message, Prompt, UserMessage
+from mcp.types import EmbeddedResource, TextContent, TextResourceContents
 
 
 class TestRenderPrompt:

tests/server/mcpserver/prompts/test_manager.py

@@ -1,7 +1,8 @@
 import pytest
 
-from mcp.server.mcpserver.prompts.base import Prompt, TextContent, UserMessage
+from mcp.server.mcpserver.prompts.base import Prompt, UserMessage
 from mcp.server.mcpserver.prompts.manager import PromptManager
+from mcp.types import TextContent
 
 
 class TestPromptManager:

tests/server/mcpserver/test_server.py

@@ -21,6 +21,9 @@ from mcp.shared.exceptions import MCPError
 from mcp.types import (
     AudioContent,
     BlobResourceContents,
+    Completion,
+    CompletionArgument,
+    CompletionContext,
     ContentBlock,
     EmbeddedResource,
     GetPromptResult,
@@ -30,6 +33,7 @@ from mcp.types import (
     Prompt,
     PromptArgument,
     PromptMessage,
+    PromptReference,
     ReadResourceResult,
     Resource,
     ResourceTemplate,
@@ -1401,6 +1405,23 @@ class TestServerPrompts:
                 await client.get_prompt("prompt_fn")
 
 
+async def test_completion_decorator() -> None:
+    """Test that the completion decorator registers a working handler."""
+    mcp = MCPServer()
+
+    @mcp.completion()
+    async def handle_completion(
+        ref: PromptReference, argument: CompletionArgument, context: CompletionContext | None
+    ) -> Completion:
+        assert argument.name == "style"
+        return Completion(values=["bold", "italic", "underline"])
+
+    async with Client(mcp) as client:
+        ref = PromptReference(type="ref/prompt", name="test")
+        result = await client.complete(ref=ref, argument={"name": "style", "value": "b"})
+        assert result.completion.values == ["bold", "italic", "underline"]
+
+
 def test_streamable_http_no_redirect() -> None:
     """Test that streamable HTTP routes are correctly configured."""
     mcp = MCPServer()