Add test_multiple_pages_async.

fauna · Sep 20, 2024 · da788b1 · da788b1
1 parent 65df6a1
commit da788b1
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/test/java/com/fauna/client/PageIteratorTest.java b/src/test/java/com/fauna/client/PageIteratorTest.java
@@ -19,6 +19,7 @@
 
 import java.io.IOException;
 import java.lang.reflect.Type;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.NoSuchElementException;
 import java.util.concurrent.CompletableFuture;
@@ -96,6 +97,22 @@ public void test_multiple_pages() throws Exception {
         assertThrows(NoSuchElementException.class, () -> pageIterator.next());
     }
 
+    @Test
+    public void test_multiple_pages_async() throws Exception {
+        when(client.asyncQuery(any(), any(ParameterizedOf.class), any())).thenReturn(
+                successFuture(true, 0), successFuture(false, 1));
+        PageIterator<String> pageIterator = new PageIterator<>(client, fql("hello"), String.class, null);
+
+        boolean hasNext = pageIterator.hasNext();
+        List<String> products = new ArrayList<>();
+        while (hasNext) {
+            hasNext = pageIterator.nextAsync().thenApply(page -> {
+                products.addAll(page.getData());
+                return pageIterator.hasNext(); }).get();
+        }
+        assertEquals(4, products.size());
+    }
+
     @Test
     public void test_error_thrown() throws IOException {
         when(client.asyncQuery(any(), any(ParameterizedOf.class), any())).thenReturn(failureFuture());