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 ```java
package com.example.mcp; 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.McpServer;
import io.mcp.server.McpServerBuilder; import io.mcp.server.McpServerBuilder;
import io.mcp.server.transport.StdioServerTransport; import io.mcp.server.transport.StdioServerTransport;
@ -612,10 +615,15 @@ class McpServerTest {
private McpServer createTestServer() { private McpServer createTestServer() {
// Same setup as main application // Same setup as main application
return McpServerBuilder.builder() McpServer server = McpServerBuilder.builder()
.serverInfo("test-server", "1.0.0") .serverInfo("test-server", "1.0.0")
.capabilities(cap -> cap.tools(true)) .capabilities(cap -> cap.tools(true))
.build(); .build();
// Register handlers
ToolHandlers.register(server);
return server;
} }
@Test @Test