update current serverless version to v1

Author: avedmala

datadog_lambda/tracing.py

@@ -424,7 +424,7 @@ def extract_context_from_step_functions(event, lambda_context):
         meta = {}
         dd_data = event.get("_datadog")
 
-        if dd_data and dd_data.get("serverless-version") == "v2":
+        if dd_data and dd_data.get("serverless-version") == "v1":
             if "x-datadog-trace-id" in dd_data:  # lambda root
                 trace_id = int(dd_data.get("x-datadog-trace-id"))
                 high_64_bit_trace_id = _parse_high_64_bits(

datadog_lambda/trigger.py

@@ -147,7 +147,7 @@ def parse_event_source(event: dict) -> _EventSource:
         event_source = _EventSource(EventTypes.CLOUDWATCH_EVENTS)
 
     if (
-        "_datadog" in event and event.get("_datadog").get("serverless-version") == "v2"
+        "_datadog" in event and event.get("_datadog").get("serverless-version") == "v1"
     ) or ("Execution" in event and "StateMachine" in event and "State" in event):
         event_source = _EventSource(EventTypes.STEPFUNCTIONS)
 

tests/event_samples/states.json

@@ -17,6 +17,6 @@
       "RetryCount": 0
     },
     "RootExecutionId": "arn:aws:states:ca-central-1:425362996713:execution:MyStateMachine-wsx8chv4d:1356a963-42a5-48b0-ba3f-73bde559a50c",
-    "serverless-version": "v2"
+    "serverless-version": "v1"
   }
 }

tests/test_tracing.py

@@ -666,7 +666,7 @@ def test_step_function_trace_data_lambda_root(self):
                 },
                 "x-datadog-trace-id": "5821803790426892636",
                 "x-datadog-tags": "_dd.p.dm=-0,_dd.p.tid=672a7cb100000000",
-                "serverless-version": "v2",
+                "serverless-version": "v1",
             }
         }
         ctx, source, event_source = extract_dd_trace_context(sfn_event, lambda_ctx)
@@ -707,7 +707,7 @@ def test_step_function_trace_data_sfn_root(self):
                     "EnteredTime": "Mon Nov 13 12:43:33 PST 2023",
                 },
                 "RootExecutionId": "4875aba4-ae31-4a4c-bf8a-63e9eee31dad",
-                "serverless-version": "v2",
+                "serverless-version": "v1",
             }
         }
         ctx, source, event_source = extract_dd_trace_context(sfn_event, lambda_ctx)