Add Testcontainers Service Connection support for docker/mcp-gateway

Signed-off-by: Eddú Meléndez <eddu.melendez@gmail.com>

Author: eddumelendez

spring-ai-docs/src/main/antora/modules/ROOT/pages/api/testcontainers.adoc

@@ -38,6 +38,9 @@ The following service connection factories are provided in the `spring-ai-spring
 | `ChromaConnectionDetails`
 | Containers of type `ChromaDBContainer`
 
+| `McpSseClientConnectionDetails`
+| Containers of type `DockerMcpGatewayContainer`
+
 | `MilvusServiceClientConnectionDetails`
 | Containers of type `MilvusContainer`
 

spring-ai-spring-boot-testcontainers/pom.xml

@@ -101,6 +101,20 @@
 <optional>true</optional>
 </dependency>
 
+<dependency>
+<groupId>org.springframework.ai</groupId>
+<artifactId>spring-ai-autoconfigure-mcp-client-httpclient</artifactId>
+<version>${project.version}</version>
+<optional>true</optional>
+</dependency>
+
+<dependency>
+<groupId>org.springframework.ai</groupId>
+<artifactId>spring-ai-autoconfigure-mcp-client-webflux</artifactId>
+<version>${project.version}</version>
+<optional>true</optional>
+</dependency>
+
 <!-- production dependencies -->
 
 <dependency>
@@ -239,6 +253,7 @@
 <dependency>
 <groupId>org.testcontainers</groupId>
 <artifactId>testcontainers</artifactId>
+<version>1.21.3</version>
 <optional>true</optional>
 </dependency>
 

spring-ai-spring-boot-testcontainers/src/main/java/org/springframework/ai/testcontainers/service/connection/docker/DockerMcpGatewayContainerConnectionDetailsFactory.java

@@ -0,0 +1,59 @@
+/*
+ * Copyright 2023-2025 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.ai.testcontainers.service.connection.docker;
+
+import java.util.Map;
+
+import org.testcontainers.containers.DockerMcpGatewayContainer;
+
+import org.springframework.ai.mcp.client.common.autoconfigure.McpSseClientConnectionDetails;
+import org.springframework.ai.mcp.client.common.autoconfigure.properties.McpSseClientProperties;
+import org.springframework.boot.testcontainers.service.connection.ContainerConnectionDetailsFactory;
+import org.springframework.boot.testcontainers.service.connection.ContainerConnectionSource;
+
+/**
+ * @author Eddú Meléndez
+ */
+class DockerMcpGatewayContainerConnectionDetailsFactory
+extends ContainerConnectionDetailsFactory<DockerMcpGatewayContainer, McpSseClientConnectionDetails> {
+
+@Override
+public McpSseClientConnectionDetails getContainerConnectionDetails(
+ContainerConnectionSource<DockerMcpGatewayContainer> source) {
+return new DockerMcpGatewayContainerConnectionDetails(source);
+}
+
+/**
+ * {@link McpSseClientConnectionDetails} backed by a
+ * {@link ContainerConnectionSource}.
+ */
+private static final class DockerMcpGatewayContainerConnectionDetails
+extends ContainerConnectionDetails<DockerMcpGatewayContainer> implements McpSseClientConnectionDetails {
+
+private DockerMcpGatewayContainerConnectionDetails(
+ContainerConnectionSource<DockerMcpGatewayContainer> source) {
+super(source);
+}
+
+@Override
+public Map<String, McpSseClientProperties.SseParameters> getConnections() {
+return Map.of("gateway", new McpSseClientProperties.SseParameters(getContainer().getEndpoint(), "/sse"));
+}
+
+}
+
+}

spring-ai-spring-boot-testcontainers/src/main/resources/META-INF/spring.factories

@@ -15,6 +15,7 @@
 #
 org.springframework.boot.autoconfigure.service.connection.ConnectionDetailsFactory=\
 org.springframework.ai.testcontainers.service.connection.chroma.ChromaContainerConnectionDetailsFactory,\
+org.springframework.ai.testcontainers.service.connection.docker.DockerMcpGatewayContainerConnectionDetailsFactory,\
 org.springframework.ai.testcontainers.service.connection.milvus.MilvusContainerConnectionDetailsFactory,\
 org.springframework.ai.testcontainers.service.connection.mongo.MongoDbAtlasLocalContainerConnectionDetailsFactory,\
 org.springframework.ai.testcontainers.service.connection.ollama.OllamaContainerConnectionDetailsFactory,\

spring-ai-spring-boot-testcontainers/src/test/java/org/springframework/ai/testcontainers/service/connection/docker/DockerMcpGatewayContainerConnectionDetailsFactoryIT.java

@@ -0,0 +1,58 @@
+/*
+ * Copyright 2023-2025 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.ai.testcontainers.service.connection.docker;
+
+import org.junit.jupiter.api.Test;
+import org.testcontainers.containers.DockerMcpGatewayContainer;
+import org.testcontainers.junit.jupiter.Container;
+import org.testcontainers.junit.jupiter.Testcontainers;
+
+import org.springframework.ai.mcp.client.common.autoconfigure.McpSseClientConnectionDetails;
+import org.springframework.ai.mcp.client.common.autoconfigure.properties.McpSseClientProperties;
+import org.springframework.ai.mcp.client.httpclient.autoconfigure.SseHttpClientTransportAutoConfiguration;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.ImportAutoConfiguration;
+import org.springframework.boot.testcontainers.service.connection.ServiceConnection;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+@SpringJUnitConfig
+@Testcontainers
+class DockerMcpGatewayContainerConnectionDetailsFactoryIT {
+
+@Container
+@ServiceConnection
+private static final DockerMcpGatewayContainer MCP_GATEWAY = new DockerMcpGatewayContainer("docker/mcp-gateway:v2");
+
+@Autowired
+private McpSseClientConnectionDetails connectionDetails;
+
+@Test
+void test() {
+assertThat(this.connectionDetails.getConnections()).containsEntry("gateway",
+new McpSseClientProperties.SseParameters(MCP_GATEWAY.getEndpoint(), "/sse"));
+}
+
+@Configuration(proxyBeanMethods = false)
+@ImportAutoConfiguration(SseHttpClientTransportAutoConfiguration.class)
+static class Config {
+
+}
+
+}