feat(tests): Add unit tests for MCP server, resources, prompts, and tools

Author: codeboyzhou

src/test/java/com/github/codeboyzhou/mcp/declarative/McpServersTest.java

@@ -0,0 +1,78 @@
+package com.github.codeboyzhou.mcp.declarative;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import com.github.codeboyzhou.mcp.declarative.test.TestSimpleMcpStdioServer;
+import io.modelcontextprotocol.client.McpClient;
+import io.modelcontextprotocol.client.McpSyncClient;
+import io.modelcontextprotocol.client.transport.ServerParameters;
+import io.modelcontextprotocol.client.transport.StdioClientTransport;
+import io.modelcontextprotocol.spec.McpSchema;
+import java.util.List;
+import org.junit.jupiter.api.Test;
+
+class McpServersTest {
+
+  String classpath = System.getProperty("java.class.path");
+
+  ServerParameters stdioServerParameters =
+      ServerParameters.builder("java")
+          .args("-cp", classpath, TestSimpleMcpStdioServer.class.getName())
+          .build();
+
+  StdioClientTransport stdioClientTransport = new StdioClientTransport(stdioServerParameters);
+
+  @Test
+  void testStartStdioServer_shouldSucceed() {
+    try (McpSyncClient client = McpClient.sync(stdioClientTransport).build()) {
+      McpSchema.InitializeResult initialized = client.initialize();
+      assertEquals("mcp-server", initialized.serverInfo().name());
+      assertEquals("1.0.0", initialized.serverInfo().version());
+      assertEquals("test", initialized.instructions());
+    }
+  }
+
+  @Test
+  void testStartStdioServer_shouldRegisteredResources() {
+    try (McpSyncClient client = McpClient.sync(stdioClientTransport).build()) {
+      client.initialize();
+      List<McpSchema.Resource> resources = client.listResources().resources();
+      assertEquals(1, resources.size());
+      McpSchema.Resource resource = resources.get(0);
+      assertEquals("test://resource1", resource.uri());
+      assertEquals("resource1_name", resource.name());
+      assertEquals("resource1_title", resource.title());
+      assertEquals("resource1_description", resource.description());
+      assertEquals("text/plain", resource.mimeType());
+    }
+  }
+
+  @Test
+  void testStartStdioServer_shouldRegisteredPrompts() {
+    try (McpSyncClient client = McpClient.sync(stdioClientTransport).build()) {
+      client.initialize();
+      List<McpSchema.Prompt> prompts = client.listPrompts().prompts();
+      assertEquals(1, prompts.size());
+      McpSchema.Prompt prompt = prompts.get(0);
+      assertEquals("prompt1_name", prompt.name());
+      assertEquals("prompt1_title", prompt.title());
+      assertEquals("prompt1_description", prompt.description());
+      assertTrue(prompt.arguments().isEmpty());
+    }
+  }
+
+  @Test
+  void testStartStdioServer_shouldRegisteredTools() {
+    try (McpSyncClient client = McpClient.sync(stdioClientTransport).build()) {
+      client.initialize();
+      List<McpSchema.Tool> tools = client.listTools().tools();
+      assertEquals(1, tools.size());
+      McpSchema.Tool tool = tools.get(0);
+      assertEquals("tool1_name", tool.name());
+      assertEquals("tool1_title", tool.title());
+      assertEquals("tool1_description", tool.description());
+      assertTrue(tool.inputSchema().properties().isEmpty());
+    }
+  }
+}

src/test/java/com/github/codeboyzhou/mcp/declarative/test/TestMcpPrompts.java

@@ -0,0 +1,13 @@
+package com.github.codeboyzhou.mcp.declarative.test;
+
+import com.github.codeboyzhou.mcp.declarative.annotation.McpPrompt;
+import com.github.codeboyzhou.mcp.declarative.annotation.McpPrompts;
+
+@McpPrompts
+public class TestMcpPrompts {
+
+  @McpPrompt(name = "prompt1_name", title = "prompt1_title", description = "prompt1_description")
+  public String prompt1() {
+    return "prompt1_content";
+  }
+}

src/test/java/com/github/codeboyzhou/mcp/declarative/test/TestMcpResources.java

@@ -0,0 +1,17 @@
+package com.github.codeboyzhou.mcp.declarative.test;
+
+import com.github.codeboyzhou.mcp.declarative.annotation.McpResource;
+import com.github.codeboyzhou.mcp.declarative.annotation.McpResources;
+
+@McpResources
+public class TestMcpResources {
+
+  @McpResource(
+      uri = "test://resource1",
+      name = "resource1_name",
+      title = "resource1_title",
+      description = "resource1_description")
+  public String resource1() {
+    return "resource1_content";
+  }
+}

src/test/java/com/github/codeboyzhou/mcp/declarative/test/TestMcpTools.java

@@ -0,0 +1,13 @@
+package com.github.codeboyzhou.mcp.declarative.test;
+
+import com.github.codeboyzhou.mcp.declarative.annotation.McpTool;
+import com.github.codeboyzhou.mcp.declarative.annotation.McpTools;
+
+@McpTools
+public class TestMcpTools {
+
+  @McpTool(name = "tool1_name", title = "tool1_title", description = "tool1_description")
+  public String tool1() {
+    return "tool1_content";
+  }
+}

src/test/java/com/github/codeboyzhou/mcp/declarative/test/TestSimpleMcpStdioServer.java

@@ -0,0 +1,19 @@
+package com.github.codeboyzhou.mcp.declarative.test;
+
+import com.github.codeboyzhou.mcp.declarative.McpServers;
+import com.github.codeboyzhou.mcp.declarative.server.simple.SimpleMcpServerBaseInfo;
+import java.time.Duration;
+
+public class TestSimpleMcpStdioServer {
+
+  public static void main(String[] args) {
+    SimpleMcpServerBaseInfo info =
+        SimpleMcpServerBaseInfo.builder()
+            .name("mcp-server")
+            .version("1.0.0")
+            .instructions("test")
+            .requestTimeout(Duration.ofSeconds(10))
+            .build();
+    McpServers.run(TestSimpleMcpStdioServer.class, args).startStdioServer(info);
+  }
+}

src/test/resources/logback.xml

@@ -4,6 +4,7 @@
     <property name="PATTERN" value="%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n"/>
 
     <appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender">
+        <target>System.err</target>
         <encoder>
             <pattern>${PATTERN}</pattern>
             <charset>UTF-8</charset>