Minor test formating improvements

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

Author: tzolov

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

@@ -286,8 +286,9 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 						""").build())
 				.callHandler((exchange, request) -> {
 
-					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 0.0, 1.0, "tool call start"));
+					var progressToken = request.progressToken();
+
+					exchange.progressNotification(new ProgressNotification(progressToken, 0.0, 1.0, "tool call start"));
 
 					exchange.ping(); // call client ping
 
@@ -301,7 +302,7 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 					ElicitResult elicitationResult = exchange.createElicitation(elicitationRequest);
 
 					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 0.50, 1.0, "elicitation completed"));
+							new ProgressNotification(progressToken, 0.50, 1.0, "elicitation completed"));
 
 					// call sampling
 					var createMessageRequest = McpSchema.CreateMessageRequest.builder()
@@ -317,8 +318,8 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 
 					CreateMessageResult samplingResponse = exchange.createMessage(createMessageRequest);
 
-					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 1.0, 1.0, "sampling completed"));
+					exchange
+						.progressNotification(new ProgressNotification(progressToken, 1.0, 1.0, "sampling completed"));
 
 					return new McpSchema.CallToolResult(List.of(new McpSchema.TextContent(
 							"CALL RESPONSE: " + samplingResponse.toString() + ", " + elicitationResult.toString())),

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

@@ -137,7 +137,6 @@ void clientServerCapabilities() {
 						CallToolRequest toolRequest = CallToolRequest.builder()
 							.name("tool1")
 							.arguments(Map.of())
-							.progressToken("test-progress-token")
 							.build();
 
 						CallToolResult response = mcpClient.callTool(toolRequest);

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

@@ -281,9 +281,9 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 						}
 						""").build())
 				.callHandler((exchange, request) -> {
+					var progressToken = request.progressToken();
 
-					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 0.0, 1.0, "tool call start"));
+					exchange.progressNotification(new ProgressNotification(progressToken, 0.0, 1.0, "tool call start"));
 
 					exchange.ping(); // call client ping
 
@@ -297,7 +297,7 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 					ElicitResult elicitationResult = exchange.createElicitation(elicitationRequest);
 
 					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 0.50, 1.0, "elicitation completed"));
+							new ProgressNotification(progressToken, 0.50, 1.0, "elicitation completed"));
 
 					// call sampling
 					var createMessageRequest = McpSchema.CreateMessageRequest.builder()
@@ -313,8 +313,8 @@ public List<McpServerFeatures.SyncToolSpecification> myTools() {
 
 					CreateMessageResult samplingResponse = exchange.createMessage(createMessageRequest);
 
-					exchange.progressNotification(
-							new ProgressNotification("test-progress-token", 1.0, 1.0, "sampling completed"));
+					exchange
+						.progressNotification(new ProgressNotification(progressToken, 1.0, 1.0, "sampling completed"));
 
 					return new McpSchema.CallToolResult(List.of(new McpSchema.TextContent(
 							"CALL RESPONSE: " + samplingResponse.toString() + ", " + elicitationResult.toString())),