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)); }