Enabling integ test case for context aware segments

RS146BIJAY · RS146BIJAY · commit 290d97075726 · 2026-01-21T22:50:26.000+05:30
Signed-off-by: RS146BIJAY &lt;rishavsagar4b1@gmail.com&gt;
diff --git a/modules/transport-grpc/src/internalClusterTest/java/org/opensearch/transport/grpc/LoadExtendingPluginServicesIT.java b/modules/transport-grpc/src/internalClusterTest/java/org/opensearch/transport/grpc/LoadExtendingPluginServicesIT.java
@@ -108,6 +108,17 @@ protected Collection<PluginInfo> additionalNodePlugins() {
                 null,
                 List.of(GrpcPlugin.class.getName()),
                 false
+            ),
+            new PluginInfo(
+                ContextAwareIndexPlugin.class.getName(),
+                "classpath plugin",
+                "NA",
+                org.opensearch.Version.CURRENT,
+                "1.8",
+                ContextAwareIndexPlugin.class.getName(),
+                null,
+                Collections.emptyList(),
+                false
             )
         );
     }
diff --git a/server/src/internalClusterTest/java/org/opensearch/action/admin/cluster/node/tasks/ConcurrentSearchTasksIT.java b/server/src/internalClusterTest/java/org/opensearch/action/admin/cluster/node/tasks/ConcurrentSearchTasksIT.java
@@ -15,6 +15,7 @@
 import org.opensearch.common.settings.FeatureFlagSettings;
 import org.opensearch.common.settings.Setting;
 import org.opensearch.common.settings.Settings;
+import org.opensearch.common.util.FeatureFlags;
 import org.opensearch.core.tasks.resourcetracker.ThreadResourceInfo;
 import org.opensearch.index.query.QueryBuilders;
 import org.opensearch.search.SearchService;
