diff --git a/tests/kafkatest/tests/core/group_mode_transactions_test.py b/tests/kafkatest/tests/core/group_mode_transactions_test.py index 18e6d97ecb41..ffb7d4cc6126 100644 --- 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,