Fix FullRollingRestartIT

Signed-off-by: Sachin Kale <kalsac@amazon.com>
soosinha · Sep 4, 2023 · 65bc7cd · 65bc7cd
1 parent 59d08b3
commit 65bc7cd
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/server/src/internalClusterTest/java/org/opensearch/recovery/FullRollingRestartIT.java b/server/src/internalClusterTest/java/org/opensearch/recovery/FullRollingRestartIT.java
@@ -72,15 +72,15 @@ public void testFullRollingRestart() throws Exception {
 
         final String healthTimeout = "1m";
 
-        for (int i = 0; i < 1000; i++) {
+        for (int i = 0; i < 100; i++) {
             client().prepareIndex("test")
                 .setId(Long.toString(i))
                 .setSource(MapBuilder.<String, Object>newMapBuilder().put("test", "value" + i).map())
                 .execute()
                 .actionGet();
         }
         flush();
-        for (int i = 1000; i < 2000; i++) {
+        for (int i = 100; i < 200; i++) {
             client().prepareIndex("test")
                 .setId(Long.toString(i))
                 .setSource(MapBuilder.<String, Object>newMapBuilder().put("test", "value" + i).map())
@@ -123,7 +123,7 @@ public void testFullRollingRestart() throws Exception {
         logger.info("--> refreshing and checking data");
         refresh();
         for (int i = 0; i < 10; i++) {
-            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 2000L);
+            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 200L);
         }
 
         // now start shutting nodes down
@@ -156,7 +156,7 @@ public void testFullRollingRestart() throws Exception {
         logger.info("--> stopped two nodes, verifying data");
         refresh();
         for (int i = 0; i < 10; i++) {
-            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 2000L);
+            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 200L);
         }
 
         // closing the 3rd node
@@ -190,7 +190,7 @@ public void testFullRollingRestart() throws Exception {
         logger.info("--> one node left, verifying data");
         refresh();
         for (int i = 0; i < 10; i++) {
-            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 2000L);
+            assertHitCount(client().prepareSearch().setPreference("_primary").setSize(0).setQuery(matchAllQuery()).get(), 200L);
         }
     }