@@ -67,6 +68,8 @@ protected Settings featureFlagSettings() {
         for (Setting builtInFlag : FeatureFlagSettings.BUILT_IN_FEATURE_FLAGS) {
             featureSettings.put(builtInFlag.getKey(), builtInFlag.getDefaultRaw(Settings.EMPTY));
         }
+
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/server/src/internalClusterTest/java/org/opensearch/cluster/routing/allocation/decider/ShardsLimitAllocationDeciderIT.java b/server/src/internalClusterTest/java/org/opensearch/cluster/routing/allocation/decider/ShardsLimitAllocationDeciderIT.java
@@ -105,6 +105,7 @@ protected Settings nodeSettings(int nodeOrdinal) {
     protected Settings featureFlagSettings() {
         Settings.Builder featureSettings = Settings.builder();
         featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/server/src/internalClusterTest/java/org/opensearch/cluster/shards/ClusterShardLimitIT.java b/server/src/internalClusterTest/java/org/opensearch/cluster/shards/ClusterShardLimitIT.java
@@ -140,6 +140,7 @@ protected Settings nodeSettings(int nodeOrdinal) {
     protected Settings featureFlagSettings() {
         Settings.Builder featureSettings = Settings.builder();
         featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/server/src/internalClusterTest/java/org/opensearch/index/shard/IndexShardIT.java b/server/src/internalClusterTest/java/org/opensearch/index/shard/IndexShardIT.java
@@ -95,6 +95,7 @@
 import org.opensearch.test.DummyShardLock;
 import org.opensearch.test.IndexSettingsModule;
 import org.opensearch.test.InternalSettingsPlugin;
+import org.opensearch.test.OpenSearchIntegTestCase;
 import org.opensearch.test.OpenSearchSingleNodeTestCase;
 import org.opensearch.test.OpenSearchTestCase;
 import org.junit.Assert;
@@ -664,7 +665,8 @@ public void postDelete(ShardId shardId, Engine.Delete delete, Engine.DeleteResul
             getInstanceFromNode(CircuitBreakerService.class),
             env.nodeId(),
             getInstanceFromNode(ClusterService.class),
-            listener
+            listener,
+            new OpenSearchIntegTestCase.ContextAwareIndexOperationListener()
         );
         shardRef.set(newShard);
         recoverShard(newShard);
diff --git a/server/src/internalClusterTest/java/org/opensearch/indices/memory/breaker/CircuitBreakerNoopIT.java b/server/src/internalClusterTest/java/org/opensearch/indices/memory/breaker/CircuitBreakerNoopIT.java
@@ -34,6 +34,7 @@
 
 import org.opensearch.action.index.IndexRequestBuilder;
 import org.opensearch.common.settings.Settings;
+import org.opensearch.common.util.FeatureFlags;
 import org.opensearch.indices.breaker.HierarchyCircuitBreakerService;
 import org.opensearch.search.sort.SortOrder;
 import org.opensearch.test.OpenSearchIntegTestCase;
@@ -59,6 +60,7 @@ protected Settings nodeSettings(int nodeOrdinal) {
             .put(HierarchyCircuitBreakerService.REQUEST_CIRCUIT_BREAKER_TYPE_SETTING.getKey(), "noop")
             // This is set low, because if the "noop" is not a noop, it will break
             .put(HierarchyCircuitBreakerService.REQUEST_CIRCUIT_BREAKER_LIMIT_SETTING.getKey(), "10b")
+            .put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true)
             .build();
     }
 
diff --git a/server/src/internalClusterTest/java/org/opensearch/indices/replication/WarmIndexSegmentReplicationIT.java b/server/src/internalClusterTest/java/org/opensearch/indices/replication/WarmIndexSegmentReplicationIT.java
@@ -157,6 +157,7 @@ protected Settings nodeSettings(int nodeOrdinal) {
     protected Settings featureFlagSettings() {
         Settings.Builder featureSettings = Settings.builder();
         featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/server/src/internalClusterTest/java/org/opensearch/plugins/ClasspathPluginIT.java b/server/src/internalClusterTest/java/org/opensearch/plugins/ClasspathPluginIT.java
@@ -65,6 +65,17 @@ protected Collection<PluginInfo> additionalNodePlugins() {
                 null,
                 List.of(SampleExtensiblePlugin.class.getName()),
                 false
+            ),
+            new PluginInfo(
+                ContextAwareIndexPlugin.class.getName(),
+                "classpath plugin",
+                "NA",
+                org.opensearch.Version.CURRENT,
+                "1.8",
+                ContextAwareIndexPlugin.class.getName(),
+                null,
+                Collections.emptyList(),
+                false
             )
         );
     }
diff --git a/server/src/internalClusterTest/java/org/opensearch/remotestore/WritableWarmIT.java b/server/src/internalClusterTest/java/org/opensearch/remotestore/WritableWarmIT.java
@@ -77,6 +77,7 @@ protected boolean addMockIndexStorePlugin() {
     protected Settings featureFlagSettings() {
         Settings.Builder featureSettings = Settings.builder();
         featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/server/src/main/java/org/opensearch/common/settings/FeatureFlagSettings.java b/server/src/main/java/org/opensearch/common/settings/FeatureFlagSettings.java
@@ -38,6 +38,7 @@ protected FeatureFlagSettings(
         FeatureFlags.APPLICATION_BASED_CONFIGURATION_TEMPLATES_SETTING,
         FeatureFlags.TERM_VERSION_PRECOMMIT_ENABLE_SETTING,
         FeatureFlags.ARROW_STREAMS_SETTING,
-        FeatureFlags.STREAM_TRANSPORT_SETTING
+        FeatureFlags.STREAM_TRANSPORT_SETTING,
+        FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_SETTING
     );
 }
diff --git a/server/src/main/java/org/opensearch/index/engine/CompositeIndexWriter.java b/server/src/main/java/org/opensearch/index/engine/CompositeIndexWriter.java
@@ -39,6 +39,7 @@
 
 import java.io.Closeable;
 import java.io.IOException;
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
@@ -347,12 +348,12 @@ public CriteriaBasedIndexWriterLookup acquire() throws EngineException {
             public CriteriaBasedIndexWriterLookup tryAcquire() {
                 boolean locked = lock.tryLock();
                 if (locked) {
-                    assert addCurrentThread();
                     if (lookup.isClosed()) {
-                        this.close();
+                        lock.unlock();
                         return null;
                     }
 
+                    assert addCurrentThread();
                     return lookup;
                 } else {
                     return null;
@@ -544,14 +545,18 @@ public void beforeRefresh() throws IOException {
     private void refreshDocumentsForParentDirectory(CriteriaBasedIndexWriterLookup oldMap) throws IOException {
         final Map<String, CompositeIndexWriter.DisposableIndexWriter> markForRefreshIndexWritersMap = oldMap.criteriaBasedIndexWriterMap;
         deletePreviousVersionsForUpdatedDocuments();
-        Directory directoryToCombine;
+        final List<Directory> directoryToCombine = new ArrayList<>();
+        final AtomicLong pendingNumDocsByOldChildWriter = new AtomicLong();
         for (CompositeIndexWriter.DisposableIndexWriter childDisposableWriter : markForRefreshIndexWritersMap.values()) {
-            directoryToCombine = childDisposableWriter.getIndexWriter().getDirectory();
+            directoryToCombine.add(childDisposableWriter.getIndexWriter().getDirectory());
             childDisposableWriter.getIndexWriter().close();
-            long pendingNumDocsByOldChildWriter = childDisposableWriter.getIndexWriter().getPendingNumDocs();
-            accumulatingIndexWriter.addIndexes(directoryToCombine);
+            pendingNumDocsByOldChildWriter.addAndGet(childDisposableWriter.getIndexWriter().getPendingNumDocs());
+        }
+
+        if (!directoryToCombine.isEmpty()) {
+            accumulatingIndexWriter.addIndexes(directoryToCombine.toArray(new Directory[0]));
             IOUtils.closeWhileHandlingException(directoryToCombine);
-            childWriterPendingNumDocs.addAndGet(-pendingNumDocsByOldChildWriter);
+            childWriterPendingNumDocs.addAndGet(-pendingNumDocsByOldChildWriter.get());
         }
 
         deleteDummyTombstoneEntry();
diff --git a/server/src/main/java/org/opensearch/index/shard/IndexShard.java b/server/src/main/java/org/opensearch/index/shard/IndexShard.java
@@ -4995,7 +4995,7 @@ public final boolean isSearchIdleSupported() {
         // refresh. If the index is context aware enabled, search idle is not supported. This will ensure periodic sync
         // between child and parent IndexWriter.
         // task continues to upload to remote store periodically.
-        if (isRemoteTranslogEnabled() || indexSettings.isAssignedOnRemoteNode() || indexSettings.isContextAwareEnabled()) {
+        if (isRemoteTranslogEnabled() || indexSettings.isAssignedOnRemoteNode()) {
             return false;
         }
         return indexSettings.isSegRepEnabledOrRemoteNode() == false || indexSettings.getNumberOfReplicas() == 0;
diff --git a/server/src/test/java/org/opensearch/action/admin/indices/tiering/TransportHotToWarmTieringActionTests.java b/server/src/test/java/org/opensearch/action/admin/indices/tiering/TransportHotToWarmTieringActionTests.java
@@ -47,6 +47,7 @@ public class TransportHotToWarmTieringActionTests extends OpenSearchIntegTestCas
     protected Settings featureFlagSettings() {
         Settings.Builder featureSettings = Settings.builder();
         featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);
+        featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);
         return featureSettings.build();
     }
 
diff --git a/test/framework/src/main/java/org/opensearch/remotestore/RemoteStoreBaseIntegTestCase.java b/test/framework/src/main/java/org/opensearch/remotestore/RemoteStoreBaseIntegTestCase.java
@@ -75,6 +75,7 @@ public class RemoteStoreBaseIntegTestCase extends OpenSearchIntegTestCase {
     protected static int REPLICA_COUNT = 1;
     protected static final String TOTAL_OPERATIONS = "total-operations";
     protected static final String REFRESHED_OR_FLUSHED_OPERATIONS = "refreshed-or-flushed-operations";
+    protected static final String FLUSHED_OPERATIONS = "flushed-operations";
     protected static final String MAX_SEQ_NO_TOTAL = "max-seq-no-total";
     protected static final String MAX_SEQ_NO_REFRESHED_OR_FLUSHED = "max-seq-no-refreshed-or-flushed";
 
@@ -101,10 +102,12 @@ protected Map<String, Long> indexData(int numberOfIterations, boolean invokeFlus
         long maxSeqNo = -1;
         long maxSeqNoRefreshedOrFlushed = -1;
         int shardId = 0;
+        long flushCount = 0;
         Map<String, Long> indexingStats = new HashMap<>();
         for (int i = 0; i < numberOfIterations; i++) {
             if (invokeFlush) {
                 flushAndRefresh(index);
+                ++flushCount;
             } else {
                 refresh(index);
             }
@@ -133,6 +136,7 @@ protected Map<String, Long> indexData(int numberOfIterations, boolean invokeFlus
 
         indexingStats.put(TOTAL_OPERATIONS, totalOperations);
         indexingStats.put(REFRESHED_OR_FLUSHED_OPERATIONS, refreshedOrFlushedOperations);
+        indexingStats.put(FLUSHED_OPERATIONS, flushCount);
         indexingStats.put(MAX_SEQ_NO_TOTAL, maxSeqNo);
         indexingStats.put(MAX_SEQ_NO_REFRESHED_OR_FLUSHED, maxSeqNoRefreshedOrFlushed);
         return indexingStats;
diff --git a/test/framework/src/main/java/org/opensearch/remotestore/RemoteStoreCoreTestCase.java b/test/framework/src/main/java/org/opensearch/remotestore/RemoteStoreCoreTestCase.java
@@ -15,6 +15,7 @@
 import org.opensearch.action.admin.indices.delete.DeleteIndexRequest;
 import org.opensearch.action.admin.indices.flush.FlushRequest;
 import org.opensearch.action.admin.indices.recovery.RecoveryResponse;
+import org.opensearch.action.admin.indices.settings.get.GetSettingsResponse;
 import org.opensearch.action.admin.indices.settings.put.UpdateSettingsRequest;
 import org.opensearch.action.index.IndexResponse;
 import org.opensearch.action.search.SearchPhaseExecutionException;
@@ -217,7 +218,7 @@ public void testStaleCommitDeletionWithInvokeFlush() throws Exception {
         String dataNode = internalCluster().startDataOnlyNode();
         createIndex(INDEX_NAME, remoteStoreIndexSettings(1, 10000L, -1));
         int numberOfIterations = randomIntBetween(1, 5);
-        indexData(numberOfIterations, true, INDEX_NAME);
+        Map<String, Long> indexingStats = indexData(numberOfIterations, true, INDEX_NAME);
         String segmentsPathFixedPrefix = RemoteStoreSettings.CLUSTER_REMOTE_STORE_SEGMENTS_PATH_PREFIX.get(getNodeSettings());
         String shardPath = getShardLevelBlobPath(
             client(),
@@ -233,8 +234,16 @@ public void testStaleCommitDeletionWithInvokeFlush() throws Exception {
         // Delete is async.
         assertBusy(() -> {
             int actualFileCount = getActualFileCount(segmentRepoPath, shardPath);
+            int extraMetadataFilesDueToFlush = 0;
+            String isCASEnabled = client().admin().indices().prepareGetSettings(INDEX_NAME).get()
+                .getSetting(INDEX_NAME, IndexSettings.INDEX_CONTEXT_AWARE_ENABLED_SETTING.getKey());
+            if (isCASEnabled != null && isCASEnabled.equals("true")) {
+                // For CAS enabled domains, there will be an extra refresh for each OpenSearch flush call.
+                extraMetadataFilesDueToFlush += indexingStats.get(FLUSHED_OPERATIONS);
+            }
             if (numberOfIterations <= lastNMetadataFilesToKeep) {
-                MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations - 1, numberOfIterations, numberOfIterations + 1)));
+                MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations - 1 + extraMetadataFilesDueToFlush,
+                    numberOfIterations + extraMetadataFilesDueToFlush, numberOfIterations + 1 + extraMetadataFilesDueToFlush)));
             } else {
                 // As delete is async its possible that the file gets created before the deletion or after
                 // deletion.
@@ -244,7 +253,8 @@ public void testStaleCommitDeletionWithInvokeFlush() throws Exception {
                 } else {
                     MatcherAssert.assertThat(
                         actualFileCount,
-                        is(oneOf(lastNMetadataFilesToKeep - 1, lastNMetadataFilesToKeep, lastNMetadataFilesToKeep + 1))
+                        is(oneOf(lastNMetadataFilesToKeep - 1 + extraMetadataFilesDueToFlush,
+                            lastNMetadataFilesToKeep + extraMetadataFilesDueToFlush, lastNMetadataFilesToKeep + 1 + extraMetadataFilesDueToFlush))
                     );
                 }
             }
@@ -263,7 +273,7 @@ public void testStaleCommitDeletionWithoutInvokeFlush() throws Exception {
         internalCluster().startDataOnlyNode();
         createIndex(INDEX_NAME, remoteStoreIndexSettings(1, 10000l, -1));
         int numberOfIterations = randomIntBetween(1, 5);
-        indexData(numberOfIterations, false, INDEX_NAME);
+        Map<String, Long> indexingStats = indexData(numberOfIterations, false, INDEX_NAME);
         String segmentsPathFixedPrefix = RemoteStoreSettings.CLUSTER_REMOTE_STORE_SEGMENTS_PATH_PREFIX.get(getNodeSettings());
         String shardPath = getShardLevelBlobPath(
             client(),
@@ -275,8 +285,16 @@ public void testStaleCommitDeletionWithoutInvokeFlush() throws Exception {
             segmentsPathFixedPrefix
         ).buildAsString();
         int actualFileCount = getActualFileCount(segmentRepoPath, shardPath);
+        int extraMetadataFilesDueToFlush = 0;
+        String isCASEnabled = client().admin().indices().prepareGetSettings(INDEX_NAME).get()
+            .getSetting(INDEX_NAME, IndexSettings.INDEX_CONTEXT_AWARE_ENABLED_SETTING.getKey());
+        if (isCASEnabled != null && isCASEnabled.equals("true")) {
+            // For CAS enabled domains, there will be an extra refresh for each OpenSearch flush call.
+            extraMetadataFilesDueToFlush += indexingStats.get(FLUSHED_OPERATIONS);
+        }
         // We also allow (numberOfIterations + 1) as index creation also triggers refresh.
-        MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations - 1, numberOfIterations, numberOfIterations + 1)));
+        MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations - 1 + extraMetadataFilesDueToFlush,
+            numberOfIterations + extraMetadataFilesDueToFlush, numberOfIterations + 1 + extraMetadataFilesDueToFlush)));
     }
 
     public void testStaleCommitDeletionWithMinSegmentFiles_3() throws Exception {
@@ -317,7 +335,7 @@ public void testStaleCommitDeletionWithMinSegmentFiles_Disabled() throws Excepti
 
         createIndex(INDEX_NAME, remoteStoreIndexSettings(1, 10000l, -1));
         int numberOfIterations = randomIntBetween(2, 5);
-        indexData(numberOfIterations, true, INDEX_NAME);
+        Map<String, Long> indexingStats = indexData(numberOfIterations, true, INDEX_NAME);
         String segmentsPathFixedPrefix = RemoteStoreSettings.CLUSTER_REMOTE_STORE_SEGMENTS_PATH_PREFIX.get(getNodeSettings());
         String shardPath = getShardLevelBlobPath(
             client(),
@@ -329,8 +347,15 @@ public void testStaleCommitDeletionWithMinSegmentFiles_Disabled() throws Excepti
             segmentsPathFixedPrefix
         ).buildAsString();
         int actualFileCount = getActualFileCount(segmentRepoPath, shardPath);
+        int extraMetadataFilesDueToFlush = 0;
+        String isCASEnabled = client().admin().indices().prepareGetSettings(INDEX_NAME).get()
+            .getSetting(INDEX_NAME, IndexSettings.INDEX_CONTEXT_AWARE_ENABLED_SETTING.getKey());
+        if (isCASEnabled != null && isCASEnabled.equals("true")) {
+            // For CAS enabled domains, there will be an extra refresh for each OpenSearch flush call.
+            extraMetadataFilesDueToFlush += indexingStats.get(FLUSHED_OPERATIONS);
+        }
         // We also allow (numberOfIterations + 1) as index creation also triggers refresh.
-        MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations, numberOfIterations + 1)));
+        MatcherAssert.assertThat(actualFileCount, is(oneOf(numberOfIterations + extraMetadataFilesDueToFlush, numberOfIterations + 1 + extraMetadataFilesDueToFlush)));
     }
 
     protected int getActualFileCount(Path segmentRepoPath, String shardPath) throws IOException {
diff --git a/test/framework/src/main/java/org/opensearch/snapshots/AbstractSnapshotIntegTestCase.java b/test/framework/src/main/java/org/opensearch/snapshots/AbstractSnapshotIntegTestCase.java
@@ -155,7 +155,7 @@ protected Settings nodeSettings(int nodeOrdinal) {
 
     @Override
     protected Collection<Class<? extends Plugin>> nodePlugins() {
-        return Arrays.asList(MockRepository.Plugin.class);
+        return List.of(MockRepository.Plugin.class, ContextAwareIndexPlugin.class);
     }
 
     @After
diff --git a/test/framework/src/main/java/org/opensearch/test/AbstractMultiClustersTestCase.java b/test/framework/src/main/java/org/opensearch/test/AbstractMultiClustersTestCase.java
@@ -78,7 +78,7 @@ protected Collection<String> remoteClusterAlias() {
     }
 
     protected Collection<Class<? extends Plugin>> nodePlugins(String clusterAlias) {
-        return Collections.emptyList();
+        return Collections.singletonList(OpenSearchIntegTestCase.ContextAwareIndexPlugin.class);
     }
 
     protected final Client client() {
diff --git a/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java b/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
diff --git a/test/framework/src/main/java/org/opensearch/test/OpenSearchSingleNodeTestCase.java b/test/framework/src/main/java/org/opensearch/test/OpenSearchSingleNodeTestCase.java

Original file line number	Diff line number	Diff line change
`@@ -105,6 +105,7 @@ protected Settings nodeSettings(int nodeOrdinal) {`
`105`	`105`	`protected Settings featureFlagSettings() {`
`106`	`106`	`Settings.Builder featureSettings = Settings.builder();`
`107`	`107`	`featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);`
	`108`	`+ featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);`
`108`	`109`	`return featureSettings.build();`
`109`	`110`	`}`
`110`	`111`
Original file line number	Diff line number	Diff line change
`@@ -140,6 +140,7 @@ protected Settings nodeSettings(int nodeOrdinal) {`
`140`	`140`	`protected Settings featureFlagSettings() {`
`141`	`141`	`Settings.Builder featureSettings = Settings.builder();`
`142`	`142`	`featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);`
	`143`	`+ featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);`
`143`	`144`	`return featureSettings.build();`
`144`	`145`	`}`
`145`	`146`
Original file line number	Diff line number	Diff line change
`@@ -157,6 +157,7 @@ protected Settings nodeSettings(int nodeOrdinal) {`
`157`	`157`	`protected Settings featureFlagSettings() {`
`158`	`158`	`Settings.Builder featureSettings = Settings.builder();`
`159`	`159`	`featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);`
	`160`	`+ featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);`
`160`	`161`	`return featureSettings.build();`
`161`	`162`	`}`
`162`	`163`
Original file line number	Diff line number	Diff line change
`@@ -77,6 +77,7 @@ protected boolean addMockIndexStorePlugin() {`
`77`	`77`	`protected Settings featureFlagSettings() {`
`78`	`78`	`Settings.Builder featureSettings = Settings.builder();`
`79`	`79`	`featureSettings.put(FeatureFlags.WRITABLE_WARM_INDEX_EXPERIMENTAL_FLAG, true);`
	`80`	`+ featureSettings.put(FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_FLAG, true);`
`80`	`81`	`return featureSettings.build();`
`81`	`82`	`}`
`82`	`83`
Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@ protected FeatureFlagSettings(`
`38`	`38`	`FeatureFlags.APPLICATION_BASED_CONFIGURATION_TEMPLATES_SETTING,`
`39`	`39`	`FeatureFlags.TERM_VERSION_PRECOMMIT_ENABLE_SETTING,`
`40`	`40`	`FeatureFlags.ARROW_STREAMS_SETTING,`
`41`		`- FeatureFlags.STREAM_TRANSPORT_SETTING`
	`41`	`+ FeatureFlags.STREAM_TRANSPORT_SETTING,`
	`42`	`+ FeatureFlags.CONTEXT_AWARE_MIGRATION_EXPERIMENTAL_SETTING`
`42`	`43`	`);`
`43`	`44`	`}`