adjust test timeout values

Signed-off-by: Christian Tzolov <[email protected]>

Author: tzolov

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/WebFluxSseMcpAsyncClientTests.java

@@ -4,6 +4,8 @@
 
 package io.modelcontextprotocol.client;
 
+import java.time.Duration;
+
 import io.modelcontextprotocol.client.transport.WebFluxSseClientTransport;
 import io.modelcontextprotocol.spec.ClientMcpTransport;
 import org.junit.jupiter.api.Timeout;
@@ -46,4 +48,8 @@ public void onClose() {
 		container.stop();
 	}
 
+	protected Duration getInitializationTimeout() {
+		return Duration.ofSeconds(1);
+	}
+
 }

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/WebFluxSseMcpSyncClientTests.java

@@ -4,6 +4,8 @@
 
 package io.modelcontextprotocol.client;
 
+import java.time.Duration;
+
 import io.modelcontextprotocol.client.transport.WebFluxSseClientTransport;
 import io.modelcontextprotocol.spec.ClientMcpTransport;
 import org.junit.jupiter.api.Timeout;
@@ -46,4 +48,8 @@ protected void onClose() {
 		container.stop();
 	}
 
+	protected Duration getInitializationTimeout() {
+		return Duration.ofSeconds(1);
+	}
+
 }

mcp-test/src/main/java/io/modelcontextprotocol/client/AbstractMcpAsyncClientTests.java

@@ -63,7 +63,7 @@ protected Duration getRequestTimeout() {
 	}
 
 	protected Duration getInitializationTimeout() {
-		return Duration.ofSeconds(1);
+		return Duration.ofSeconds(2);
 	}
 
 	@BeforeEach
@@ -90,10 +90,10 @@ void tearDown() {
 
 	@Test
 	void testConstructorWithInvalidArguments() {
-		assertThatThrownBy(() -> McpClient.sync(null).build()).isInstanceOf(IllegalArgumentException.class)
+		assertThatThrownBy(() -> McpClient.async(null).build()).isInstanceOf(IllegalArgumentException.class)
 			.hasMessage("Transport must not be null");
 
-		assertThatThrownBy(() -> McpClient.sync(mcpTransport).requestTimeout(null).build())
+		assertThatThrownBy(() -> McpClient.async(mcpTransport).requestTimeout(null).build())
 			.isInstanceOf(IllegalArgumentException.class)
 			.hasMessage("Request timeout must not be null");
 	}

mcp-test/src/main/java/io/modelcontextprotocol/client/AbstractMcpSyncClientTests.java

@@ -48,16 +48,18 @@ public abstract class AbstractMcpSyncClientTests {
 
 	abstract protected ClientMcpTransport createMcpTransport();
 
-	abstract protected void onStart();
+	protected void onStart() {
+	}
 
-	abstract protected void onClose();
+	protected void onClose() {
+	}
 
 	protected Duration getRequestTimeout() {
 		return Duration.ofSeconds(10);
 	}
 
 	protected Duration getInitializationTimeout() {
-		return Duration.ofSeconds(1);
+		return Duration.ofSeconds(2);
 	}
 
 	@BeforeEach

mcp/src/test/java/io/modelcontextprotocol/client/AbstractMcpSyncClientTests.java

@@ -49,9 +49,11 @@ public abstract class AbstractMcpSyncClientTests {
 
 	abstract protected ClientMcpTransport createMcpTransport();
 
-	abstract protected void onStart();
+	protected void onStart() {
+	}
 
-	abstract protected void onClose();
+	protected void onClose() {
+	}
 
 	protected Duration getRequestTimeout() {
 		return Duration.ofSeconds(10);

mcp/src/test/java/io/modelcontextprotocol/client/StdioMcpAsyncClientTests.java

@@ -4,6 +4,8 @@
 
 package io.modelcontextprotocol.client;
 
+import java.time.Duration;
+
 import io.modelcontextprotocol.client.transport.ServerParameters;
 import io.modelcontextprotocol.client.transport.StdioClientTransport;
 import io.modelcontextprotocol.spec.ClientMcpTransport;
@@ -26,4 +28,8 @@ protected ClientMcpTransport createMcpTransport() {
 		return new StdioClientTransport(stdioParams);
 	}
 
+	protected Duration getInitializationTimeout() {
+		return Duration.ofSeconds(6);
+	}
+
 }

mcp/src/test/java/io/modelcontextprotocol/client/StdioMcpSyncClientTests.java

@@ -4,6 +4,7 @@
 
 package io.modelcontextprotocol.client;
 
+import java.time.Duration;
 import java.util.concurrent.atomic.AtomicReference;
 
 import io.modelcontextprotocol.client.transport.ServerParameters;
@@ -44,12 +45,8 @@ void customErrorHandlerShouldReceiveErrors() {
 		assertThat(receivedError.get()).isNotNull().isEqualTo(errorMessage);
 	}
 
-	@Override
-	protected void onStart() {
-	}
-
-	@Override
-	protected void onClose() {
+	protected Duration getInitializationTimeout() {
+		return Duration.ofSeconds(6);
 	}
 
 }