Fix Java MCP generator prompt

- Add missing import statements for handler classes
- Add handler registration in test server setup
This commit is contained in:
Troy Taylor 2025-10-16 14:54:02 -04:00
parent 5a493169d0
commit 6ce404ebec

View File

@ -186,6 +186,9 @@ tasks.test {
```java
package com.example.mcp;
import com.example.mcp.tools.ToolHandlers;
import com.example.mcp.resources.ResourceHandlers;
import com.example.mcp.prompts.PromptHandlers;
import io.mcp.server.McpServer;
import io.mcp.server.McpServerBuilder;
import io.mcp.server.transport.StdioServerTransport;
@ -612,10 +615,15 @@ class McpServerTest {
private McpServer createTestServer() {
// Same setup as main application
return McpServerBuilder.builder()
McpServer server = McpServerBuilder.builder()
.serverInfo("test-server", "1.0.0")
.capabilities(cap -> cap.tools(true))
.build();
// Register handlers
ToolHandlers.register(server);
return server;
}
@Test