acknowledgments on case

Signed-off-by: Santhosh Gandhe <[email protected]>
opensearch-project · Jan 22, 2025 · 2589f1f · 2589f1f
1 parent bc2449f
commit 2589f1f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...dataprepper/plugins/source/source_crawler/coordination/scheduler/WorkerSchedulerTest.java b/...dataprepper/plugins/source/source_crawler/coordination/scheduler/WorkerSchedulerTest.java
@@ -75,6 +75,7 @@ void testLeaderPartitionsCreation() throws InterruptedException {
         String state = "{\"keyAttributes\":{\"project\":\"project-1\"},\"totalItems\":0,\"loadedItems\":20,\"exportStartTime\":1729391235717,\"itemIds\":[\"GTMS-25\",\"GTMS-24\"]}";
         when(sourcePartitionStoreItem.getPartitionProgressState()).thenReturn(state);
         when(sourcePartitionStoreItem.getSourceIdentifier()).thenReturn(sourceId);
+        when(sourceConfig.isAcknowledgments()).thenReturn(true);
         when(acknowledgementSetManager.create(any(), any())).thenReturn(acknowledgementSet);
         PartitionFactory factory = new PartitionFactory();
         EnhancedSourcePartition sourcePartition = factory.apply(sourcePartitionStoreItem);