Addressed PR comments

Signed-off-by: Owais Kazi <[email protected]>
opensearch-project · Oct 5, 2023 · 9d4a996 · 9d4a996
1 parent 5576712
commit 9d4a996
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/client/src/test/java/org/opensearch/ml/client/MachineLearningNodeClientTest.java b/client/src/test/java/org/opensearch/ml/client/MachineLearningNodeClientTest.java
@@ -74,9 +74,7 @@
 import org.opensearch.search.suggest.Suggest;
 
 import java.io.IOException;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Answers.RETURNS_DEEP_STUBS;
@@ -650,7 +648,8 @@ public void createConnector() {
         ArgumentCaptor<MLCreateConnectorResponse> argumentCaptor = ArgumentCaptor.forClass(MLCreateConnectorResponse.class);
 
         Map<String, String> params = Map.ofEntries(Map.entry("endpoint", "endpoint"), Map.entry("temp", "7"));
-        Map<String, String> credentials = Map.ofEntries(Map.entry("key1", "key1"), Map.entry("key2", "key2"));
+        Map<String, String> credentials = Map.ofEntries(Map.entry("key1", "value1"), Map.entry("key2", "value2"));
+        List<String> backendRoles = Arrays.asList("IT", "HR");
 
         MLCreateConnectorInput mlCreateConnectorInput = MLCreateConnectorInput.builder()
                 .name("test")
@@ -660,7 +659,7 @@ public void createConnector() {
                 .parameters(params)
                 .credential(credentials)
                 .actions(null)
-                .backendRoles(null)
+                .backendRoles(backendRoles)
                 .addAllBackendRoles(false)
                 .access(AccessMode.from("private"))
                 .dryRun(false)