Fixed build erros

Signed-off-by: Kondaka <krishkdk@amazon.com>
opensearch-project · Sep 16, 2024 · e1f8fcb · e1f8fcb
1 parent 785b782
commit e1f8fcb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...rc/test/java/org/opensearch/dataprepper/plugins/source/rds/export/DataFileLoaderTest.java b/...rc/test/java/org/opensearch/dataprepper/plugins/source/rds/export/DataFileLoaderTest.java
@@ -148,7 +148,7 @@ void test_run_success() throws Exception {
 
         verify(bufferAccumulator).add(any(Record.class));
         verify(bufferAccumulator).flush();
-        verify(acknowledgementSet).add(event);
+        verify(acknowledgementSet).add(event.getEventHandle());
         verify(acknowledgementSet).complete();
 
         verify(exportRecordsTotalCounter).increment();
@@ -198,7 +198,7 @@ void test_flush_failure_then_error_metric_updated() throws Exception {
 
         verify(bufferAccumulator).add(any(Record.class));
         verify(bufferAccumulator).flush();
-        verify(acknowledgementSet).add(event);
+        verify(acknowledgementSet).add(event.getEventHandle());
         verify(acknowledgementSet, never()).complete();
 
         verify(exportRecordsTotalCounter).increment();
@@ -213,4 +213,4 @@ private DataFileLoader createObjectUnderTest() {
         return DataFileLoader.create(dataFilePartition, codec, buffer, s3ObjectReader, recordConverter,
                 pluginMetrics, sourceCoordinator, acknowledgementSet, acknowledgmentTimeout);
     }
-}
+}