Check URPs in group_mode_transactions_test.py as well

apache · Aug 26, 2021 · 4eaa389 · 4eaa389
1 parent e05a85b
commit 4eaa389
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/kafkatest/tests/core/group_mode_transactions_test.py b/tests/kafkatest/tests/core/group_mode_transactions_test.py
@@ -110,6 +110,10 @@ def bounce_brokers(self, clean_shutdown):
                     time.sleep(brokerSessionTimeoutSecs + gracePeriodSecs)
                 self.kafka.start_node(node)
 
+            wait_until(lambda: not self.kafka.has_under_replicated_partitions(),
+                       timeout_sec = 30,
+                       err_msg="Timed out waiting for under-replicated-partitions to clear after bouncing broker %s" % str(node.account))
+
     def create_and_start_message_copier(self, input_topic, output_topic, transactional_id):
         message_copier = TransactionalMessageCopier(
             context=self.test_context,