account for missing redrive count

Author: avedmala

datadog_lambda/tracing.py

@@ -392,7 +392,7 @@ def _generate_sfn_parent_id(context: dict) -> int:
     version of the Lambda layer that doesn't use this value for its parentID generation.
     """
     execution_id = context.get("Execution").get("Id")
-    redrive_count = context.get("Execution").get("RedriveCount")
+    redrive_count = context.get("Execution").get("RedriveCount", 0)
     state_name = context.get("State").get("Name")
     state_entered_time = context.get("State").get("EnteredTime")
 

tests/test_tracing.py

@@ -623,7 +623,6 @@ def test_step_function_trace_data(self):
                 "Name": "72a7ca3e-901c-41bb-b5a3-5f279b92a316",
                 "RoleArn": "arn:aws:iam::425362996713:role/service-role/StepFunctions-abhinav-activity-state-machine-role-22jpbgl6j",
                 "StartTime": "2024-12-04T19:38:04.069Z",
-                "RedriveCount": 0,
             },
             "State": {
                 "Name": "Lambda Invoke",