Adding few changes for setMaxUris and in the unit test.

Author: anu3990

marklogic-client-api-functionaltests/src/test/java/com/marklogic/client/datamovement/functionaltests/QueryBatcherJobReportTest.java

@@ -714,10 +714,9 @@ public void run() {
 				}			
 			}
 
-			JobTicket jobTicket = dmManager.startJob(qb);
+			dmManager.startJob(qb);
 
-			Thread tMBStop;
-			tMBStop = new Thread(new MaxBatchesThread());
+			Thread tMBStop = new Thread(new MaxBatchesThread());
 
 			tMBStop.start();
 			tMBStop.join();
@@ -735,7 +734,7 @@ public void run() {
 			Collection<String> batchResults2 = new LinkedHashSet<String>();
 			QueryBatcher qb2 = dmManager.newQueryBatcher(urisList.iterator())
 					.withBatchSize(12)
-					.withThreadCount(1)
+					.withThreadCount(20)
 					.withJobId("ListenerCompletionTest2")	            
 					.onUrisReady((QueryBatch batch) -> {
 
@@ -747,25 +746,32 @@ public void run() {
 					.onQueryFailure(throwable-> {
 						failedBatch2.addAndGet(1);                
 					});
-			
+			qb2.setMaxBatches(203);
 			class BatchesSoFarThread implements Runnable {
 
 				@Override
 				public void run() {
 					// Test 2
+					try {
+						Thread.sleep(1000);
+					} catch (InterruptedException e) {
+						e.printStackTrace();
+					}
 					qb2.setMaxBatches();					
 				}			
 			}
 
-			JobTicket jobTicket2 = dmManager.startJob(qb2);
-
-			Thread tMBStop2;
-			tMBStop2 = new Thread(new BatchesSoFarThread());
+			dmManager.startJob(qb2);
 
+			Thread tMBStop2 = new Thread(new BatchesSoFarThread());
+			int initialUrisSize = batchResults2.size();
+			
 			tMBStop2.start();
-			tMBStop2.join();
-			System.out.println("URI size so far is : " + batchResults2.size());
-			assertTrue("Batches of URIs collected so far", batchResults.size() > 0);
+			qb2.awaitCompletion();
+			dmManager.stopJob(qb2);
+			
+			assertTrue("Batches of URIs collected so far", batchResults2.size() > 0);
+			assertTrue("Number of Uris collected does not fall in the range", (batchResults2.size()>initialUrisSize && batchResults2.size()< 50000));
 		}
 		catch (Exception ex) {
 			ex.printStackTrace();

marklogic-client-api/src/main/java/com/marklogic/client/datamovement/impl/QueryBatcherImpl.java

@@ -1053,6 +1053,7 @@ public long getMaxBatches() {
 @Override
 public void setMaxBatches() {
 	this.maxBatches = -1L;
-	
+	if(isStarted())
+		setMaxUris(getMaxBatches());
 }
 }