Skip to content

Commit 1b350b9

Browse files
committed
removed event_type assertion
1 parent eec4743 commit 1b350b9

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/test_tracing.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,12 @@
3636
_convert_xray_trace_id,
3737
_convert_xray_entity_id,
3838
_convert_xray_sampling,
39-
InferredSpanInfo,
4039
create_service_mapping,
4140
determine_service_name,
4241
service_mapping as global_service_mapping,
4342
propagator,
4443
emit_telemetry_on_exception_outside_of_handler,
4544
)
46-
from datadog_lambda.trigger import EventTypes
4745

4846
from tests.utils import get_mock_context
4947

@@ -630,10 +628,9 @@ def _test_step_function_trace_data_common(
630628
TraceHeader.TAGS: f"_dd.p.tid={expected_tid}",
631629
}
632630

633-
ctx, source, event_source = extract_dd_trace_context(event, lambda_ctx)
631+
ctx, source, _ = extract_dd_trace_context(event, lambda_ctx)
634632

635633
self.assertEqual(source, "event")
636-
self.assertEqual(event_source.event_type, EventTypes.STEPFUNCTIONS)
637634
self.assertEqual(ctx, expected_context)
638635
self.assertEqual(get_dd_trace_context(), expected_headers)
639636

0 commit comments

Comments
 (0)