increased wait times in aggregate processor test

Signed-off-by: Kondaka <[email protected]>
kkondaka · Nov 27, 2023 · 31fdce4 · 31fdce4
1 parent a1cbd6f
commit 31fdce4
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/...est/java/org/opensearch/dataprepper/plugins/processor/aggregate/AggregateProcessorIT.java b/...est/java/org/opensearch/dataprepper/plugins/processor/aggregate/AggregateProcessorIT.java
@@ -159,7 +159,7 @@ void aggregateWithNoConcludingGroupsReturnsExpectedResult() throws InterruptedEx
             });
         }
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
 
         assertThat(allThreadsFinished, equalTo(true));
         assertThat(aggregatedResult.size(), equalTo(NUM_UNIQUE_EVENTS_PER_BATCH));
@@ -198,7 +198,7 @@ void aggregateWithConcludingGroupsOnceReturnsExpectedResult() throws Interrupted
             });
         }
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
 
         assertThat(allThreadsFinished, equalTo(true));
         assertThat(aggregatedResult.size(), equalTo(NUM_UNIQUE_EVENTS_PER_BATCH));
@@ -249,7 +249,7 @@ void aggregateWithPutAllActionAndCondition() throws InterruptedException {
             });
         }
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
 
         assertThat(allThreadsFinished, equalTo(true));
         assertThat(aggregatedResult.size(), equalTo(NUM_UNIQUE_EVENTS_PER_BATCH/2));
@@ -320,7 +320,7 @@ void aggregateWithRateLimiterAction() throws InterruptedException {
             });
         }
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
 
         assertThat(allThreadsFinished, equalTo(true));
         // Expect less number of events to be received, because of rate limiting
@@ -355,7 +355,7 @@ void aggregateWithRateLimiterActionNoDrops() throws InterruptedException {
             });
         }
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
 
         assertThat(allThreadsFinished, equalTo(true));
         // Expect all events to be received even with rate limiting because no events are dropped
@@ -385,7 +385,7 @@ void aggregateWithCountAggregateAction() throws InterruptedException, NoSuchFiel
         }
         Thread.sleep(GROUP_DURATION_FOR_ONLY_SINGLE_CONCLUDE * 1000);
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
         assertThat(allThreadsFinished, equalTo(true));
 
         Collection<Record<Event>> results = objectUnderTest.doExecute(new ArrayList<Record<Event>>());
@@ -431,7 +431,7 @@ void aggregateWithCountAggregateActionWithCondition() throws InterruptedExceptio
         }
         Thread.sleep(GROUP_DURATION_FOR_ONLY_SINGLE_CONCLUDE * 1000);
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
         assertThat(allThreadsFinished, equalTo(true));
 
         Collection<Record<Event>> results = objectUnderTest.doExecute(new ArrayList<Record<Event>>());
@@ -491,7 +491,7 @@ void aggregateWithHistogramAggregateAction() throws InterruptedException, NoSuch
         }
         Thread.sleep(GROUP_DURATION_FOR_ONLY_SINGLE_CONCLUDE * 1000);
 
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
         assertThat(allThreadsFinished, equalTo(true));
 
         Collection<Record<Event>> results = objectUnderTest.doExecute(new ArrayList<Record<Event>>());
@@ -545,7 +545,7 @@ void aggregateWithTailSamplerAction(final int testPercent) throws InterruptedExc
             });
         }
         Thread.sleep(GROUP_DURATION_FOR_ONLY_SINGLE_CONCLUDE * 1000);
-        boolean allThreadsFinished = countDownLatch.await(10L, TimeUnit.SECONDS);
+        boolean allThreadsFinished = countDownLatch.await(20L, TimeUnit.SECONDS);
         assertThat(allThreadsFinished, equalTo(true));
         List<Event> errorEventList = eventBatch.stream().map(Record::getData).filter(event -> {
             Event ev = ((Event)event);