Bump Mockito dependencies (#1711)

Signed-off-by: Andriy Redko <andriy.redko@aiven.io>
opensearch-project · Jan 15, 2024 · dec09d4 · dec09d4
1 parent 0d639f2
commit dec09d4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/src/test/java/org/opensearch/ml/bwc/MLCommonsBackwardsCompatibilityRestTestCase.java b/plugin/src/test/java/org/opensearch/ml/bwc/MLCommonsBackwardsCompatibilityRestTestCase.java
@@ -5,6 +5,7 @@
 
 package org.opensearch.ml.bwc;
 
+import static org.hamcrest.Matchers.equalTo;
 import static org.opensearch.commons.ConfigConstants.OPENSEARCH_SECURITY_SSL_HTTP_ENABLED;
 import static org.opensearch.commons.ConfigConstants.OPENSEARCH_SECURITY_SSL_HTTP_KEYSTORE_FILEPATH;
 import static org.opensearch.commons.ConfigConstants.OPENSEARCH_SECURITY_SSL_HTTP_KEYSTORE_KEYPASSWORD;
@@ -735,7 +736,7 @@ public String getTaskState(String taskId) throws IOException {
         return (String) task.get("state");
     }
 
-    public void waitForTask(String taskId, MLTaskState targetState) throws InterruptedException {
+    public void waitForTask(String taskId, MLTaskState targetState) throws InterruptedException, IOException {
         AtomicBoolean taskDone = new AtomicBoolean(false);
         waitUntil(() -> {
             try {
@@ -748,6 +749,7 @@ public void waitForTask(String taskId, MLTaskState targetState) throws Interrupt
             }
             return taskDone.get();
         }, CUSTOM_MODEL_TIMEOUT, TimeUnit.SECONDS);
+        assertThat(getTaskState(taskId), equalTo(targetState.name()));
         assertTrue(taskDone.get());
     }
 }