Fixed failing tests

Signed-off-by: Krishna Kondaka <[email protected]>
kkondaka · Oct 18, 2024 · 7bd2d63 · 7bd2d63
1 parent d1d3bfd
commit 7bd2d63
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...t/java/org/opensearch/dataprepper/core/pipeline/router/RouterCopyRecordStrategyTests.java b/...t/java/org/opensearch/dataprepper/core/pipeline/router/RouterCopyRecordStrategyTests.java
@@ -280,10 +280,10 @@ void test_multiple_records_with_acknowledgements_and_multi_components() {
         attachEventHandlesToRecordsIn(eventHandles);
         try {
             doAnswer((i) -> {
-                JacksonEvent e1 = (JacksonEvent) i.getArgument(0);
-                ((DefaultEventHandle)e1.getEventHandle()).addAcknowledgementSet(acknowledgementSet1);
+                EventHandle handle = (EventHandle) i.getArgument(0);
+                ((DefaultEventHandle)handle).addAcknowledgementSet(acknowledgementSet1);
                 return null;
-            }).when(acknowledgementSet1).add(any(JacksonEvent.class));
+            }).when(acknowledgementSet1).add(any(EventHandle.class));
         } catch (Exception e){}
 
         eventBuilder = mock(EventBuilder.class);