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)