(pulsar) 02/08: [fix][test] Improve reliability of IncrementPartitionsTest (#24172)

[lhotari]

This is an automated email from the ASF dual-hosted git repository.

lhotari pushed a commit to branch branch-3.0
in repository https://gitbox.apache.org/repos/asf/pulsar.git

commit 5fd1f1ab7538c94e7d39c2348eb2691ce1c4140c
Author: Shuta Kondo <takekon...@gmail.com>
AuthorDate: Fri Apr 11 17:13:08 2025 +0900

    [fix][test] Improve reliability of IncrementPartitionsTest (#24172)
    
    (cherry picked from commit 65a2d5e256ffc655bfe0b6a7b5d11584ddf427db)
---
 .../org/apache/pulsar/broker/admin/IncrementPartitionsTest.java  | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git 
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/IncrementPartitionsTest.java
 
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/IncrementPartitionsTest.java
index 0032bb3f267..091f4ae6c07 100644
--- 
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/IncrementPartitionsTest.java
+++ 
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/IncrementPartitionsTest.java
@@ -25,6 +25,7 @@ import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.UUID;
+import java.util.concurrent.atomic.AtomicReference;
 import lombok.Cleanup;
 import org.apache.pulsar.broker.BrokerTestUtil;
 import org.apache.pulsar.broker.admin.AdminApiTest.MockedPulsarService;
@@ -107,8 +108,12 @@ public class IncrementPartitionsTest extends 
MockedPulsarServiceBaseTest {
         admin.topics().updatePartitionedTopic(partitionedTopicName, 20);
         
assertEquals(admin.topics().getPartitionedTopicMetadata(partitionedTopicName).partitions,
 20);
 
-        for (TopicStats topic : 
admin.topics().getPartitionedStats(partitionedTopicName, true).getPartitions()
-                                .values()) {
+        AtomicReference<Map<String, ? extends TopicStats>> partitionsStats = 
new AtomicReference<>();
+        Awaitility.await().until(() -> {
+            
partitionsStats.set(admin.topics().getPartitionedStats(partitionedTopicName, 
true).getPartitions());
+            return partitionsStats.get().size() == 20;
+        });
+        for (TopicStats topic : partitionsStats.get().values()) {
                 assertEquals(topic.getSubscriptions().keySet(), 
List.of("sub-1"));
         }