Wait all brokers to start running

apache · Jun 22, 2024 · 4583d28 · 4583d28
1 parent 9433123
commit 4583d28
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/src/test/java/kafka/test/junit/RaftClusterInvocationContext.java b/core/src/test/java/kafka/test/junit/RaftClusterInvocationContext.java
@@ -189,7 +189,9 @@ public void start() {
                 if (started.compareAndSet(false, true)) {
                     clusterTestKit.startup();
                     kafka.utils.TestUtils.waitUntilTrue(
-                            () -> this.clusterTestKit.brokers().get(0).brokerState() == BrokerState.RUNNING,
+                            () -> this.clusterTestKit.brokers().values().stream().allMatch(
+                                    brokers -> brokers.brokerState() == BrokerState.RUNNING
+                            ),
                             () -> "Broker never made it to RUNNING state.",
                             org.apache.kafka.test.TestUtils.DEFAULT_MAX_WAIT_MS,
                             100L);