From 785b7824c64371f4135601e56945d9a59fabf224 Mon Sep 17 00:00:00 2001 From: Kondaka Date: Sun, 15 Sep 2024 08:58:20 -0700 Subject: [PATCH] Fixed build erros Signed-off-by: Kondaka --- .../plugins/mongo/buffer/ExportRecordBufferWriterTest.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/data-prepper-plugins/mongodb/src/test/java/org/opensearch/dataprepper/plugins/mongo/buffer/ExportRecordBufferWriterTest.java b/data-prepper-plugins/mongodb/src/test/java/org/opensearch/dataprepper/plugins/mongo/buffer/ExportRecordBufferWriterTest.java index f1efdbcad4..57fd63681a 100644 --- a/data-prepper-plugins/mongodb/src/test/java/org/opensearch/dataprepper/plugins/mongo/buffer/ExportRecordBufferWriterTest.java +++ b/data-prepper-plugins/mongodb/src/test/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); } -} \ No newline at end of file +}