apply spotless

Signed-off-by: Jiaping Zeng <[email protected]>

Author: jiapingzeng

ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/agent/AgentUtilsTest.java

@@ -2005,10 +2005,14 @@ private void mockMcpStreamableHttpConnector(MockedStatic<Connector> connectorSta
     @Test
     public void testExtractRequestHeaders_WithValidHeaders() {
         Map<String, String> parameters = new HashMap<>();
-        parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, "{\"Authorization\":\"Bearer token123\",\"Content-Type\":\"application/json\"}");
-        
+        parameters
+            .put(
+                org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS,
+                "{\"Authorization\":\"Bearer token123\",\"Content-Type\":\"application/json\"}"
+            );
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(2, result.size());
         assertEquals("Bearer token123", result.get("Authorization"));
         assertEquals("application/json", result.get("Content-Type"));
@@ -2017,7 +2021,7 @@ public void testExtractRequestHeaders_WithValidHeaders() {
     @Test
     public void testExtractRequestHeaders_WithNullParameters() {
         Map<String, String> result = AgentUtils.extractRequestHeaders(null);
-        
+
         assertEquals(0, result.size());
         assertEquals(Collections.emptyMap(), result);
     }
@@ -2026,9 +2030,9 @@ public void testExtractRequestHeaders_WithNullParameters() {
     public void testExtractRequestHeaders_WithEmptyHeadersJson() {
         Map<String, String> parameters = new HashMap<>();
         parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, "");
-        
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(0, result.size());
         assertEquals(Collections.emptyMap(), result);
     }
@@ -2037,9 +2041,9 @@ public void testExtractRequestHeaders_WithEmptyHeadersJson() {
     public void testExtractRequestHeaders_WithNullHeadersJson() {
         Map<String, String> parameters = new HashMap<>();
         parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, null);
-        
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(0, result.size());
         assertEquals(Collections.emptyMap(), result);
     }
@@ -2048,9 +2052,9 @@ public void testExtractRequestHeaders_WithNullHeadersJson() {
     public void testExtractRequestHeaders_WithWhitespaceHeadersJson() {
         Map<String, String> parameters = new HashMap<>();
         parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, "   ");
-        
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(0, result.size());
         assertEquals(Collections.emptyMap(), result);
     }
@@ -2059,9 +2063,9 @@ public void testExtractRequestHeaders_WithWhitespaceHeadersJson() {
     public void testExtractRequestHeaders_WithInvalidJson() {
         Map<String, String> parameters = new HashMap<>();
         parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, "{invalid json}");
-        
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(0, result.size());
         assertEquals(Collections.emptyMap(), result);
     }
@@ -2070,9 +2074,9 @@ public void testExtractRequestHeaders_WithInvalidJson() {
     public void testExtractRequestHeaders_WithEmptyJsonObject() {
         Map<String, String> parameters = new HashMap<>();
         parameters.put(org.opensearch.ml.common.CommonValue.MCP_REQUEST_HEADERS, "{}");
-        
+
         Map<String, String> result = AgentUtils.extractRequestHeaders(parameters);
-        
+
         assertEquals(0, result.size());
     }
 
@@ -2081,13 +2085,13 @@ public void testMergeHeaders_BothNonEmpty() {
         Map<String, String> staticHeaders = new HashMap<>();
         staticHeaders.put("X-Static-Header", "static-value");
         staticHeaders.put("Authorization", "Bearer static-token");
-        
+
         Map<String, String> requestHeaders = new HashMap<>();
         requestHeaders.put("X-Request-Header", "request-value");
         requestHeaders.put("Authorization", "Bearer request-token");
-        
+
         Map<String, String> result = AgentUtils.mergeHeaders(staticHeaders, requestHeaders);
-        
+
         assertEquals(3, result.size());
         assertEquals("static-value", result.get("X-Static-Header"));
         assertEquals("request-value", result.get("X-Request-Header"));
@@ -2098,9 +2102,9 @@ public void testMergeHeaders_BothNonEmpty() {
     public void testMergeHeaders_OnlyStaticHeaders() {
         Map<String, String> staticHeaders = new HashMap<>();
         staticHeaders.put("X-Static-Header", "static-value");
-        
+
         Map<String, String> result = AgentUtils.mergeHeaders(staticHeaders, null);
-        
+
         assertEquals(1, result.size());
         assertEquals("static-value", result.get("X-Static-Header"));
     }
@@ -2109,34 +2113,34 @@ public void testMergeHeaders_OnlyStaticHeaders() {
     public void testMergeHeaders_OnlyRequestHeaders() {
         Map<String, String> requestHeaders = new HashMap<>();
         requestHeaders.put("X-Request-Header", "request-value");
-        
+
         Map<String, String> result = AgentUtils.mergeHeaders(null, requestHeaders);
-        
+
         assertEquals(1, result.size());
         assertEquals("request-value", result.get("X-Request-Header"));
     }
 
     @Test
     public void testMergeHeaders_BothNull() {
         Map<String, String> result = AgentUtils.mergeHeaders(null, null);
-        
+
         assertEquals(0, result.size());
     }
 
     @Test
     public void testMergeHeaders_BothEmpty() {
         Map<String, String> result = AgentUtils.mergeHeaders(new HashMap<>(), new HashMap<>());
-        
+
         assertEquals(0, result.size());
     }
 
     @Test
     public void testMergeHeaders_EmptyStaticNonEmptyRequest() {
         Map<String, String> requestHeaders = new HashMap<>();
         requestHeaders.put("X-Request-Header", "request-value");
-        
+
         Map<String, String> result = AgentUtils.mergeHeaders(new HashMap<>(), requestHeaders);
-        
+
         assertEquals(1, result.size());
         assertEquals("request-value", result.get("X-Request-Header"));
     }
@@ -2150,7 +2154,7 @@ public void testCleanupMcpClients_WithNullTools() {
     @Test
     public void testCleanupMcpClients_WithEmptyTools() {
         Map<String, Tool> tools = new HashMap<>();
-        
+
         // Should not throw exception
         AgentUtils.cleanupMcpClients(tools);
     }
@@ -2161,7 +2165,7 @@ public void testCleanupMcpClients_WithToolsWithoutMcpClient() {
         Tool tool = mock(Tool.class);
         when(tool.getAttributes()).thenReturn(new HashMap<>());
         tools.put("tool1", tool);
-        
+
         // Should not throw exception
         AgentUtils.cleanupMcpClients(tools);
     }
@@ -2172,7 +2176,7 @@ public void testCleanupMcpClients_WithToolsWithNullAttributes() {
         Tool tool = mock(Tool.class);
         when(tool.getAttributes()).thenReturn(null);
         tools.put("tool1", tool);
-        
+
         // Should not throw exception
         AgentUtils.cleanupMcpClients(tools);
     }
@@ -2182,41 +2186,41 @@ public void testCleanupMcpClients_WithMcpClient() {
         Map<String, Tool> tools = new HashMap<>();
         Tool tool = mock(Tool.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Map<String, Object> attributes = new HashMap<>();
         attributes.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool.getAttributes()).thenReturn(attributes);
         when(tool.getName()).thenReturn("test-tool");
         tools.put("tool1", tool);
-        
+
         AgentUtils.cleanupMcpClients(tools);
-        
+
         verify(mcpClient).closeGracefully();
     }
 
     @Test
     public void testCleanupMcpClients_WithMultipleToolsSameMcpClient() {
         Map<String, Tool> tools = new HashMap<>();
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         // Create two tools sharing the same MCP client
         Tool tool1 = mock(Tool.class);
         Map<String, Object> attributes1 = new HashMap<>();
         attributes1.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool1.getAttributes()).thenReturn(attributes1);
         when(tool1.getName()).thenReturn("test-tool-1");
-        
+
         Tool tool2 = mock(Tool.class);
         Map<String, Object> attributes2 = new HashMap<>();
         attributes2.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool2.getAttributes()).thenReturn(attributes2);
         when(tool2.getName()).thenReturn("test-tool-2");
-        
+
         tools.put("tool1", tool1);
         tools.put("tool2", tool2);
-        
+
         AgentUtils.cleanupMcpClients(tools);
-        
+
         // Should only close once even though two tools share the same client
         verify(mcpClient).closeGracefully();
     }
@@ -2226,24 +2230,24 @@ public void testCleanupMcpClients_WithMultipleToolsDifferentMcpClients() {
         Map<String, Tool> tools = new HashMap<>();
         io.modelcontextprotocol.client.McpSyncClient mcpClient1 = mock(io.modelcontextprotocol.client.McpSyncClient.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient2 = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Tool tool1 = mock(Tool.class);
         Map<String, Object> attributes1 = new HashMap<>();
         attributes1.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient1);
         when(tool1.getAttributes()).thenReturn(attributes1);
         when(tool1.getName()).thenReturn("test-tool-1");
-        
+
         Tool tool2 = mock(Tool.class);
         Map<String, Object> attributes2 = new HashMap<>();
         attributes2.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient2);
         when(tool2.getAttributes()).thenReturn(attributes2);
         when(tool2.getName()).thenReturn("test-tool-2");
-        
+
         tools.put("tool1", tool1);
         tools.put("tool2", tool2);
-        
+
         AgentUtils.cleanupMcpClients(tools);
-        
+
         verify(mcpClient1).closeGracefully();
         verify(mcpClient2).closeGracefully();
     }
@@ -2253,19 +2257,19 @@ public void testCleanupMcpClients_WithExceptionDuringClose() {
         Map<String, Tool> tools = new HashMap<>();
         Tool tool = mock(Tool.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Map<String, Object> attributes = new HashMap<>();
         attributes.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool.getAttributes()).thenReturn(attributes);
         when(tool.getName()).thenReturn("test-tool");
         tools.put("tool1", tool);
-        
+
         // Make closeGracefully throw an exception
         doThrow(new RuntimeException("Close failed")).when(mcpClient).closeGracefully();
-        
+
         // Should not throw exception, just log warning
         AgentUtils.cleanupMcpClients(tools);
-        
+
         verify(mcpClient).closeGracefully();
     }
 
@@ -2274,18 +2278,18 @@ public void testWrapListenerWithMcpCleanup_OnResponse() {
         Map<String, Tool> tools = new HashMap<>();
         Tool tool = mock(Tool.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Map<String, Object> attributes = new HashMap<>();
         attributes.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool.getAttributes()).thenReturn(attributes);
         when(tool.getName()).thenReturn("test-tool");
         tools.put("tool1", tool);
-        
+
         ActionListener<String> delegate = mock(ActionListener.class);
         ActionListener<String> wrapped = AgentUtils.wrapListenerWithMcpCleanup(delegate, tools);
-        
+
         wrapped.onResponse("test-response");
-        
+
         verify(mcpClient).closeGracefully();
         verify(delegate).onResponse("test-response");
     }
@@ -2295,19 +2299,19 @@ public void testWrapListenerWithMcpCleanup_OnFailure() {
         Map<String, Tool> tools = new HashMap<>();
         Tool tool = mock(Tool.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Map<String, Object> attributes = new HashMap<>();
         attributes.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool.getAttributes()).thenReturn(attributes);
         when(tool.getName()).thenReturn("test-tool");
         tools.put("tool1", tool);
-        
+
         ActionListener<String> delegate = mock(ActionListener.class);
         ActionListener<String> wrapped = AgentUtils.wrapListenerWithMcpCleanup(delegate, tools);
-        
+
         Exception testException = new RuntimeException("test exception");
         wrapped.onFailure(testException);
-        
+
         verify(mcpClient).closeGracefully();
         verify(delegate).onFailure(testException);
     }
@@ -2317,21 +2321,21 @@ public void testWrapListenerWithMcpCleanup_CleanupFailureDoesNotPreventDelegateC
         Map<String, Tool> tools = new HashMap<>();
         Tool tool = mock(Tool.class);
         io.modelcontextprotocol.client.McpSyncClient mcpClient = mock(io.modelcontextprotocol.client.McpSyncClient.class);
-        
+
         Map<String, Object> attributes = new HashMap<>();
         attributes.put(org.opensearch.ml.common.CommonValue.MCP_SYNC_CLIENT, mcpClient);
         when(tool.getAttributes()).thenReturn(attributes);
         when(tool.getName()).thenReturn("test-tool");
         tools.put("tool1", tool);
-        
+
         // Make cleanup fail
         doThrow(new RuntimeException("Cleanup failed")).when(mcpClient).closeGracefully();
-        
+
         ActionListener<String> delegate = mock(ActionListener.class);
         ActionListener<String> wrapped = AgentUtils.wrapListenerWithMcpCleanup(delegate, tools);
-        
+
         wrapped.onResponse("test-response");
-        
+
         // Delegate should still be called even if cleanup fails
         verify(delegate).onResponse("test-response");
     }