Make headers a tuple to be hashable

aio-libs · Jan 20, 2019 · 70e4af6 · 70e4af6
1 parent 2777c9f
commit 70e4af6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/aiokafka/consumer/fetcher.py b/aiokafka/consumer/fetcher.py
@@ -292,7 +292,7 @@ def _consumer_record(self, tp, record):
         return ConsumerRecord(
             tp.topic, tp.partition, record.offset, record.timestamp,
             record.timestamp_type, key, value, record.checksum,
-            key_size, value_size, record.headers)
+            key_size, value_size, tuple(record.headers))
 
 
 class Fetcher:

diff --git a/tests/test_consumer.py b/tests/test_consumer.py
@@ -1897,4 +1897,4 @@ def test_consumer_with_headers(self):
 
         message = yield from consumer.getone()
         self.assertEqual(message.value, b"0")
-        self.assertEqual(message.headers, [("header1", b"17")])
+        self.assertEqual(message.headers, (("header1", b"17"), ))
diff --git a/tests/test_fetcher.py b/tests/test_fetcher.py
@@ -353,7 +353,7 @@ def _setup_error_after_data(self):
         messages = [ConsumerRecord(
             topic="some_topic", partition=1, offset=0, timestamp=0,
             timestamp_type=0, key=None, value=b"some", checksum=None,
-            serialized_key_size=0, serialized_value_size=4)]
+            serialized_key_size=0, serialized_value_size=4, headers=[])]
         partition_records = PartitionRecords(
             tp2, mock.Mock(), [], 0,
             None, None, False, READ_UNCOMMITTED)