From bfed90ee3f1e156007ef56d9237acc1a6dbcb7aa Mon Sep 17 00:00:00 2001
From: Emanuele Sabellico <esabellico@confluent.io>
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)