Fix tests by restarting the container in each test

The reason is that the server performs shutdown after SSE connection is
closed.

Author: chemicL

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/AbstractMcpAsyncClientTests.java

@@ -55,9 +55,13 @@ public abstract class AbstractMcpAsyncClientTests {
 
 	abstract protected void createMcpTransport();
 
+	abstract protected void onStart();
+
+	abstract protected void onClose();
+
 	@BeforeEach
 	void setUp() {
-
+		onStart();
 		createMcpTransport();
 
 		assertThatCode(() -> {
@@ -72,6 +76,7 @@ void tearDown() {
 			assertThatCode(() -> mcpAsyncClient.closeGracefully().block(Duration.ofSeconds(10)))
 				.doesNotThrowAnyException();
 		}
+		onClose();
 	}
 
 	@Test

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/AbstractMcpSyncClientTests.java

@@ -55,8 +55,13 @@ public abstract class AbstractMcpSyncClientTests {
 
 	abstract protected void createMcpTransport();
 
+	abstract protected void onStart();
+
+	abstract protected void onClose();
+
 	@BeforeEach
 	void setUp() {
+		onStart();
 		createMcpTransport();
 
 		assertThatCode(() -> {
@@ -70,6 +75,7 @@ void tearDown() {
 		if (mcpSyncClient != null) {
 			assertThatCode(() -> mcpSyncClient.close()).doesNotThrowAnyException();
 		}
+		onClose();
 	}
 
 	@Test

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/sse/SseMcpAsyncClientTests.java

@@ -16,8 +16,6 @@
 
 package org.springframework.ai.mcp.client.sse;
 
-import org.junit.jupiter.api.AfterAll;
-import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Timeout;
 import org.testcontainers.containers.GenericContainer;
 import org.testcontainers.containers.wait.strategy.Wait;
@@ -38,7 +36,7 @@ class SseMcpAsyncClientTests extends AbstractMcpAsyncClientTests {
 
 	// Uses the https://github.com/tzolov/mcp-everything-server-docker-image
 	@SuppressWarnings("resource")
-	static GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
+	GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
 		.withLogConsumer(outputFrame -> System.out.println(outputFrame.getUtf8String()))
 		.withExposedPorts(3001)
 		.waitingFor(Wait.forHttp("/").forStatusCode(404));
@@ -48,15 +46,15 @@ protected void createMcpTransport() {
 		this.mcpTransport = new SseServerTransport(WebClient.builder().baseUrl(host));
 	}
 
-	@BeforeAll
-	static void beforeAll() {
+	@Override
+	protected void onStart() {
 		container.start();
 		int port = container.getMappedPort(3001);
 		host = "http://" + container.getHost() + ":" + port;
 	}
 
-	@AfterAll
-	public static void cleanup() {
+	@Override
+	public void onClose() {
 		container.stop();
 	}
 

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/sse/SseMcpSyncClientTests.java

@@ -16,8 +16,8 @@
 
 package org.springframework.ai.mcp.client.sse;
 
-import org.junit.jupiter.api.AfterAll;
-import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Timeout;
 import org.testcontainers.containers.GenericContainer;
 import org.testcontainers.containers.wait.strategy.Wait;
@@ -38,7 +38,7 @@ class SseMcpSyncClientTests extends AbstractMcpSyncClientTests {
 
 	// Uses the https://github.com/tzolov/mcp-everything-server-docker-image
 	@SuppressWarnings("resource")
-	static GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
+	GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
 		.withLogConsumer(outputFrame -> System.out.println(outputFrame.getUtf8String()))
 		.withExposedPorts(3001)
 		.waitingFor(Wait.forHttp("/").forStatusCode(404));
@@ -48,15 +48,15 @@ protected void createMcpTransport() {
 		this.mcpTransport = new SseServerTransport(WebClient.builder().baseUrl(host));
 	}
 
-	@BeforeAll
-	static void beforeAll() {
+	@Override
+	protected void onStart() {
 		container.start();
 		int port = container.getMappedPort(3001);
 		host = "http://" + container.getHost() + ":" + port;
 	}
 
-	@AfterAll
-	public static void cleanup() {
+	@Override
+	protected void onClose() {
 		container.stop();
 	}
 

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/sse/SseServerTransportTests.java

@@ -51,7 +51,7 @@ class SseServerTransportTests {
 	static String host = "http://localhost:3001";
 
 	@SuppressWarnings("resource")
-	static GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
+	GenericContainer<?> container = new GenericContainer<>("docker.io/tzolov/mcp-everything-server:v1")
 		.withLogConsumer(outputFrame -> System.out.println(outputFrame.getUtf8String()))
 		.withExposedPorts(3001)
 		.waitingFor(Wait.forHttp("/").forStatusCode(404));
@@ -98,15 +98,15 @@ public void simulateMessageEvent(String jsonMessage) {
 
 	}
 
-	@BeforeAll
-	static void beforeAll() {
+	void startContainer() {
 		container.start();
 		int port = container.getMappedPort(3001);
 		host = "http://" + container.getHost() + ":" + port;
 	}
 
 	@BeforeEach
 	void setUp() {
+		startContainer();
 		webClientBuilder = WebClient.builder().baseUrl(host);
 		objectMapper = new ObjectMapper();
 		transport = new TestSseServerTransport(webClientBuilder, objectMapper);
@@ -117,10 +117,10 @@ void afterEach() {
 		if (transport != null) {
 			assertThatCode(() -> transport.closeGracefully().block(Duration.ofSeconds(10))).doesNotThrowAnyException();
 		}
+		cleanup();
 	}
 
-	@AfterAll
-	static void cleanup() {
+	void cleanup() {
 		container.stop();
 	}
 

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/stdio/McpAsyncClientTests.java

@@ -37,4 +37,14 @@ protected void createMcpTransport() {
 		this.mcpTransport = new StdioServerTransport(stdioParams);
 	}
 
+	@Override
+	protected void onStart() {
+
+	}
+
+	@Override
+	protected void onClose() {
+
+	}
+
 }

spring-ai-mcp-core/src/test/java/org/springframework/ai/mcp/client/stdio/McpSyncClientTests.java

@@ -37,4 +37,14 @@ protected void createMcpTransport() {
 		this.mcpTransport = new StdioServerTransport(stdioParams);
 	}
 
+	@Override
+	protected void onStart() {
+
+	}
+
+	@Override
+	protected void onClose() {
+
+	}
+
 }