[Issue 12897] Fix flaky test in testReplicatorProducerName (#12898)

apache · Nov 21, 2021 · 78c2981 · 78c2981
1 parent ec715f2
commit 78c2981
Showing 1 changed file with 6 additions and 4 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
@@ -724,10 +724,12 @@ public void testReplicatorProducerName() throws Exception {
         });
         Optional<Topic> topic = pulsar2.getBrokerService().getTopicReference(topicName);
         assertTrue(topic.isPresent());
-        Set<String> remoteClusters = topic.get().getProducers().values().stream()
-                .map(org.apache.pulsar.broker.service.Producer::getRemoteCluster)
-                .collect(Collectors.toSet());
-        assertTrue(remoteClusters.contains("r1"));
+        Awaitility.await().untilAsserted(() -> {
+            Set<String> remoteClusters = topic.get().getProducers().values().stream()
+                    .map(org.apache.pulsar.broker.service.Producer::getRemoteCluster)
+                    .collect(Collectors.toSet());
+            assertTrue(remoteClusters.contains("r1"));
+        });
     }
 
     @Test(priority = 5, timeOut = 30000)