chore: clean up docs example

Signed-off-by: heitorlessa <[email protected]>

Author: heitorlessa

examples/event_sources/src/kinesis_firehose_response.py

@@ -1,25 +1,24 @@
-import base64
-
 from aws_lambda_powertools.utilities.data_classes import (
     KinesisFirehoseDataTransformationRecord,
     KinesisFirehoseDataTransformationResponse,
+    KinesisFirehoseEvent,
 )
 from aws_lambda_powertools.utilities.serialization import base64_from_json
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
 
 def lambda_handler(event: dict, context: LambdaContext):
+    firehose_event = KinesisFirehoseEvent(event)
     result = KinesisFirehoseDataTransformationResponse()
 
-    for record in event["records"]:
-        print(record["recordId"])
-        payload = base64.b64decode(record["data"]).decode("utf-8")
-        ## do all kind of stuff with payload
+    for record in firehose_event.records:
+        payload = record.data_as_text  # base64 decoded data as str
+
         ## generate data to return
         transformed_data = {"tool_used": "powertools_dataclass", "original_payload": payload}
 
         processed_record = KinesisFirehoseDataTransformationRecord(
-            record_id=record["recordId"],
+            record_id=record.record_id,
             data=base64_from_json(transformed_data),
         )
         result.add_record(processed_record)

tests/unit/data_classes/test_kinesis_firehose_response.py

@@ -92,7 +92,6 @@ def test_kinesis_firehose_create_response():
     for record in parsed_event.records:
         metadata_partition = KinesisFirehoseDataTransformationRecordMetadata(partition_keys={"year": "2023"})
         processed_record = record.build_data_transformation_response(
-            result="Ok",
             metadata=metadata_partition,
             data=base64_from_str(arbitrary_data),
         )