File tree Expand file tree Collapse file tree 1 file changed +33
-0
lines changed Expand file tree Collapse file tree 1 file changed +33
-0
lines changed Original file line number Diff line number Diff line change @@ -191,6 +191,39 @@ def test_with_complete_datadog_trace_headers(self):
191
191
expected_context ,
192
192
)
193
193
194
+ def test_with_w3c_trace_headers (self ):
195
+ lambda_ctx = get_mock_context ()
196
+ ctx , source , event_source = extract_dd_trace_context (
197
+ {
198
+ "headers" : {
199
+ "traceparent" : "00-0000000000000000000000000000007b-0000000000000141-01" ,
200
+ }
201
+ },
202
+ lambda_ctx ,
203
+ )
204
+ self .assertEqual (source , "event" )
205
+ expected_context = Context (
206
+ trace_id = 123 ,
207
+ span_id = 321 ,
208
+ sampling_priority = 1 ,
209
+ meta = {"traceparent" : "00-0000000000000000000000000000007b-0000000000000141-01" },
210
+ )
211
+ self .assertEqual (ctx , expected_context )
212
+ self .assertEqual (
213
+ get_dd_trace_context (),
214
+ Context (
215
+ trace_id = 123 ,
216
+ span_id = fake_xray_header_value_parent_decimal ,
217
+ sampling_priority = 1 ,
218
+ ),
219
+ )
220
+ create_dd_dummy_metadata_subsegment (ctx , XraySubsegment .TRACE_KEY )
221
+ self .mock_send_segment .assert_called ()
222
+ self .mock_send_segment .assert_called_with (
223
+ XraySubsegment .TRACE_KEY ,
224
+ expected_context ,
225
+ )
226
+
194
227
def test_with_extractor_function (self ):
195
228
def extractor_foo (event , context ):
196
229
foo = event .get ("foo" , {})
You can’t perform that action at this time.
0 commit comments