fixed trigger tags and tests

Author: avedmala

datadog_lambda/tracing.py

@@ -403,8 +403,10 @@ def extract_context_from_step_functions(event, lambda_context):
 
         if dd_data and dd_data.get("serverless-version") == "v2":
             if "x-datadog-trace-id" in dd_data:  # lambda root
-                trace_id = dd_data.get("x-datadog-trace-id")
-                high_64_bit_trace_id = _parse_high_64_bits(dd_data.get("x-datadog-tags"))
+                trace_id = int(dd_data.get("x-datadog-trace-id"))
+                high_64_bit_trace_id = _parse_high_64_bits(
+                    dd_data.get("x-datadog-tags")
+                )
                 if high_64_bit_trace_id:
                     meta["_dd.p.tid"] = high_64_bit_trace_id
             else:  # sfn root
@@ -709,7 +711,6 @@ def create_inferred_span(
     event_source: _EventSource = None,
     decode_authorizer_context: bool = True,
 ):
-    logger.debug("abhinav event %s", event)
     if event_source is None:
         event_source = parse_event_source(event)
     try:

datadog_lambda/trigger.py

@@ -256,6 +256,14 @@ def parse_event_source_arn(source: _EventSource, event: dict, context: Any) -> s
     if source.event_type == EventTypes.CLOUDWATCH_EVENTS and event.get("resources"):
         return event.get("resources")[0]
 
+    # todo: testme
+    # e.g. arn:aws:states:us-east-1:123456789012:stateMachine:stateMachineName
+    if source.event_type == EventTypes.STEPFUNCTIONS:
+        context = event
+        if "_datadog" in event:
+            context = event.get("_datadog")
+        return context.get("StateMachine").get("Id")
+
 
 def get_event_source_arn(source: _EventSource, event: dict, context: Any) -> str:
     event_source_arn = event.get("eventSourceARN") or event.get("eventSourceArn")