Skip to content

Commit a1e417f

Browse files
committed
Fix McpServerAutoConfigurationIT
- Fix MCP client mock for client info
1 parent 9571501 commit a1e417f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

auto-configurations/spring-ai-mcp-server/src/test/java/org/springframework/ai/autoconfigure/mcp/server/McpServerAutoConfigurationIT.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@
4242
import java.util.function.Function;
4343

4444
import static org.assertj.core.api.Assertions.assertThat;
45+
import static org.mockito.Mockito.when;
4546

4647
public class McpServerAutoConfigurationIT {
4748

@@ -250,6 +251,7 @@ List<ToolCallback> testTool() {
250251
Mockito.when(mockTool.name()).thenReturn("test-tool");
251252
Mockito.when(mockTool.description()).thenReturn("Test Tool");
252253
Mockito.when(mockClient.callTool(Mockito.any(McpSchema.CallToolRequest.class))).thenReturn(mockResult);
254+
when(mockClient.getClientInfo()).thenReturn(new McpSchema.Implementation("testClient", "1.0.0"));
253255

254256
return List.of(new SyncMcpToolCallback(mockClient, mockTool));
255257
}

0 commit comments

Comments
 (0)