From e335e18f94090c5a058fdaf4234d9e95c434bbd1 Mon Sep 17 00:00:00 2001 From: Rob Rudin Date: Mon, 21 Aug 2023 15:06:18 -0400 Subject: [PATCH] DEVEXP-155 Added disabled test for setMaxBatches --- .../datamovement/SetMaxBatchesTest.java | 36 +++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 marklogic-client-api-functionaltests/src/test/java/com/marklogic/client/fastfunctest/datamovement/SetMaxBatchesTest.java diff --git a/marklogic-client-api-functionaltests/src/test/java/com/marklogic/client/fastfunctest/datamovement/SetMaxBatchesTest.java b/marklogic-client-api-functionaltests/src/test/java/com/marklogic/client/fastfunctest/datamovement/SetMaxBatchesTest.java new file mode 100644 index 000000000..576f8750f --- /dev/null +++ b/marklogic-client-api-functionaltests/src/test/java/com/marklogic/client/fastfunctest/datamovement/SetMaxBatchesTest.java @@ -0,0 +1,36 @@ +package com.marklogic.client.fastfunctest.datamovement; + +import com.marklogic.client.datamovement.DataMovementManager; +import com.marklogic.client.datamovement.QueryBatcher; +import com.marklogic.client.fastfunctest.AbstractFunctionalTest; +import org.junit.jupiter.api.Disabled; +import org.junit.jupiter.api.Test; + +import java.util.concurrent.atomic.AtomicInteger; + +import static org.junit.jupiter.api.Assertions.assertEquals; + +@Disabled("Getting this test in place to verify that setMaxBatches does not appear to work yet.") +public class SetMaxBatchesTest extends AbstractFunctionalTest { + + @Test + void test() { + writeJsonDocs(50, "max-batches-test"); + + DataMovementManager dmm = client.newDataMovementManager(); + AtomicInteger uriCount = new AtomicInteger(); + QueryBatcher queryBatcher = dmm + .newQueryBatcher(client.newQueryManager().newStructuredQueryBuilder().collection("max-batches-test")) + .withThreadCount(1) + .withBatchSize(10) + .onUrisReady(batch -> uriCount.addAndGet(batch.getItems().length)); + queryBatcher.setMaxBatches(2); + dmm.startJob(queryBatcher); + queryBatcher.awaitCompletion(); + dmm.stopJob(queryBatcher); + + assertEquals(20, uriCount.get(), "Because the batch size is 10 and we asked for 2 batches back, we only " + + "expect 20 URIs back. But through 6.2.2 (and probably going back much further), all URIs are returned. " + + "Modifying the thread count and batch size do not appear to affect this at all."); + } +}