From 9485921ac3166bfe48ddc4275f4c3a4aaea75f85 Mon Sep 17 00:00:00 2001 From: Emanuel Trandafir Date: Fri, 12 Jun 2026 18:04:25 +0300 Subject: [PATCH] fix failing test --- .../mcp/test/ExchangeRateMcpToolSseIntegrationTest.java | 4 +++- .../test/ExchangeRateMcpToolStreamableIntegrationTest.java | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolSseIntegrationTest.java b/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolSseIntegrationTest.java index f4e245b575e6..bee08404f0e4 100644 --- a/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolSseIntegrationTest.java +++ b/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolSseIntegrationTest.java @@ -62,7 +62,9 @@ void whenMcpClientCallTool_thenTheToolReturnsMockedResponse() { .findFirst() .orElseThrow(); - String argumentName = exchangeRateTool.inputSchema().keySet().stream() + String argumentName = ((Map) exchangeRateTool.inputSchema().get("properties")) + .keySet() + .stream() .findFirst() .orElseThrow(); diff --git a/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolStreamableIntegrationTest.java b/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolStreamableIntegrationTest.java index 77058d29c978..9c01404d6eeb 100644 --- a/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolStreamableIntegrationTest.java +++ b/spring-ai-modules/spring-ai-mcp/src/test/java/com/baeldung/springai/mcp/test/ExchangeRateMcpToolStreamableIntegrationTest.java @@ -64,7 +64,9 @@ void whenMcpClientCallTool_thenTheToolReturnsMockedResponse() { .findFirst() .orElseThrow(); - String argumentName = exchangeRateTool.inputSchema().keySet().stream() + String argumentName = ((Map) exchangeRateTool.inputSchema().get("properties")) + .keySet() + .stream() .findFirst() .orElseThrow();