handle a case where the record is some customized item (#616)

Author: joeyzhao2018

datadog_lambda/trigger.py

@@ -153,7 +153,7 @@ def parse_event_source(event: dict) -> _EventSource:
         event_source = _EventSource(EventTypes.STEPFUNCTIONS)
 
     event_record = get_first_record(event)
-    if event_record:
+    if event_record and isinstance(event_record, dict):
         aws_event_source = event_record.get("eventSource") or event_record.get(
             "EventSource"
         )

tests/test_trigger.py

@@ -280,6 +280,13 @@ def test_detect_lambda_function_url_domain_with_invalid_input(self):
         # Test with string that would normally cause an exception when split
         self.assertFalse(detect_lambda_function_url_domain(""))
 
+    def test_event_source_with_non_dict_event_record(self):
+        # Test with event_record that's not a dictionary
+        event = {"Records": "not_a_dict"}
+        event_source = parse_event_source(event)
+        # Should handle the first non-dict record gracefully and return unknown
+        self.assertEqual(event_source.to_string(), "unknown")
+
 
 class GetTriggerTags(unittest.TestCase):
     def test_extract_trigger_tags_api_gateway(self):