kafka topic partition definition

kafka topic partition definition

Issue-ID: DCAEGEN2-565
Signed-off-by: marekpl <marek.pondel@nokia.com>
Change-Id: Iae72717b3da37c278d399eca4607c9ebd4bb6f75

diff --git a/robotframework-onap/ONAPLibrary/KafkaKeywords.py b/robotframework-onap/ONAPLibrary/KafkaKeywords.py
index f5adce5..6cdf85f 100644 (file)
--- 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())