RemoteIndexRecoveryIT.testRerouteRecovery: fix simluated slow down fo…

…r remote backed index recovery Signed-off-by: bansvaru <[email protected]>
opensearch-project · Aug 30, 2023 · ba41aa9 · ba41aa9
1 parent 8324b88
commit ba41aa9
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/server/src/internalClusterTest/java/org/opensearch/indices/recovery/IndexRecoveryIT.java b/server/src/internalClusterTest/java/org/opensearch/indices/recovery/IndexRecoveryIT.java
@@ -252,7 +252,7 @@ private void assertOnGoingRecoveryState(
  assertThat(state.getStage(), not(equalTo(Stage.DONE)));
  }
 
- private void slowDownRecovery(ByteSizeValue shardSize) {
+ public void slowDownRecovery(ByteSizeValue shardSize) {
  long chunkSize = Math.max(1, shardSize.getBytes() / 10);
  assertTrue(
  client().admin()

diff --git a/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteIndexRecoveryIT.java b/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteIndexRecoveryIT.java
@@ -11,6 +11,8 @@
 import org.opensearch.cluster.metadata.IndexMetadata;
 import org.opensearch.common.settings.Settings;
 import org.opensearch.common.util.FeatureFlags;
+import org.opensearch.core.common.unit.ByteSizeUnit;
+import org.opensearch.core.common.unit.ByteSizeValue;
 import org.opensearch.index.IndexModule;
 import org.opensearch.index.IndexSettings;
 import org.opensearch.indices.recovery.IndexRecoveryIT;
@@ -171,4 +173,25 @@ public void testDisconnectsDuringRecovery() {
  public void testReplicaRecovery() {
 
  }
+
+ /**
+ * Slowing down recoveries for remote backed indices requires to limit download bytes for remote repository
+ */
+ @Override
+ public void slowDownRecovery(ByteSizeValue shardSize) {
+ long chunkSize = Math.max(1, shardSize.getBytes() / 10);
+ assertAcked(
+ client().admin()
+ .cluster()
+ .preparePutRepository(REPOSITORY_NAME)
+ .setType("fs")
+ .setSettings(
+ Settings.builder()
+ .put("location", absolutePath)
+ .put("max_remote_download_bytes_per_sec", chunkSize, ByteSizeUnit.BYTES)
+ .put("chunk_size", chunkSize, ByteSizeUnit.BYTES)
+
+ )
+ );
+ }
 }