From 927d0cc0592c72ecc0fb5bde65cd6f46b525499b Mon Sep 17 00:00:00 2001
From: Rishab Nahata <rnnahata@amazon.com>
Date: Tue, 10 Jan 2023 22:58:43 +0530
Subject: [PATCH] Test fix

Signed-off-by: Rishab Nahata <rnnahata@amazon.com>
---
 .../cluster/decommission/DecommissionServiceTests.java        | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java b/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java
index 2ea6fc9eff4e3..51cd0e6eb23ed 100644
--- a/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java
+++ b/server/src/test/java/org/opensearch/cluster/decommission/DecommissionServiceTests.java
@@ -343,10 +343,11 @@ public void testScheduleNodesDecommissionOnTimeout() {
 
     public void testDrainNodesWithDecommissionedAttributeWithNoDelay() {
         DecommissionAttribute decommissionAttribute = new DecommissionAttribute("zone", "zone-2");
+        String requestID = randomAlphaOfLength(10);
         DecommissionAttributeMetadata decommissionAttributeMetadata = new DecommissionAttributeMetadata(
             decommissionAttribute,
             DecommissionStatus.INIT,
-            randomAlphaOfLength(10)
+            requestID
         );
 
         Metadata metadata = Metadata.builder().putCustom(DecommissionAttributeMetadata.TYPE, decommissionAttributeMetadata).build();
@@ -354,6 +355,7 @@ public void testDrainNodesWithDecommissionedAttributeWithNoDelay() {
 
         DecommissionRequest request = new DecommissionRequest(decommissionAttribute);
         request.setNoDelay(true);
+        request.setRequestID(requestID);
 
         setState(clusterService, state);
         decommissionService.drainNodesWithDecommissionedAttribute(request);