Fix sqs worker integration test failure

Signed-off-by: Krishna Kondaka <[email protected]>
kkondaka · Nov 2, 2023 · e8455cf · e8455cf
1 parent 211ac55
commit e8455cf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/...ce/src/integrationTest/java/org/opensearch/dataprepper/plugins/source/s3/SqsWorkerIT.java b/...ce/src/integrationTest/java/org/opensearch/dataprepper/plugins/source/s3/SqsWorkerIT.java
@@ -312,7 +312,7 @@ void processSqsMessages_with_acks_and_progress_check_callbacks(final int numberO
 
         lenient().doAnswer((val) -> {
             if (val.getArgument(0) != null) {
-                delCount.getAndAdd((int)(double)val.getArgument(0));
+                deletedCount += (double)val.getArgument(0);
             }
             return null;
         }).when(deletedCounter).increment(any(Double.class));
@@ -360,7 +360,7 @@ void processSqsMessages_with_acks_and_progress_check_callbacks(final int numberO
         await().atMost(Duration.ofSeconds(20))
                 .untilAsserted(() -> {
                     assertThat(visibilityTimeoutChangedCount, greaterThanOrEqualTo((double)numberOfObjectsToWrite));
-                    assertThat(delCount.get(), equalTo(numberOfObjectsToWrite));
+                    assertThat(deletedCount, equalTo((double)numberOfObjectsToWrite));
                     assertThat(ackCallbackCount, equalTo((double)numberOfObjectsToWrite));
                 });
     }
@@ -387,7 +387,6 @@ void processSqsMessages_with_acks_and_progress_check_callbacks_expires(final int
         lenient().doAnswer((val) -> {
             if (val.getArgument(0) != null) {
                 deletedCount += (double)val.getArgument(0);
-                delCount.getAndAdd((int)(double)val.getArgument(0));
             }
             return null;
         }).when(deletedCounter).increment(any(Double.class));
@@ -429,7 +428,7 @@ void processSqsMessages_with_acks_and_progress_check_callbacks_expires(final int
         await().atMost(Duration.ofSeconds(10))
                 .untilAsserted(() -> {
                     assertThat(visibilityTimeoutChangedCount, greaterThanOrEqualTo((double)numberOfObjectsToWrite));
-                    assertThat(delCount.get(), equalTo(0));
+                    assertThat(deletedCount, equalTo(0.0));
                     assertThat(ackCallbackCount, equalTo(0.0));
                 });