@@ -18,12 +18,11 @@ import * as assert from 'assert';
18
18
import {
19
19
defaultTextMapGetter ,
20
20
defaultTextMapSetter ,
21
- getSpan ,
22
21
INVALID_SPAN_CONTEXT ,
23
22
ROOT_CONTEXT ,
24
- setSpanContext ,
25
23
SpanContext ,
26
24
TraceFlags ,
25
+ trace ,
27
26
} from '@opentelemetry/api' ;
28
27
import { TraceState } from '@opentelemetry/core' ;
29
28
@@ -50,7 +49,7 @@ describe('AWSXRayPropagator', () => {
50
49
traceFlags : SAMPLED_TRACE_FLAG ,
51
50
} ;
52
51
xrayPropagator . inject (
53
- setSpanContext ( ROOT_CONTEXT , spanContext ) ,
52
+ trace . setSpan ( ROOT_CONTEXT , trace . wrapSpanContext ( spanContext ) ) ,
54
53
carrier ,
55
54
defaultTextMapSetter
56
55
) ;
@@ -68,7 +67,7 @@ describe('AWSXRayPropagator', () => {
68
67
traceFlags : NOT_SAMPLED_TRACE_FLAG ,
69
68
} ;
70
69
xrayPropagator . inject (
71
- setSpanContext ( ROOT_CONTEXT , spanContext ) ,
70
+ trace . setSpan ( ROOT_CONTEXT , trace . wrapSpanContext ( spanContext ) ) ,
72
71
carrier ,
73
72
defaultTextMapSetter
74
73
) ;
@@ -89,7 +88,7 @@ describe('AWSXRayPropagator', () => {
89
88
traceState : traceState ,
90
89
} ;
91
90
xrayPropagator . inject (
92
- setSpanContext ( ROOT_CONTEXT , spanContext ) ,
91
+ trace . setSpan ( ROOT_CONTEXT , trace . wrapSpanContext ( spanContext ) ) ,
93
92
carrier ,
94
93
defaultTextMapSetter
95
94
) ;
@@ -109,7 +108,10 @@ describe('AWSXRayPropagator', () => {
109
108
110
109
it ( 'inject default invalid spanContext - should inject nothing' , ( ) => {
111
110
xrayPropagator . inject (
112
- setSpanContext ( ROOT_CONTEXT , INVALID_SPAN_CONTEXT ) ,
111
+ trace . setSpan (
112
+ ROOT_CONTEXT ,
113
+ trace . wrapSpanContext ( INVALID_SPAN_CONTEXT )
114
+ ) ,
113
115
carrier ,
114
116
defaultTextMapSetter
115
117
) ;
@@ -130,9 +132,11 @@ describe('AWSXRayPropagator', () => {
130
132
it ( 'should extract sampled context' , ( ) => {
131
133
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
132
134
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1' ;
133
- const extractedSpanContext = getSpan (
134
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
135
- ) ?. context ( ) ;
135
+ const extractedSpanContext = trace
136
+ . getSpan (
137
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
138
+ )
139
+ ?. spanContext ( ) ;
136
140
137
141
assert . deepStrictEqual ( extractedSpanContext , {
138
142
traceId : TRACE_ID ,
@@ -145,9 +149,11 @@ describe('AWSXRayPropagator', () => {
145
149
it ( 'should extract sampled context with arbitrary order' , ( ) => {
146
150
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
147
151
'Parent=53995c3f42cd8ad8;Sampled=1;Root=1-8a3c60f7-d188f8fa79d48a391a778fa6' ;
148
- const extractedSpanContext = getSpan (
149
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
150
- ) ?. context ( ) ;
152
+ const extractedSpanContext = trace
153
+ . getSpan (
154
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
155
+ )
156
+ ?. spanContext ( ) ;
151
157
152
158
assert . deepStrictEqual ( extractedSpanContext , {
153
159
traceId : TRACE_ID ,
@@ -160,9 +166,11 @@ describe('AWSXRayPropagator', () => {
160
166
it ( 'should extract context with additional fields' , ( ) => {
161
167
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
162
168
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1;Foo=Bar' ;
163
- const extractedSpanContext = getSpan (
164
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
165
- ) ?. context ( ) ;
169
+ const extractedSpanContext = trace
170
+ . getSpan (
171
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
172
+ )
173
+ ?. spanContext ( ) ;
166
174
167
175
// TODO: assert additional fields when the propagator supports it
168
176
assert . deepStrictEqual ( extractedSpanContext , {
@@ -175,99 +183,119 @@ describe('AWSXRayPropagator', () => {
175
183
176
184
it ( 'extract empty header value - should return undefined' , ( ) => {
177
185
carrier [ AWSXRAY_TRACE_ID_HEADER ] = '' ;
178
- const extractedSpanContext = getSpan (
179
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
180
- ) ?. context ( ) ;
186
+ const extractedSpanContext = trace
187
+ . getSpan (
188
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
189
+ )
190
+ ?. spanContext ( ) ;
181
191
182
192
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
183
193
} ) ;
184
194
185
195
it ( 'extract invalid traceId - should return undefined' , ( ) => {
186
196
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
187
197
'Root=1-abcdefgh-ijklmnopabcdefghijklmnop;Parent=53995c3f42cd8ad8;Sampled=0' ;
188
- const extractedSpanContext = getSpan (
189
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
190
- ) ?. context ( ) ;
198
+ const extractedSpanContext = trace
199
+ . getSpan (
200
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
201
+ )
202
+ ?. spanContext ( ) ;
191
203
192
204
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
193
205
} ) ;
194
206
195
207
it ( 'extract invalid traceId size - should return undefined' , ( ) => {
196
208
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
197
209
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa600;Parent=53995c3f42cd8ad8;Sampled=0' ;
198
- const extractedSpanContext = getSpan (
199
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
200
- ) ?. context ( ) ;
210
+ const extractedSpanContext = trace
211
+ . getSpan (
212
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
213
+ )
214
+ ?. spanContext ( ) ;
201
215
202
216
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
203
217
} ) ;
204
218
205
219
it ( 'extract invalid traceId delimiter - should return undefined' , ( ) => {
206
220
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
207
221
'Root=1*8a3c60f7+d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1;Foo=Bar' ;
208
- const extractedSpanContext = getSpan (
209
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
210
- ) ?. context ( ) ;
222
+ const extractedSpanContext = trace
223
+ . getSpan (
224
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
225
+ )
226
+ ?. spanContext ( ) ;
211
227
212
228
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
213
229
} ) ;
214
230
215
231
it ( 'extract invalid spanId - should return undefined' , ( ) => {
216
232
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
217
233
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=abcdefghijklmnop;Sampled=0' ;
218
- const extractedSpanContext = getSpan (
219
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
220
- ) ?. context ( ) ;
234
+ const extractedSpanContext = trace
235
+ . getSpan (
236
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
237
+ )
238
+ ?. spanContext ( ) ;
221
239
222
240
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
223
241
} ) ;
224
242
225
243
it ( 'extract invalid spanId size - should return undefined' , ( ) => {
226
244
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
227
245
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad800;Sampled=0' ;
228
- const extractedSpanContext = getSpan (
229
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
230
- ) ?. context ( ) ;
246
+ const extractedSpanContext = trace
247
+ . getSpan (
248
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
249
+ )
250
+ ?. spanContext ( ) ;
231
251
232
252
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
233
253
} ) ;
234
254
235
255
it ( 'extract invalid traceFlags - should return undefined' , ( ) => {
236
256
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
237
257
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=' ;
238
- const extractedSpanContext = getSpan (
239
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
240
- ) ?. context ( ) ;
258
+ const extractedSpanContext = trace
259
+ . getSpan (
260
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
261
+ )
262
+ ?. spanContext ( ) ;
241
263
242
264
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
243
265
} ) ;
244
266
245
267
it ( 'extract invalid traceFlags length - should return undefined' , ( ) => {
246
268
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
247
269
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=10220' ;
248
- const extractedSpanContext = getSpan (
249
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
250
- ) ?. context ( ) ;
270
+ const extractedSpanContext = trace
271
+ . getSpan (
272
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
273
+ )
274
+ ?. spanContext ( ) ;
251
275
252
276
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
253
277
} ) ;
254
278
255
279
it ( 'extract nonnumeric invalid traceFlags - should return undefined' , ( ) => {
256
280
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
257
281
'Root=1-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=a' ;
258
- const extractedSpanContext = getSpan (
259
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
260
- ) ?. context ( ) ;
282
+ const extractedSpanContext = trace
283
+ . getSpan (
284
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
285
+ )
286
+ ?. spanContext ( ) ;
261
287
262
288
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
263
289
} ) ;
264
290
265
291
it ( 'extract invalid aws xray version - should return undefined' , ( ) => {
266
292
carrier [ AWSXRAY_TRACE_ID_HEADER ] =
267
293
'Root=2-8a3c60f7-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1' ;
268
- const extractedSpanContext = getSpan (
269
- xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
270
- ) ?. context ( ) ;
294
+ const extractedSpanContext = trace
295
+ . getSpan (
296
+ xrayPropagator . extract ( ROOT_CONTEXT , carrier , defaultTextMapGetter )
297
+ )
298
+ ?. spanContext ( ) ;
271
299
272
300
assert . deepStrictEqual ( extractedSpanContext , undefined ) ;
273
301
} ) ;
0 commit comments