Fixing tests

Signed-off-by: Aditya Khera <kheraadi@amazon.com>

Author: Aditya Khera

server/src/test/java/org/opensearch/index/engine/MergeRateLimitingTests.java

@@ -19,6 +19,7 @@
 import org.opensearch.common.settings.Settings;
 import org.opensearch.core.index.shard.ShardId;
 import org.opensearch.index.IndexSettings;
+import org.opensearch.index.merge.MergedSegmentTransferTracker;
 import org.opensearch.test.OpenSearchTestCase;
 
 import static org.opensearch.cluster.metadata.IndexMetadata.SETTING_NUMBER_OF_REPLICAS;
@@ -77,15 +78,15 @@ public void testSettingPrecedence() {
         IndexSettings indexSettings = new IndexSettings(newIndexMeta("test_index", indexBuilder.build()), nodeSettings);
         ShardId shardId = new ShardId("test_index", "test_uuid", 0);
 
-        OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings);
+        OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings, new MergedSegmentTransferTracker());
 
         // Should use cluster-level setting
         assertThat(scheduler.getForceMergeMBPerSec(), equalTo(75.0));
 
         // Test with both index and cluster-level settings - index should take precedence
         indexBuilder.put(MAX_FORCE_MERGE_MB_PER_SEC_SETTING.getKey(), "25.0");
         indexSettings = new IndexSettings(newIndexMeta("test_index", indexBuilder.build()), nodeSettings);
-        scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings);
+        scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings, new MergedSegmentTransferTracker());
 
         // Should use index-level setting
         assertThat(scheduler.getForceMergeMBPerSec(), equalTo(25.0));
@@ -104,7 +105,7 @@ public void testDisabledRateLimiting() {
         IndexSettings indexSettings = new IndexSettings(newIndexMeta("test_index", builder.build()), Settings.EMPTY);
         ShardId shardId = new ShardId("test_index", "test_uuid", 0);
 
-        OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings);
+        OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings, new MergedSegmentTransferTracker());
 
         // Should have no rate limiting
         assertThat(scheduler.getForceMergeMBPerSec(), equalTo(Double.POSITIVE_INFINITY));
@@ -130,7 +131,7 @@ public void testDynamicRateLimitUpdates() throws Exception {
             IndexSettings indexSettings = new IndexSettings(newIndexMeta("test_index", builder.build()), Settings.EMPTY);
             ShardId shardId = new ShardId("test_index", "test_uuid", 0);
 
-            OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings);
+            OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings, new MergedSegmentTransferTracker());
             assertThat(scheduler.getForceMergeMBPerSec(), equalTo(10.0));
 
             // Update to a different rate limit
@@ -190,7 +191,7 @@ public void testFallbackToClusterSettingWhenIndexSettingRemoved() throws Excepti
             IndexSettings indexSettings = new IndexSettings(newIndexMeta("test_index", builder.build()), nodeSettings);
             ShardId shardId = new ShardId("test_index", "test_uuid", 0);
 
-            OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings);
+            OpenSearchConcurrentMergeScheduler scheduler = new OpenSearchConcurrentMergeScheduler(shardId, indexSettings, new MergedSegmentTransferTracker());
 
             // Should initially use index-level setting
             assertThat(scheduler.getForceMergeMBPerSec(), equalTo(25.0));