From 077963febeb67ab755c40e281bc7cb699ac4d561 Mon Sep 17 00:00:00 2001
From: Pranshu Shukla <pranshushukla06@gmail.com>
Date: Sun, 21 Apr 2024 15:51:21 +0530
Subject: [PATCH] Fix Flaky Test
 PersistentTasksExecutorFullRestartIT.testFullClusterRestart

Signed-off-by: Pranshu Shukla <pranshushukla06@gmail.com>
---
 .../persistent/PersistentTasksExecutorFullRestartIT.java       | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/server/src/internalClusterTest/java/org/opensearch/persistent/PersistentTasksExecutorFullRestartIT.java b/server/src/internalClusterTest/java/org/opensearch/persistent/PersistentTasksExecutorFullRestartIT.java
index 708388b3328f0..151a207d2c191 100644
--- a/server/src/internalClusterTest/java/org/opensearch/persistent/PersistentTasksExecutorFullRestartIT.java
+++ b/server/src/internalClusterTest/java/org/opensearch/persistent/PersistentTasksExecutorFullRestartIT.java
@@ -43,6 +43,7 @@
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 import static org.hamcrest.Matchers.empty;
 import static org.hamcrest.Matchers.equalTo;
@@ -129,7 +130,7 @@ public void testFullClusterRestart() throws Exception {
                     .custom(PersistentTasksCustomMetadata.TYPE)).tasks(),
                 empty()
             );
-        });
+        }, 20, TimeUnit.SECONDS);
 
     }
 }