Fixed flaky test in LocalMemoryMetadataStoreTest (apache#12540)

mfleming · Oct 30, 2021 · f527a22 · f527a22
1 parent 1c9fc07
commit f527a22
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/LocalMemoryMetadataStoreTest.java b/pulsar-metadata/src/test/java/org/apache/pulsar/metadata/LocalMemoryMetadataStoreTest.java
@@ -45,6 +45,7 @@
 import org.apache.pulsar.metadata.api.Notification;
 import org.apache.pulsar.metadata.api.NotificationType;
 import org.apache.pulsar.metadata.api.Stat;
+import org.awaitility.Awaitility;
 import org.testng.annotations.Test;
 
 public class LocalMemoryMetadataStoreTest {
@@ -84,7 +85,11 @@ public void testSharedInstance() throws Exception {
 
         store2.delete("/test", Optional.empty()).join();
 
-        assertFalse(store1.exists("/test").join());
         assertFalse(store2.exists("/test").join());
+
+        // The exists will be updated based on the cache invalidation in store1
+        Awaitility.await().untilAsserted(() -> {
+            assertFalse(store1.exists("/test").join());
+        });
     }
 }