Fixing the Key decode issue for multi-events (#146)

* fixing the issue

* adding unit test case

* fixing the flake8 issue

Author: shrohilla

azure/functions/kafka.py

@@ -262,8 +262,7 @@ def decode_multiple_events(cls, data: meta.Datum,
             event = KafkaEvent(
                 body=parsed_data[i],
                 timestamp=parsed_timestamp_props[i],
-                key=cls._decode_typed_data(
-                    parsed_key_props[i], python_type=str),
+                key=parsed_key_props[i],
                 partition=parsed_partition_props[i],
                 offset=parsed_offset_props[i],
                 topic=parsed_topic_props[i],

tests/test_kafka.py

@@ -22,10 +22,12 @@ class Kafka(unittest.TestCase):
     MULTIPLE_KAFKA_TIMESTAMP_1 = "2020-06-20T05:06:25.945Z"
     MULTIPLE_KAFKA_DATA_0 = '{"Offset":62,"Partition":1,"Topic":"message",'\
         '"Timestamp":"2020-06-20T05:06:25.139Z","Value":"a", ' \
-                            '"Headers":[{"Key":"test","Value":"1"}]}'
+                            '"Headers":[{"Key":"test","Value":"1"}], ' \
+                            '"Key" : "1"}'
     MULTIPLE_KAFKA_DATA_1 = '{"Offset":63,"Partition":1,"Topic":"message",'\
         '"Timestamp":"2020-06-20T05:06:25.945Z","Value":"a", ' \
-                            '"Headers":[{"Key":"test2","Value":"2"}]}'
+                            '"Headers":[{"Key":"test2","Value":"2"}], ' \
+                            '"Key": "2"}'
 
     def test_kafka_input_type(self):
         check_input_type = (
@@ -238,10 +240,10 @@ def _generate_single_kafka_datum(self, datum_type='string'):
     def _generate_multiple_kafka_data(self, data_type='json'):
         data = '[{"Offset":62,"Partition":1,"Topic":"message",'\
                '"Timestamp":"2020-06-20T05:06:25.139Z","Value":"a",' \
-               ' "Headers":[{"Key":"test","Value":"1"}]},'\
+               ' "Headers":[{"Key":"test","Value":"1"}], "Key": "1"},'\
                ' {"Offset":63,"Partition":1,"Topic":"message",'\
                '"Timestamp":"2020-06-20T05:06:25.945Z","Value":"a", ' \
-               '"Headers":[{"Key":"test2","Value":"2"}]}]'
+               '"Headers":[{"Key":"test2","Value":"2"}], "Key": "2"}]'
         if data_type == 'collection_bytes':
             data = list(
                 map(lambda x: json.dumps(x).encode('utf-8'),
@@ -282,7 +284,7 @@ def _generate_single_trigger_metadatum(self):
         }
 
     def _generate_multiple_trigger_metadata(self):
-        key_array = [None, None]
+        key_array = ["1", "2"]
         partition_array = [1, 2]
         timestamp_array = ["2020-06-20T05:06:25.139Z",
                            "2020-06-20T05:06:25.945Z"]