Addressed failing test

Signed-off-by: Krishna Kondaka <[email protected]>
opensearch-project · Jul 3, 2023 · fdb7444 · fdb7444
1 parent 141527b
commit fdb7444
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...java/org/opensearch/dataprepper/plugins/kafka/consumer/KafkaSourceCustomConsumerTest.java b/...java/org/opensearch/dataprepper/plugins/kafka/consumer/KafkaSourceCustomConsumerTest.java
@@ -118,7 +118,7 @@ public void setUp() {
 
     public KafkaSourceCustomConsumer createObjectUnderTest(String schemaType, boolean acknowledgementsEnabled) {
         when(sourceConfig.getAcknowledgementsEnabled()).thenReturn(acknowledgementsEnabled);
-        when(sourceConfig.getAcknowledgementsTimeout()).thenReturn(Duration.ofSeconds(5));
+        when(sourceConfig.getAcknowledgementsTimeout()).thenReturn(KafkaSourceConfig.DEFAULT_ACKNOWLEDGEMENTS_TIMEOUT);
         return new KafkaSourceCustomConsumer(kafkaConsumer, shutdownInProgress, buffer, sourceConfig, topicConfig, schemaType, acknowledgementSetManager, pluginMetrics);
     }