9
9
import static io .opentelemetry .api .trace .SpanKind .CLIENT ;
10
10
import static io .opentelemetry .sdk .testing .assertj .OpenTelemetryAssertions .equalTo ;
11
11
import static io .opentelemetry .sdk .testing .assertj .OpenTelemetryAssertions .satisfies ;
12
+ import static io .opentelemetry .semconv .HttpAttributes .HTTP_REQUEST_METHOD ;
13
+ import static io .opentelemetry .semconv .HttpAttributes .HTTP_RESPONSE_STATUS_CODE ;
14
+ import static io .opentelemetry .semconv .NetworkAttributes .NETWORK_PROTOCOL_VERSION ;
15
+ import static io .opentelemetry .semconv .ServerAttributes .SERVER_ADDRESS ;
16
+ import static io .opentelemetry .semconv .ServerAttributes .SERVER_PORT ;
17
+ import static io .opentelemetry .semconv .UrlAttributes .URL_FULL ;
18
+ import static io .opentelemetry .semconv .incubating .AwsIncubatingAttributes .AWS_REQUEST_ID ;
19
+ import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MESSAGING_DESTINATION_NAME ;
20
+ import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MESSAGING_MESSAGE_ID ;
21
+ import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MESSAGING_OPERATION ;
22
+ import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MESSAGING_SYSTEM ;
23
+ import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MessagingSystemIncubatingValues .AWS_SQS ;
24
+ import static io .opentelemetry .semconv .incubating .RpcIncubatingAttributes .RPC_METHOD ;
25
+ import static io .opentelemetry .semconv .incubating .RpcIncubatingAttributes .RPC_SERVICE ;
26
+ import static io .opentelemetry .semconv .incubating .RpcIncubatingAttributes .RPC_SYSTEM ;
12
27
import static org .assertj .core .api .Assertions .assertThat ;
13
28
14
29
import io .opentelemetry .api .trace .SpanKind ;
15
30
import io .opentelemetry .sdk .testing .assertj .AttributeAssertion ;
16
31
import io .opentelemetry .sdk .testing .assertj .SpanDataAssert ;
17
- import io .opentelemetry .semconv .HttpAttributes ;
18
- import io .opentelemetry .semconv .NetworkAttributes ;
19
- import io .opentelemetry .semconv .ServerAttributes ;
20
- import io .opentelemetry .semconv .UrlAttributes ;
21
- import io .opentelemetry .semconv .incubating .AwsIncubatingAttributes ;
22
- import io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes ;
23
- import io .opentelemetry .semconv .incubating .RpcIncubatingAttributes ;
24
32
import java .util .ArrayList ;
25
33
import java .util .Arrays ;
26
34
import java .util .List ;
@@ -72,53 +80,38 @@ static SpanDataAssert sqs(
72
80
val ->
73
81
val .satisfiesAnyOf (
74
82
v -> assertThat (v ).isEqualTo (queueUrl ), v -> assertThat (v ).isNull ())),
83
+ satisfies (AWS_REQUEST_ID , val -> val .isInstanceOf (String .class )),
84
+ equalTo (HTTP_REQUEST_METHOD , "POST" ),
85
+ equalTo (HTTP_RESPONSE_STATUS_CODE , 200 ),
86
+ satisfies (URL_FULL , val -> val .isInstanceOf (String .class )),
87
+ satisfies (SERVER_ADDRESS , stringAssert -> stringAssert .isInstanceOf (String .class )),
75
88
satisfies (
76
- AwsIncubatingAttributes .AWS_REQUEST_ID , val -> val .isInstanceOf (String .class )),
77
- equalTo (HttpAttributes .HTTP_REQUEST_METHOD , "POST" ),
78
- equalTo (HttpAttributes .HTTP_RESPONSE_STATUS_CODE , 200 ),
79
- satisfies (UrlAttributes .URL_FULL , val -> val .isInstanceOf (String .class )),
80
- satisfies (
81
- ServerAttributes .SERVER_ADDRESS ,
82
- stringAssert -> stringAssert .isInstanceOf (String .class )),
83
- satisfies (
84
- ServerAttributes .SERVER_PORT ,
89
+ SERVER_PORT ,
85
90
val ->
86
91
val .satisfiesAnyOf (
87
92
v -> assertThat (v ).isNull (),
88
93
v -> assertThat (v ).isInstanceOf (Number .class ))),
89
- equalTo (NetworkAttributes .NETWORK_PROTOCOL_VERSION , "1.1" ),
90
- equalTo (RpcIncubatingAttributes .RPC_SYSTEM , "aws-api" ),
91
- satisfies (
92
- RpcIncubatingAttributes .RPC_METHOD ,
93
- stringAssert -> stringAssert .isEqualTo (rpcMethod )),
94
- equalTo (RpcIncubatingAttributes .RPC_SERVICE , "AmazonSQS" )));
94
+ equalTo (NETWORK_PROTOCOL_VERSION , "1.1" ),
95
+ equalTo (RPC_SYSTEM , "aws-api" ),
96
+ satisfies (RPC_METHOD , stringAssert -> stringAssert .isEqualTo (rpcMethod )),
97
+ equalTo (RPC_SERVICE , "AmazonSQS" )));
95
98
96
99
if (spanName .endsWith ("receive" )
97
100
|| spanName .endsWith ("process" )
98
101
|| spanName .endsWith ("publish" )) {
99
102
attributeAssertions .addAll (
100
103
Arrays .asList (
101
- equalTo (MessagingIncubatingAttributes .MESSAGING_DESTINATION_NAME , queueName ),
102
- equalTo (
103
- MessagingIncubatingAttributes .MESSAGING_SYSTEM ,
104
- MessagingIncubatingAttributes .MessagingSystemIncubatingValues .AWS_SQS )));
104
+ equalTo (MESSAGING_DESTINATION_NAME , queueName ), equalTo (MESSAGING_SYSTEM , AWS_SQS )));
105
105
if (spanName .endsWith ("receive" )) {
106
- attributeAssertions .add (
107
- equalTo (MessagingIncubatingAttributes .MESSAGING_OPERATION , "receive" ));
106
+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "receive" ));
108
107
} else if (spanName .endsWith ("process" )) {
108
+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "process" ));
109
109
attributeAssertions .add (
110
- equalTo (MessagingIncubatingAttributes .MESSAGING_OPERATION , "process" ));
111
- attributeAssertions .add (
112
- satisfies (
113
- MessagingIncubatingAttributes .MESSAGING_MESSAGE_ID ,
114
- val -> assertThat (val ).isNotNull ()));
110
+ satisfies (MESSAGING_MESSAGE_ID , val -> assertThat (val ).isNotNull ()));
115
111
} else if (spanName .endsWith ("publish" )) {
112
+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "publish" ));
116
113
attributeAssertions .add (
117
- equalTo (MessagingIncubatingAttributes .MESSAGING_OPERATION , "publish" ));
118
- attributeAssertions .add (
119
- satisfies (
120
- MessagingIncubatingAttributes .MESSAGING_MESSAGE_ID ,
121
- val -> assertThat (val ).isNotNull ()));
114
+ satisfies (MESSAGING_MESSAGE_ID , val -> assertThat (val ).isNotNull ()));
122
115
}
123
116
}
124
117
@@ -133,16 +126,16 @@ static SpanDataAssert s3(SpanDataAssert span, String spanName, String bucketName
133
126
equalTo (stringKey ("aws.agent" ), "java-aws-sdk" ),
134
127
satisfies (stringKey ("aws.endpoint" ), val -> val .isInstanceOf (String .class )),
135
128
equalTo (stringKey ("aws.bucket.name" ), bucketName ),
136
- equalTo (RpcIncubatingAttributes . RPC_SYSTEM , "aws-api" ),
137
- equalTo (RpcIncubatingAttributes . RPC_METHOD , spanName .substring (3 )),
138
- equalTo (RpcIncubatingAttributes . RPC_SERVICE , "Amazon S3" ),
139
- equalTo (HttpAttributes . HTTP_REQUEST_METHOD , method ),
140
- equalTo (HttpAttributes . HTTP_RESPONSE_STATUS_CODE , 200 ),
141
- satisfies (UrlAttributes . URL_FULL , val -> val .isInstanceOf (String .class )),
142
- equalTo (NetworkAttributes . NETWORK_PROTOCOL_VERSION , "1.1" ),
143
- satisfies (ServerAttributes . SERVER_ADDRESS , val -> val .isInstanceOf (String .class )),
129
+ equalTo (RPC_SYSTEM , "aws-api" ),
130
+ equalTo (RPC_METHOD , spanName .substring (3 )),
131
+ equalTo (RPC_SERVICE , "Amazon S3" ),
132
+ equalTo (HTTP_REQUEST_METHOD , method ),
133
+ equalTo (HTTP_RESPONSE_STATUS_CODE , 200 ),
134
+ satisfies (URL_FULL , val -> val .isInstanceOf (String .class )),
135
+ equalTo (NETWORK_PROTOCOL_VERSION , "1.1" ),
136
+ satisfies (SERVER_ADDRESS , val -> val .isInstanceOf (String .class )),
144
137
satisfies (
145
- ServerAttributes . SERVER_PORT ,
138
+ SERVER_PORT ,
146
139
val ->
147
140
val .satisfiesAnyOf (
148
141
v -> val .isInstanceOf (Number .class ), v -> assertThat (v ).isNull ())));
@@ -154,19 +147,18 @@ static SpanDataAssert sns(SpanDataAssert span, String spanName, String topicArn)
154
147
.hasAttributesSatisfyingExactly (
155
148
equalTo (stringKey ("aws.agent" ), "java-aws-sdk" ),
156
149
satisfies (stringKey ("aws.endpoint" ), val -> val .isInstanceOf (String .class )),
150
+ satisfies (AWS_REQUEST_ID , val -> val .isInstanceOf (String .class )),
151
+ equalTo (RPC_SYSTEM , "aws-api" ),
152
+ equalTo (RPC_METHOD , spanName .substring (4 )),
153
+ equalTo (RPC_SERVICE , "AmazonSNS" ),
154
+ equalTo (MESSAGING_DESTINATION_NAME , topicArn ),
155
+ equalTo (HTTP_REQUEST_METHOD , "POST" ),
156
+ equalTo (HTTP_RESPONSE_STATUS_CODE , 200 ),
157
+ satisfies (URL_FULL , val -> val .isInstanceOf (String .class )),
158
+ equalTo (NETWORK_PROTOCOL_VERSION , "1.1" ),
159
+ satisfies (SERVER_ADDRESS , val -> val .isInstanceOf (String .class )),
157
160
satisfies (
158
- AwsIncubatingAttributes .AWS_REQUEST_ID , val -> val .isInstanceOf (String .class )),
159
- equalTo (RpcIncubatingAttributes .RPC_SYSTEM , "aws-api" ),
160
- equalTo (RpcIncubatingAttributes .RPC_METHOD , spanName .substring (4 )),
161
- equalTo (RpcIncubatingAttributes .RPC_SERVICE , "AmazonSNS" ),
162
- equalTo (MessagingIncubatingAttributes .MESSAGING_DESTINATION_NAME , topicArn ),
163
- equalTo (HttpAttributes .HTTP_REQUEST_METHOD , "POST" ),
164
- equalTo (HttpAttributes .HTTP_RESPONSE_STATUS_CODE , 200 ),
165
- satisfies (UrlAttributes .URL_FULL , val -> val .isInstanceOf (String .class )),
166
- equalTo (NetworkAttributes .NETWORK_PROTOCOL_VERSION , "1.1" ),
167
- satisfies (ServerAttributes .SERVER_ADDRESS , val -> val .isInstanceOf (String .class )),
168
- satisfies (
169
- ServerAttributes .SERVER_PORT ,
161
+ SERVER_PORT ,
170
162
val ->
171
163
val .satisfiesAnyOf (
172
164
v -> val .isInstanceOf (Number .class ), v -> assertThat (v ).isNull ())));
0 commit comments