BDD test added for compression in sha extractro

Jakub007d · Feb 13, 2024 · 22d63bb · 22d63bb
1 parent 55419bb
commit 22d63bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/features/src/kafka_util.py b/features/src/kafka_util.py
@@ -85,7 +85,7 @@ def consume_event(bootstrap, topic, group_id=None):
     consumer.subscribe(topics=topic)
     return consumer.poll()
 
-def consume_one_message_from_topic(bootsrap, topic):
+def consume_message_from_topic(bootsrap, topic):
     """Consume one messages in given topic."""
     consumer = KafkaConsumer(
         topic,

diff --git a/features/steps/insights_sha_extractor.py b/features/steps/insights_sha_extractor.py
@@ -234,7 +234,7 @@ def compressed_archive_sent_to_topic(context):
     """Check that sha extractor did not process any event."""
     decoded = None
     error= None
-    message = kafka_util.consume_one_message_from_topic(context.kafka_hostname, context.outgoing_topic)
+    message = kafka_util.consume_message_from_topic(context.kafka_hostname, context.outgoing_topic)
     try:
         decoded = gzip.decompress(message.value)
     except Exception as err:
@@ -249,7 +249,7 @@ def no_compressed_archive_sent_to_topic(context):
     """Check that sha extractor did not process any event."""
     decoded = None
     error= None
-    message = kafka_util.consume_one_message_from_topic(context.kafka_hostname, context.outgoing_topic)
+    message = kafka_util.consume_message_from_topic(context.kafka_hostname, context.outgoing_topic)
     try:
         decoded = gzip.decompress(message.value)
     except Exception as err: