Fix flaky test

streamnative · Dec 29, 2023 · d20dede · d20dede
1 parent b773e5b
commit d20dede
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ReplicatorTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ReplicatorTest.java
@@ -1683,6 +1683,7 @@ public void testReplicatorWithFailedAck() throws Exception {
         MessageIdImpl lastMessageId = (MessageIdImpl) topic.getLastMessageId().get();
         Position lastPosition = PositionImpl.get(lastMessageId.getLedgerId(), lastMessageId.getEntryId());
         ConcurrentOpenHashMap<String, Replicator> replicators = topic.getReplicators();
+        Awaitility.await().untilAsserted(() -> Assert.assertNotNull(replicators.get("r2")));
         PersistentReplicator replicator = (PersistentReplicator) replicators.get("r2");
 
         Awaitility.await().pollInterval(1, TimeUnit.SECONDS).timeout(30, TimeUnit.SECONDS)