From 7e4873084798223944afff95f93bc11896e4bc88 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edd=C3=BA=20Mel=C3=A9ndez?= Date: Sat, 28 Jun 2025 10:26:15 -0600 Subject: [PATCH] Support docker/mcp-gateway image --- .../containers/DockerMcpGatewayContainer.java | 10 +++++++--- .../containers/DockerMcpGatewayContainerTest.java | 4 ++-- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/core/src/main/java/org/testcontainers/containers/DockerMcpGatewayContainer.java b/core/src/main/java/org/testcontainers/containers/DockerMcpGatewayContainer.java index 592b02cb0ad..8a91f9e651e 100644 --- a/core/src/main/java/org/testcontainers/containers/DockerMcpGatewayContainer.java +++ b/core/src/main/java/org/testcontainers/containers/DockerMcpGatewayContainer.java @@ -20,9 +20,13 @@ */ public class DockerMcpGatewayContainer extends GenericContainer { - private static final String DOCKER_AGENT_GATEWAY_IMAGE = "docker/agents_gateway"; + private static final String DOCKER_MCP_GATEWAY_IMAGE = "docker/mcp-gateway"; - private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse(DOCKER_AGENT_GATEWAY_IMAGE); + private static final String DOCKER_AGENTS_GATEWAY_IMAGE = "docker/agents_gateway"; + + private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse(DOCKER_MCP_GATEWAY_IMAGE); + + private static final DockerImageName DOCKER_AGENTS_IMAGE_NAME = DockerImageName.parse(DOCKER_AGENTS_GATEWAY_IMAGE); private static final int DEFAULT_PORT = 8811; @@ -40,7 +44,7 @@ public DockerMcpGatewayContainer(String dockerImageName) { public DockerMcpGatewayContainer(DockerImageName dockerImageName) { super(dockerImageName); - dockerImageName.assertCompatibleWith(DEFAULT_IMAGE_NAME); + dockerImageName.assertCompatibleWith(DEFAULT_IMAGE_NAME, DOCKER_AGENTS_IMAGE_NAME); withExposedPorts(DEFAULT_PORT); withFileSystemBind(DockerClientFactory.instance().getRemoteDockerUnixSocketPath(), "/var/run/docker.sock"); waitingFor(Wait.forLogMessage(".*Start sse server on port.*", 1)); diff --git a/core/src/test/java/org/testcontainers/containers/DockerMcpGatewayContainerTest.java b/core/src/test/java/org/testcontainers/containers/DockerMcpGatewayContainerTest.java index ab9305cdcfd..28e4cf6fb6a 100644 --- a/core/src/test/java/org/testcontainers/containers/DockerMcpGatewayContainerTest.java +++ b/core/src/test/java/org/testcontainers/containers/DockerMcpGatewayContainerTest.java @@ -10,7 +10,7 @@ public class DockerMcpGatewayContainerTest { @Test public void serviceSuccessfullyStarts() { - try (DockerMcpGatewayContainer gateway = new DockerMcpGatewayContainer("docker/agents_gateway:v2")) { + try (DockerMcpGatewayContainer gateway = new DockerMcpGatewayContainer("docker/mcp-gateway:latest")) { gateway.start(); assertThat(gateway.isRunning()).isTrue(); @@ -21,7 +21,7 @@ public void serviceSuccessfullyStarts() { public void gatewayStartsWithServers() { try ( // container { - DockerMcpGatewayContainer gateway = new DockerMcpGatewayContainer("docker/agents_gateway:v2") + DockerMcpGatewayContainer gateway = new DockerMcpGatewayContainer("docker/mcp-gateway:latest") .withServer("curl", "curl") .withServer("brave", "brave_local_search", "brave_web_search") .withServer("github-official", Collections.singletonList("add_issue_comment"))