Fixed build erros

Signed-off-by: Kondaka <krishkdk@amazon.com>
opensearch-project · Sep 15, 2024 · 785b782 · 785b782
1 parent d402bfe
commit 785b782
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...st/java/org/opensearch/dataprepper/plugins/mongo/buffer/ExportRecordBufferWriterTest.java b/...st/java/org/opensearch/dataprepper/plugins/mongo/buffer/ExportRecordBufferWriterTest.java
@@ -112,7 +112,7 @@ void test_writeSingleRecordToBufferWithAcknowledgementSet() throws Exception {
                 pluginMetrics);
         recordBufferWriter.writeToBuffer(acknowledgementSet, List.of(event));
         verify(bufferAccumulator).add(recordArgumentCaptor.capture());
-        verify(acknowledgementSet).add(event);
+        verify(acknowledgementSet).add(event.getEventHandle());
         assertThat(recordArgumentCaptor.getValue().getData(), equalTo(event));
 
         verify(bufferAccumulator).flush();
@@ -129,10 +129,10 @@ void test_writeSingleRecordFlushException() throws Exception {
 
         recordBufferWriter.writeToBuffer(acknowledgementSet, List.of(event));
         verify(bufferAccumulator).add(recordArgumentCaptor.capture());
-        verify(acknowledgementSet).add(event);
+        verify(acknowledgementSet).add(event.getEventHandle());
         assertThat(recordArgumentCaptor.getValue().getData(), equalTo(event));
 
         verify(bufferAccumulator).flush();
         verify(exportRecordErrors).increment(1);
     }
-}
+}