From 2589f1fc9ff78fd401b2a07ada9e9a118d7192ac Mon Sep 17 00:00:00 2001
From: Santhosh Gandhe <1909520+san81@users.noreply.github.com>
Date: Tue, 21 Jan 2025 19:02:19 -0800
Subject: [PATCH] acknowledgments on case

Signed-off-by: Santhosh Gandhe <1909520+san81@users.noreply.github.com>
---
 .../coordination/scheduler/WorkerSchedulerTest.java              | 1 +
 1 file changed, 1 insertion(+)

diff --git a/data-prepper-plugins/saas-source-plugins/source-crawler/src/test/java/org/opensearch/dataprepper/plugins/source/source_crawler/coordination/scheduler/WorkerSchedulerTest.java b/data-prepper-plugins/saas-source-plugins/source-crawler/src/test/java/org/opensearch/dataprepper/plugins/source/source_crawler/coordination/scheduler/WorkerSchedulerTest.java
index 884d9c5b82..9220fa6e80 100644
--- a/data-prepper-plugins/saas-source-plugins/source-crawler/src/test/java/org/opensearch/dataprepper/plugins/source/source_crawler/coordination/scheduler/WorkerSchedulerTest.java
+++ b/data-prepper-plugins/saas-source-plugins/source-crawler/src/test/java/org/opensearch/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);