add more batch processing tests

michael-zhao459 · michael-zhao459 · commit 2f8dfaa668db · 2025-08-06T11:11:00.000-04:00
diff --git a/datadog_lambda/tracing.py b/datadog_lambda/tracing.py
@@ -328,13 +328,11 @@ def extract_context_from_sqs_or_sns_event_or_context(
                             logger.debug(
                                 "Found dd-trace injected trace context from AWSTraceHeader"
                             )
-                            context = Context(
+                            return Context(
                                 trace_id=int(trace_id_parts[2][8:], 16),
                                 span_id=int(x_ray_context["parent_id"], 16),
                                 sampling_priority=float(x_ray_context["sampled"]),
                             )
-                            if not config.data_streams_enabled:
-                                break
         except Exception as e:
             logger.debug("The trace extractor returned with error %s", e)
 

Original file line number	Diff line number	Diff line change
`@@ -328,13 +328,11 @@ def extract_context_from_sqs_or_sns_event_or_context(`
`328`	`328`	`logger.debug(`
`329`	`329`	`"Found dd-trace injected trace context from AWSTraceHeader"`
`330`	`330`	`)`
`331`		`- context = Context(`
	`331`	`+ return Context(`
`332`	`332`	`trace_id=int(trace_id_parts[2][8:], 16),`
`333`	`333`	`span_id=int(x_ray_context["parent_id"], 16),`
`334`	`334`	`sampling_priority=float(x_ray_context["sampled"]),`
`335`	`335`	`)`
`336`		`- if not config.data_streams_enabled:`
`337`		`- break`
`338`	`336`	`except Exception as e:`
`339`	`337`	`logger.debug("The trace extractor returned with error %s", e)`
`340`	`338`