lucasbru commented on code in PR #19560:
URL: https://github.com/apache/kafka/pull/19560#discussion_r2063009976


##########
tests/kafkatest/tests/streams/base_streams_test.py:
##########
@@ -13,22 +13,40 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from ducktape.tests.test import Test
 from ducktape.utils.util import wait_until
 from kafkatest.services.verifiable_consumer import VerifiableConsumer
 from kafkatest.services.verifiable_producer import VerifiableProducer
-from kafkatest.tests.kafka_test import KafkaTest
+from kafkatest.services.zookeeper import ZookeeperService
+from kafkatest.services.kafka import KafkaService, quorum
 
 
-class BaseStreamsTest(KafkaTest):
+class BaseStreamsTest(Test):
     """
     Helper class that contains methods for producing and consuming
     messages and verification of results from log files
 
     Extends KafkaTest which manages setting up Kafka Cluster and Zookeeper
     see tests/kafkatest/tests/kafka_test.py for more info
     """
-    def __init__(self, test_context,  topics, num_controllers=1, 
num_brokers=3):
-        super(BaseStreamsTest, self).__init__(test_context, num_controllers, 
num_brokers, topics)
+    def __init__(self, test_context, topics, num_controllers=1, num_brokers=3):
+        self.num_zk = num_controllers
+        self.num_brokers = num_brokers
+        self.topics = topics
+
+        self.zk = ZookeeperService(test_context, self.num_zk) if 
quorum.for_test(test_context) == quorum.zk else None

Review Comment:
   I'm just preserving the code that is already there (in KafkaTest). the 
quorum parameter is still everywhere in the streams tests. I can push a 
clean-up PR, for this PR, I just preserved what's tehre.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to