From 2af2478e6edc251f73e3d028d3b4f366706c187e Mon Sep 17 00:00:00 2001
From: David Arthur <mumrah@gmail.com>
Date: Thu, 5 Sep 2024 20:45:50 -0400
Subject: [PATCH] add wait on ISR

---
 .../integration/kafka/zk/ZkMigrationIntegrationTest.scala     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/core/src/test/scala/integration/kafka/zk/ZkMigrationIntegrationTest.scala b/core/src/test/scala/integration/kafka/zk/ZkMigrationIntegrationTest.scala
index 12a37a076e526..e8946bbd6f32b 100644
--- a/core/src/test/scala/integration/kafka/zk/ZkMigrationIntegrationTest.scala
+++ b/core/src/test/scala/integration/kafka/zk/ZkMigrationIntegrationTest.scala
@@ -308,7 +308,9 @@ class ZkMigrationIntegrationTest {
           false
         } else {
           topicDescribe.topicNameValues().values().stream().allMatch {
-            topic => topic.get(32, TimeUnit.SECONDS).partitions().stream().allMatch(part => part.leader() != null)
+            topic => topic.get(62, TimeUnit.SECONDS).partitions().stream().allMatch(part => {
+              part.leader() != null && part.isr().size() == 3
+            })
           }
         }
       }, msg="waiting for topics to be available", waitTimeMs=303000)