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 index 05ad33bec7..fb16be554e 100644 --- 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 {