feat(data-classes): adding more tests to kafka events

Author: leandrodamascena

tests/events/kafkaEvent.json renamed to tests/events/kafkaEventMsk.json

@@ -0,0 +1,34 @@
+{
+  "eventSource":"aws:aws:SelfManagedKafka",
+  "bootstrapServers":"b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
+  "records":{
+     "mytopic-0":[
+        {
+           "topic":"mytopic",
+           "partition":0,
+           "offset":15,
+           "timestamp":1545084650987,
+           "timestampType":"CREATE_TIME",
+           "key":"cmVjb3JkS2V5",
+           "value":"eyJrZXkiOiJ2YWx1ZSJ9",
+           "headers":[
+              {
+                 "headerKey":[
+                    104,
+                    101,
+                    97,
+                    100,
+                    101,
+                    114,
+                    86,
+                    97,
+                    108,
+                    117,
+                    101
+                 ]
+              }
+           ]
+        }
+     ]
+  }
+}

tests/events/kafkaEventSelfManaged.json

@@ -1139,15 +1139,52 @@ def test_base_proxy_event_json_body_with_base64_encoded_data():
     assert event.json_body == data
 
 
-def test_kafka_event():
-    event = KafkaEvent(load_event("kafkaEvent.json"))
+def test_kafka_msk_event():
+    event = KafkaEvent(load_event("kafkaEventMsk.json"))
     assert event.event_source == "aws:kafka"
-    assert event.event_source_arn == "arn:aws:kafka:us-east-1:0123456789019:cluster/SalesCluster/abcd1234-abcd-cafe-abab-9876543210ab-4"
-    bootstrap_servers = [
+    assert (
+        event.event_source_arn
+        == "arn:aws:kafka:us-east-1:0123456789019:cluster/SalesCluster/abcd1234-abcd-cafe-abab-9876543210ab-4"
+    )
+
+    bootstrap_servers_raw = "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092"  # noqa E501
+
+    bootstrap_servers_list = [
         "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
         "b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
     ]
-    assert event.decoded_bootstrap_servers == bootstrap_servers
+
+    assert event.bootstrap_servers == bootstrap_servers_raw
+    assert event.decoded_bootstrap_servers == bootstrap_servers_list
+
+    records = list(event.records)
+    assert len(records) == 1
+    record = records[0]
+    assert record.topic == "mytopic"
+    assert record.partition == 0
+    assert record.offset == 15
+    assert record.timestamp == 1545084650987
+    assert record.timestamp_type == "CREATE_TIME"
+    assert record.decoded_key == b"recordKey"
+    assert record.value == "eyJrZXkiOiJ2YWx1ZSJ9"
+    assert record.json_value == {"key": "value"}
+    assert record.decoded_headers == {"headerKey": b"headerValue"}
+    assert record.get_header_value("HeaderKey", case_sensitive=False) == b"headerValue"
+
+
+def test_kafka_self_managed_event():
+    event = KafkaEvent(load_event("kafkaEventSelfManaged.json"))
+    assert event.event_source == "aws:aws:SelfManagedKafka"
+
+    bootstrap_servers_raw = "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092"  # noqa E501
+
+    bootstrap_servers_list = [
+        "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
+        "b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
+    ]
+
+    assert event.bootstrap_servers == bootstrap_servers_raw
+    assert event.decoded_bootstrap_servers == bootstrap_servers_list
 
     records = list(event.records)
     assert len(records) == 1