Skip to content

Commit 0a814c9

Browse files
committed
Fix soaktest to run without TestConsumer
as new protocol configuration can be put in configuration properties file
1 parent 888829a commit 0a814c9

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/soak/soakclient.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
#
2727

2828
from confluent_kafka import KafkaError, KafkaException, version
29-
from confluent_kafka import Producer
29+
from confluent_kafka import Producer, Consumer
3030
from confluent_kafka.admin import AdminClient, NewTopic
3131
from collections import defaultdict
3232
from builtins import int
3333
from opentelemetry import metrics
34-
from common import TestConsumer
3534
import argparse
3635
import threading
3736
import time
@@ -447,7 +446,7 @@ def filter_config(conf, filter_out, strip_prefix):
447446
cconf['error_cb'] = self.consumer_error_cb
448447
cconf['on_commit'] = self.consumer_commit_cb
449448
self.logger.info("consumer: using group.id {}".format(cconf['group.id']))
450-
self.consumer = TestConsumer(cconf)
449+
self.consumer = Consumer(cconf)
451450

452451
# Create and start producer thread
453452
self.producer_thread = threading.Thread(target=self.producer_thread_main)

0 commit comments

Comments
 (0)