Fix flaky test BrokerBkEnsemblesTests.testCrashBrokerWithoutCursorLed…

…gerLeak (apache#14519) - wrap assertion with Awaitility
drianpauL · Mar 2, 2022 · 40e408d · 40e408d
1 parent daed6a0
commit 40e408d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/BrokerBkEnsemblesTests.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/BrokerBkEnsemblesTests.java
@@ -48,6 +48,7 @@
 import org.apache.pulsar.client.api.PulsarClient;
 import org.apache.pulsar.client.api.Schema;
 import org.apache.zookeeper.ZooKeeper;
+import org.awaitility.Awaitility;
 import org.testng.Assert;
 import org.testng.annotations.Test;
 
@@ -150,7 +151,9 @@ public void testCrashBrokerWithoutCursorLedgerLeak() throws Exception {
         Assert.assertNotEquals(cursorLedgerId, newCursorLedgerId);
 
         // cursor node must be deleted
-        Assert.assertNull(zk.exists(ledgerPath, false));
+        Awaitility.await().untilAsserted(() -> {
+            Assert.assertNull(zk.exists(ledgerPath, false));
+        });
 
         producer.close();
         consumer.close();