From bfed90ee3f1e156007ef56d9237acc1a6dbcb7aa Mon Sep 17 00:00:00 2001 From: Emanuele Sabellico Date: Mon, 16 Dec 2024 14:50:15 +0100 Subject: [PATCH] Fix soaktest to run without TestConsumer as new protocol configuration can be put in configuration properties file --- tests/soak/soakclient.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/soak/soakclient.py b/tests/soak/soakclient.py index 027cc2e83..453c7e5a5 100755 --- a/tests/soak/soakclient.py +++ b/tests/soak/soakclient.py @@ -26,12 +26,11 @@ # from confluent_kafka import KafkaError, KafkaException, version -from confluent_kafka import Producer +from confluent_kafka import Producer, Consumer from confluent_kafka.admin import AdminClient, NewTopic from collections import defaultdict from builtins import int from opentelemetry import metrics -from common import TestConsumer import argparse import threading import time @@ -447,7 +446,7 @@ def filter_config(conf, filter_out, strip_prefix): cconf['error_cb'] = self.consumer_error_cb cconf['on_commit'] = self.consumer_commit_cb self.logger.info("consumer: using group.id {}".format(cconf['group.id'])) - self.consumer = TestConsumer(cconf) + self.consumer = Consumer(cconf) # Create and start producer thread self.producer_thread = threading.Thread(target=self.producer_thread_main)