From a1e417f3503ffa270722914a5147917aba18ca62 Mon Sep 17 00:00:00 2001 From: Ilayaperumal Gopinathan Date: Tue, 11 Mar 2025 14:31:03 +0000 Subject: [PATCH] Fix McpServerAutoConfigurationIT - Fix MCP client mock for client info --- .../autoconfigure/mcp/server/McpServerAutoConfigurationIT.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpServerAutoConfigurationIT.java b/auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpServerAutoConfigurationIT.java index 48abdba9c..3cce88218 100644 --- a/auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpServerAutoConfigurationIT.java +++ b/auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpServerAutoConfigurationIT.java @@ -42,6 +42,7 @@ import java.util.function.Consumer; import java.util.function.Function; import static org.assertj.core.api.Assertions.assertThat; +import static org.mockito.Mockito.when; public class McpServerAutoConfigurationIT { @@ -250,6 +251,7 @@ public class McpServerAutoConfigurationIT { Mockito.when(mockTool.name()).thenReturn("test-tool"); Mockito.when(mockTool.description()).thenReturn("Test Tool"); Mockito.when(mockClient.callTool(Mockito.any(McpSchema.CallToolRequest.class))).thenReturn(mockResult); + when(mockClient.getClientInfo()).thenReturn(new McpSchema.Implementation("testClient", "1.0.0")); return List.of(new SyncMcpToolCallback(mockClient, mockTool)); }