diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApi2Test.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApi2Test.java index 1128911911722..580aa90693656 100644 --- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApi2Test.java +++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApi2Test.java @@ -1669,11 +1669,13 @@ public void testBacklogNoDelayedForPartitionedTopic() throws PulsarClientExcepti producer.send("message-1".getBytes(StandardCharsets.UTF_8)); } } - - TopicStats topicStats = admin.topics().getPartitionedStats(topic, false, true, true, true); - assertEquals(topicStats.getSubscriptions().get(subName).getMsgBacklog(), 10); - assertEquals(topicStats.getSubscriptions().get(subName).getBacklogSize(), 440); - assertEquals(topicStats.getSubscriptions().get(subName).getMsgBacklogNoDelayed(), 5); + // wait until the message add to delay queue. + Awaitility.await().untilAsserted(() -> { + TopicStats topicStats = admin.topics().getPartitionedStats(topic, false, true, true, true); + assertEquals(topicStats.getSubscriptions().get(subName).getMsgBacklog(), 10); + assertEquals(topicStats.getSubscriptions().get(subName).getBacklogSize(), 440); + assertEquals(topicStats.getSubscriptions().get(subName).getMsgBacklogNoDelayed(), 5); + }); for (int i = 0; i < 5; i++) { consumer.acknowledge(consumer.receive());