From eda36ec2bbbe5f28d532eb71cc582e350d7b3844 Mon Sep 17 00:00:00 2001
From: chenhang <chenhang@apache.org>
Date: Fri, 27 Aug 2021 16:17:55 +0800
Subject: [PATCH] fix flaky test testReacquireLocksAfterSessionLost

---
 .../src/test/java/org/apache/pulsar/metadata/ZKSessionTest.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/ZKSessionTest.java b/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/ZKSessionTest.java
index 5f8c690752b4f..bb61edda08687 100644
--- a/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/ZKSessionTest.java
+++ b/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/ZKSessionTest.java
@@ -127,7 +127,7 @@ public void testReacquireLocksAfterSessionLost() throws Exception {
         e = sessionEvents.poll(10, TimeUnit.SECONDS);
         assertEquals(e, SessionEvent.SessionReestablished);
 
-        Awaitility.await().untilAsserted(() -> {
+        Awaitility.waitAtMost(30, TimeUnit.SECONDS).untilAsserted(() -> {
             assertFalse(lock.getLockExpiredFuture().isDone());
         });