5
5
6
6
package io .opentelemetry .contrib .awsxray .propagator ;
7
7
8
+ import static io .opentelemetry .contrib .awsxray .propagator .AwsXrayPropagator .TRACE_HEADER_KEY ;
9
+ import static org .assertj .core .api .Assertions .assertThat ;
10
+
8
11
import io .opentelemetry .api .trace .Span ;
9
12
import io .opentelemetry .api .trace .SpanContext ;
10
13
import io .opentelemetry .api .trace .TraceFlags ;
15
18
import io .opentelemetry .sdk .trace .ReadableSpan ;
16
19
import io .opentelemetry .sdk .trace .SdkTracerProvider ;
17
20
import io .opentelemetry .sdk .trace .data .LinkData ;
21
+ import java .util .Collections ;
22
+ import java .util .Map ;
18
23
import org .junit .jupiter .api .BeforeEach ;
19
24
import org .junit .jupiter .api .Test ;
20
25
import org .junit .jupiter .api .extension .ExtendWith ;
23
28
import uk .org .webcompere .systemstubs .jupiter .SystemStubsExtension ;
24
29
import uk .org .webcompere .systemstubs .properties .SystemProperties ;
25
30
26
- import java .util .Collections ;
27
- import java .util .Map ;
28
-
29
- import static io .opentelemetry .contrib .awsxray .propagator .AwsXrayPropagator .TRACE_HEADER_KEY ;
30
- import static org .assertj .core .api .Assertions .assertThat ;
31
-
32
31
@ ExtendWith (SystemStubsExtension .class )
33
32
class AwsXrayLambdaPropagatorTest extends AwsXrayPropagatorTest {
34
33
@@ -54,7 +53,7 @@ void extract_fromEnvironmentVariable() {
54
53
"Root=1-00000001-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1;Foo=Bar" );
55
54
56
55
assertThat (
57
- getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
56
+ getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
58
57
.isEqualTo (
59
58
SpanContext .createFromRemoteParent (
60
59
"00000001d188f8fa79d48a391a778fa6" ,
@@ -70,7 +69,7 @@ void extract_fromSystemProperty() {
70
69
"Root=1-00000001-d188f8fa79d48a391a778fa6;Parent=53995c3f42cd8ad8;Sampled=1;Foo=Bar" );
71
70
72
71
assertThat (
73
- getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
72
+ getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
74
73
.isEqualTo (
75
74
SpanContext .createFromRemoteParent (
76
75
"00000001d188f8fa79d48a391a778fa6" ,
@@ -89,7 +88,7 @@ void extract_systemPropertyBeforeEnvironmentVariable() {
89
88
"Root=1-00000002-240000000000000000000002;Parent=1600000000000002;Sampled=1;Foo=Baz" );
90
89
91
90
assertThat (
92
- getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
91
+ getSpanContext (propagator ().extract (Context .current (), Collections .emptyMap (), GETTER )))
93
92
.isEqualTo (
94
93
SpanContext .createFromRemoteParent (
95
94
"00000002240000000000000000000002" ,
@@ -112,10 +111,15 @@ void addLink_SystemProperty() {
112
111
"Root=1-00000003-240000000000000000000003;Parent=1600000000000003;Sampled=1;Foo=Baz" );
113
112
114
113
Context extract = propagator ().extract (Context .current (), carrier , GETTER );
115
- ReadableSpan span = (ReadableSpan ) tracer .spanBuilder ("test" )
116
- .setParent (extract )
117
- .addLink (Span .fromContext (propagator ().extract (extract , carrier , GETTER )).getSpanContext ())
118
- .startSpan ();
114
+ ReadableSpan span =
115
+ (ReadableSpan )
116
+ tracer
117
+ .spanBuilder ("test" )
118
+ .setParent (extract )
119
+ .addLink (
120
+ Span .fromContext (propagator ().extract (extract , carrier , GETTER ))
121
+ .getSpanContext ())
122
+ .startSpan ();
119
123
assertThat (span .getParentSpanContext ())
120
124
.isEqualTo (
121
125
SpanContext .createFromRemoteParent (
@@ -125,12 +129,13 @@ void addLink_SystemProperty() {
125
129
TraceState .getDefault ()));
126
130
127
131
assertThat (span .toSpanData ().getLinks ())
128
- .isEqualTo (Collections .singletonList (LinkData .create (
129
- SpanContext .createFromRemoteParent (
130
- "00000001240000000000000000000001" ,
131
- "1600000000000001" ,
132
- TraceFlags .getSampled (),
133
- TraceState .getDefault ()))));
134
-
132
+ .isEqualTo (
133
+ Collections .singletonList (
134
+ LinkData .create (
135
+ SpanContext .createFromRemoteParent (
136
+ "00000001240000000000000000000001" ,
137
+ "1600000000000001" ,
138
+ TraceFlags .getSampled (),
139
+ TraceState .getDefault ()))));
135
140
}
136
141
}
0 commit comments