fix failed ut

Signed-off-by: Yaliang Wu <[email protected]>
opensearch-project · Oct 3, 2023 · 3063dfe · 3063dfe
1 parent 2948938
commit 3063dfe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/src/test/java/org/opensearch/ml/action/tasks/SearchTaskTransportActionTests.java b/plugin/src/test/java/org/opensearch/ml/action/tasks/SearchTaskTransportActionTests.java
@@ -5,6 +5,8 @@
 
 package org.opensearch.ml.action.tasks;
 
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
@@ -61,6 +63,6 @@ public void setup() {
 
     public void test_DoExecute() {
         searchTaskTransportAction.doExecute(null, searchRequest, actionListener);
-        verify(client).search(searchRequest, actionListener);
+        verify(client).search(eq(searchRequest), any());
     }
 }