From: marekpl Date: Tue, 30 Jul 2019 14:16:26 +0000 (+0200) Subject: kafka topic partition definition X-Git-Tag: 6.0.0-ONAP~30 X-Git-Url: https://gerrit.onap.org/r/gitweb?a=commitdiff_plain;h=refs%2Fchanges%2F82%2F92282%2F1;p=testsuite%2Fpython-testing-utils.git kafka topic partition definition kafka topic partition definition Issue-ID: DCAEGEN2-565 Signed-off-by: marekpl Change-Id: Iae72717b3da37c278d399eca4607c9ebd4bb6f75 --- diff --git a/robotframework-onap/ONAPLibrary/KafkaKeywords.py b/robotframework-onap/ONAPLibrary/KafkaKeywords.py index f5adce5..6cdf85f 100644 --- a/robotframework-onap/ONAPLibrary/KafkaKeywords.py +++ b/robotframework-onap/ONAPLibrary/KafkaKeywords.py @@ -91,7 +91,11 @@ class KafkaKeywords(object): group_id=cgn, request_timeout_ms=10001) - partitions = [TopicPartition(str(topic_name), 0), TopicPartition(str(topic_name), 1), TopicPartition(str(topic_name), 2)] + consumer.topics() + partition_set = consumer.partitions_for_topic(str(topic_name)) + partitions = [] + for val in partition_set: + partitions.append(TopicPartition(str(topic_name), val)) consumer.assign(partitions) last = consumer.end_offsets(partitions) offset = max(last.values())