Fix test case

Signed-off-by: pranikum <109206473+pranikum@users.noreply.github.com>
opensearch-project · Sep 25, 2022 · f8c1882 · f8c1882
1 parent 96bb379
commit f8c1882
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -19,6 +19,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - Add BWC version 2.3.1 ([#4513](https://github.com/opensearch-project/OpenSearch/pull/4513))
 - [Segment Replication] Add snapshot and restore tests for segment replication feature ([#3993](https://github.com/opensearch-project/OpenSearch/pull/3993))
 - Added missing javadocs for `:example-plugins` modules ([#4540](https://github.com/opensearch-project/OpenSearch/pull/4540))
+- Added changes for graceful node decommission ([#4405](https://github.com/opensearch-project/OpenSearch/pull/4405))
 ### Dependencies
 - Bumps `log4j-core` from 2.18.0 to 2.19.0
 

diff --git a/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java b/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java
@@ -26,6 +26,7 @@
 import org.opensearch.cluster.service.ClusterService;
 import org.opensearch.common.settings.ClusterSettings;
 import org.opensearch.common.settings.Settings;
+import org.opensearch.common.unit.TimeValue;
 import org.opensearch.test.OpenSearchTestCase;
 import org.opensearch.test.transport.MockTransport;
 import org.opensearch.threadpool.TestThreadPool;
@@ -133,7 +134,7 @@ public void onFailure(Exception e) {
                 countDownLatch.countDown();
             }
         };
-        decommissionService.startDecommissionAction(decommissionAttribute, listener);
+        decommissionService.startDecommissionAction(decommissionAttribute, listener, TimeValue.timeValueSeconds(30));
         assertTrue(countDownLatch.await(30, TimeUnit.SECONDS));
     }
 
@@ -160,7 +161,7 @@ public void onFailure(Exception e) {
                 countDownLatch.countDown();
             }
         };
-        decommissionService.startDecommissionAction(decommissionAttribute, listener);
+        decommissionService.startDecommissionAction(decommissionAttribute, listener, TimeValue.timeValueSeconds(30));
         assertTrue(countDownLatch.await(30, TimeUnit.SECONDS));
     }
 
@@ -197,7 +198,7 @@ public void onFailure(Exception e) {
                 countDownLatch.countDown();
             }
         };
-        decommissionService.startDecommissionAction(new DecommissionAttribute("zone", "zone_2"), listener);
+        decommissionService.startDecommissionAction(new DecommissionAttribute("zone", "zone_2"), listener, TimeValue.timeValueSeconds(30));
         assertTrue(countDownLatch.await(30, TimeUnit.SECONDS));
     }