update itest

Signed-off-by: bfindlay <bfindlay@acm.org>
opensearch-project · Nov 26, 2023 · 687c546 · 687c546
1 parent aa9f389
commit 687c546
Showing 1 changed file with 28 additions and 31 deletions.
diff --git a/java-client/src/test/java/org/opensearch/client/opensearch/integTest/AbstractTasksIT.java b/java-client/src/test/java/org/opensearch/client/opensearch/integTest/AbstractTasksIT.java
@@ -1,49 +1,46 @@
 package org.opensearch.client.opensearch.integTest;
 
+import java.io.IOException;
 import org.junit.Test;
 import org.opensearch.client.opensearch.tasks.Status;
 
 public abstract class AbstractTasksIT extends OpenSearchJavaClientTestCase {
 
     @Test
-    public void getTasks_waitForCompletionFalse_jsonDataStatusCanBeDeserialized() {
-        try {
-            javaClient().indices().create(b -> b.index("test-index-a").settings(s -> s.refreshInterval(v -> v.time("1s"))));
-            javaClient().index(b -> b.index("test-index-a").id("a").document(new IndexData("test")));
-            javaClient().indices().refresh(b -> b.index("test-index-a"));
+    public void getTasks_waitForCompletionFalse_jsonDataStatusCanBeDeserialized() throws IOException, InterruptedException {
+        javaClient().indices().create(b -> b.index("test-index-a").settings(s -> s.refreshInterval(v -> v.time("1s"))));
+        javaClient().index(b -> b.index("test-index-a").id("a").document(new IndexData("test")));
+        javaClient().indices().refresh(b -> b.index("test-index-a"));
+        Thread.sleep(2000);
 
-            final var deleteByQueryResponse = javaClient().deleteByQuery(
-                d -> d.index("test-index-a").query(q -> q.matchAll(m -> m.queryName("match-all"))).waitForCompletion(false)
-            );
+        final var deleteByQueryResponse = javaClient().deleteByQuery(
+            d -> d.index("test-index-a").query(q -> q.matchAll(m -> m.queryName("match-all"))).waitForCompletion(false)
+        );
 
-            javaClient().indices().refresh(b -> b.index("test-index-a"));
+        javaClient().indices().refresh(b -> b.index("test-index-a"));
 
-            Thread.sleep(2000);
-            final var tasksResponse = javaClient().tasks().get(t -> t.taskId(deleteByQueryResponse.task()));
+        Thread.sleep(3000);
+        final var tasksResponse = javaClient().tasks().get(t -> t.taskId(deleteByQueryResponse.task()));
 
-            assertNotNull(tasksResponse.response());
-            assertTrue(tasksResponse.completed());
+        assertNotNull(tasksResponse.response());
+        assertTrue(tasksResponse.completed());
 
-            // Deserialize the JsonData to a typed Status response
-            final Status taskStatus = tasksResponse.task().status().to(Status.class);
+        // Deserialize the JsonData to a typed Status response
+        final Status taskStatus = tasksResponse.task().status().to(Status.class);
 
-            assertTrue(tasksResponse.completed());
+        assertTrue(tasksResponse.completed());
 
-            // Ensure the JsonData can be deserialized
-            assertEquals(1, taskStatus.total());
-            assertEquals(0, taskStatus.updated());
-            assertEquals(0, taskStatus.created());
-            assertEquals(1, taskStatus.deleted());
-            assertEquals(1, taskStatus.batches());
-            assertEquals(0, taskStatus.noops());
-            assertEquals(0, taskStatus.versionConflicts());
-            assertEquals(0, taskStatus.retries().bulk());
-            assertEquals(0, taskStatus.retries().search());
-            assertEquals(0, taskStatus.throttledMillis());
-            assertEquals(-1f, taskStatus.requestsPerSecond(), 0.01);
-        } catch (Exception e) {
-            fail(e.getMessage());
-        }
+        // Ensure the JsonData can be deserialized
+        assertEquals(1, taskStatus.total());
+        assertEquals(1, taskStatus.deleted());
+        assertEquals(1, taskStatus.batches());
+        assertEquals(1, taskStatus.batches());
+        assertEquals(0, taskStatus.updated());
+        assertEquals(0, taskStatus.created());
+        assertEquals(0, taskStatus.noops());
+        assertEquals(0, taskStatus.versionConflicts());
+        assertEquals(0, taskStatus.retries().bulk());
+        assertEquals(0, taskStatus.retries().search());
     }
 
     public class IndexData {