check if event is a dict before getting

Author: avedmala

datadog_lambda/tracing.py

@@ -411,6 +411,9 @@ def is_legacy_lambda_step_function(event):
     """
     Check if the event is a step function that called a legacy lambda
     """
+    if not isinstance(event, dict):
+        return False
+
     event = event.get("Payload", {})
     return "Execution" in event and "StateMachine" in event and "State" in event
 

tests/test_tracing.py

@@ -678,6 +678,9 @@ def test_is_legacy_lambda_step_function(self):
         }
         self.assertFalse(is_legacy_lambda_step_function(sf_event))
 
+        other_event = ["foo", "bar"]
+        self.assertFalse(is_legacy_lambda_step_function(other_event))
+
 
 class TestXRayContextConversion(unittest.TestCase):
     def test_convert_xray_trace_id(self):