Merge branch 'StreamableHttpServerTransportProvider' of https://github.com/ZachGerman/mcp-java-sdk into StreamableHttpServerTransportProvider

Author: Zachary German

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/server/transport/StreamableHttpServerTransportProviderIntegrationTests.java

@@ -0,0 +1,125 @@
+/*
+ * Copyright 2024-2024 the original author or authors.
+ */
+
+package io.modelcontextprotocol.server.transport;
+
+import java.util.Map;
+import java.util.List;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import io.modelcontextprotocol.client.McpClient;
+import io.modelcontextprotocol.client.transport.WebClientStreamableHttpTransport;
+import io.modelcontextprotocol.server.McpServer;
+import io.modelcontextprotocol.server.McpServerFeatures;
+import io.modelcontextprotocol.spec.McpSchema;
+import io.modelcontextprotocol.spec.McpSchema.CallToolResult;
+import io.modelcontextprotocol.spec.McpSchema.InitializeResult;
+import org.apache.catalina.LifecycleException;
+import org.apache.catalina.LifecycleState;
+import org.apache.catalina.startup.Tomcat;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.springframework.web.reactive.function.client.WebClient;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+/**
+ * Integration tests for {@link StreamableHttpServerTransportProvider} with
+ * {@link WebClientStreamableHttpTransport}.
+ */
+class StreamableHttpServerTransportProviderIntegrationTests {
+
+	private static final int PORT = TomcatTestUtil.findAvailablePort();
+
+	private static final String ENDPOINT = "/mcp";
+
+	private StreamableHttpServerTransportProvider serverTransportProvider;
+
+	private McpClient.SyncSpec clientBuilder;
+
+	private Tomcat tomcat;
+
+	@BeforeEach
+	void setUp() {
+		serverTransportProvider = new StreamableHttpServerTransportProvider(new ObjectMapper(), ENDPOINT, null);
+
+		tomcat = TomcatTestUtil.createTomcatServer("", PORT, serverTransportProvider);
+		try {
+			tomcat.start();
+			assertThat(tomcat.getServer().getState()).isEqualTo(LifecycleState.STARTED);
+		}
+		catch (Exception e) {
+			throw new RuntimeException("Failed to start Tomcat", e);
+		}
+
+		WebClientStreamableHttpTransport clientTransport = WebClientStreamableHttpTransport
+			.builder(WebClient.builder().baseUrl("http://localhost:" + PORT))
+			.endpoint(ENDPOINT)
+			.objectMapper(new ObjectMapper())
+			.build();
+
+		clientBuilder = McpClient.sync(clientTransport)
+			.clientInfo(new McpSchema.Implementation("Test Client", "1.0.0"));
+	}
+
+	@AfterEach
+	void tearDown() {
+		if (serverTransportProvider != null) {
+			serverTransportProvider.closeGracefully().block();
+		}
+		if (tomcat != null) {
+			try {
+				tomcat.stop();
+				tomcat.destroy();
+			}
+			catch (LifecycleException e) {
+				throw new RuntimeException("Failed to stop Tomcat", e);
+			}
+		}
+	}
+
+	@Test
+	void shouldInitializeSuccessfully() {
+		var mcpServer = McpServer.sync(serverTransportProvider).serverInfo("Test Server", "1.0.0").build();
+
+		try (var mcpClient = clientBuilder.build()) {
+			InitializeResult result = mcpClient.initialize();
+			assertThat(result).isNotNull();
+			assertThat(result.serverInfo().name()).isEqualTo("Test Server");
+		}
+
+		mcpServer.close();
+	}
+
+	@Test
+	void shouldCallToolSuccessfully() {
+		String emptyJsonSchema = """
+				{
+					"$schema": "http://json-schema.org/draft-07/schema#",
+					"type": "object",
+					"properties": {}
+				}
+				""";
+
+		var callResponse = new CallToolResult(List.of(new McpSchema.TextContent("Tool executed successfully")), null);
+		McpServerFeatures.SyncToolSpecification tool = new McpServerFeatures.SyncToolSpecification(
+				new McpSchema.Tool("test-tool", "Test tool description", emptyJsonSchema),
+				(exchange, request) -> callResponse);
+
+		var mcpServer = McpServer.sync(serverTransportProvider).serverInfo("Test Server", "1.0.0").tools(tool).build();
+
+		try (var mcpClient = clientBuilder.build()) {
+			mcpClient.initialize();
+
+			CallToolResult response = mcpClient.callTool(new McpSchema.CallToolRequest("test-tool", Map.of()));
+
+			assertThat(response).isNotNull();
+			assertThat(response).isEqualTo(callResponse);
+		}
+
+		mcpServer.close();
+	}
+
+}