From 8e142453d33060fc7f0ba5a5654bad433e58a0a9 Mon Sep 17 00:00:00 2001 From: Michal Maslanka Date: Wed, 11 May 2022 11:12:38 +0200 Subject: [PATCH] tests/kafka_cli_consumer: added wait_for_started method Signed-off-by: Michal Maslanka --- tests/rptest/services/kafka_cli_consumer.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/rptest/services/kafka_cli_consumer.py b/tests/rptest/services/kafka_cli_consumer.py index 16778094be82c..ee54865a5f260 100644 --- a/tests/rptest/services/kafka_cli_consumer.py +++ b/tests/rptest/services/kafka_cli_consumer.py @@ -91,6 +91,15 @@ def wait_for_messages(self, messages, timeout=30): timeout, backoff_sec=2) + def wait_for_started(self, timeout=10): + def all_started(): + return all([ + len(node.account.java_pids("ConsoleConsumer")) == 1 + for node in self.nodes + ]) + + wait_until(all_started, timeout, backoff_sec=1) + def stop_node(self, node): self._stopping.set() node.account.kill_process("java", clean_shutdown=True)