mjsax commented on code in PR #19560:
URL: https://github.com/apache/kafka/pull/19560#discussion_r2064260309
##########
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:
Thanks. Might be a good idea to also go over other tests and clean this up,
if it's still there?
I would only expect some upgrade/downgrade tests which use older broker
versions to still use ZK parameter.
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]