add tests for KinesisFirehoseEvent

Author: ryandeivert

tests/events/kinesisFirehoseEvent.json

@@ -0,0 +1,31 @@
+{
+  "invocationId": "2b4d1ad9-2f48-94bd-a088-767c317e994a",
+  "deliveryStreamArn": "arn:aws:firehose:us-east-2:123456789012:deliverystream/delivery-stream-name",
+  "region": "us-east-2",
+  "records": [
+    {
+      "data": "SGVsbG8gV29ybGQ=",
+      "recordId": "record1",
+      "approximateArrivalTimestamp": 1510772160000,
+      "kinesisRecordMetadata": {
+        "shardId": "shardId-000000000000",
+        "partitionKey": "4d1ad2b9-24f8-4b9d-a088-76e9947c317a",
+        "approximateArrivalTimestamp": "2012-04-23T18:25:43.511Z",
+        "sequenceNumber": "49546986683135544286507457936321625675700192471156785154",
+        "subsequenceNumber": ""
+      }
+    },
+    {
+      "data": "eyJIZWxsbyI6ICJXb3JsZCJ9",
+      "recordId": "record2",
+      "approximateArrivalTimestamp": 151077216000,
+      "kinesisRecordMetadata": {
+        "shardId": "shardId-000000000001",
+        "partitionKey": "4d1ad2b9-24f8-4b9d-a088-76e9947c318a",
+        "approximateArrivalTimestamp": "2012-04-23T19:25:43.511Z",
+        "sequenceNumber": "49546986683135544286507457936321625675700192471156785155",
+        "subsequenceNumber": ""
+      }
+    }
+  ]
+}

tests/functional/test_data_classes.py

@@ -18,6 +18,7 @@
     CodePipelineJobEvent,
     EventBridgeEvent,
     KafkaEvent,
+    KinesisFirehoseEvent,
     KinesisStreamEvent,
     S3Event,
     SESEvent,
@@ -1201,6 +1202,30 @@ def test_kafka_self_managed_event():
     assert record.get_header_value("HeaderKey", case_sensitive=False) == b"headerValue"
 
 
+def test_kinesis_firehose_event():
+    event = KinesisFirehoseEvent(load_event("kinesisFirehoseEvent.json"))
+    
+    assert event.region == "us-east-2"
+    assert event.invocation_id == "2b4d1ad9-2f48-94bd-a088-767c317e994a"
+    assert event.delivery_stream_arn == "arn:aws:firehose:us-east-2:123456789012:deliverystream/delivery-stream-name"
+    assert event.source_kinesis_stream_arn is None
+    
+    records = list(event.records)
+    assert len(records) == 2
+    record_01, record_02 = records[:]
+    
+    assert record_01.approximate_arrival_timestamp == 1510772160000
+    assert record_01.record_id == "record1"
+    assert record_01.data == "SGVsbG8gV29ybGQ="
+    assert record_01.data_as_bytes == b'Hello World'
+    assert record_01.data_as_text == 'Hello World'
+    
+    assert record_02.data == "eyJIZWxsbyI6ICJXb3JsZCJ9"
+    assert record_02.data_as_bytes == b'{"Hello": "World"}'
+    assert record_02.data_as_text == '{"Hello": "World"}'
+    assert record_02.data_as_json == {"Hello": "World"}
+
+
 def test_kinesis_stream_event():
     event = KinesisStreamEvent(load_event("kinesisStreamEvent.json"))