99import static io .opentelemetry .api .trace .SpanKind .CLIENT ;
1010import static io .opentelemetry .sdk .testing .assertj .OpenTelemetryAssertions .equalTo ;
1111import 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 ;
1227import static org .assertj .core .api .Assertions .assertThat ;
1328
1429import io .opentelemetry .api .trace .SpanKind ;
1530import io .opentelemetry .sdk .testing .assertj .AttributeAssertion ;
1631import 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 ;
2432import java .util .ArrayList ;
2533import java .util .Arrays ;
2634import java .util .List ;
@@ -72,53 +80,38 @@ static SpanDataAssert sqs(
7280 val ->
7381 val .satisfiesAnyOf (
7482 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 )),
7588 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 ,
8590 val ->
8691 val .satisfiesAnyOf (
8792 v -> assertThat (v ).isNull (),
8893 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" )));
9598
9699 if (spanName .endsWith ("receive" )
97100 || spanName .endsWith ("process" )
98101 || spanName .endsWith ("publish" )) {
99102 attributeAssertions .addAll (
100103 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 )));
105105 if (spanName .endsWith ("receive" )) {
106- attributeAssertions .add (
107- equalTo (MessagingIncubatingAttributes .MESSAGING_OPERATION , "receive" ));
106+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "receive" ));
108107 } else if (spanName .endsWith ("process" )) {
108+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "process" ));
109109 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 ()));
115111 } else if (spanName .endsWith ("publish" )) {
112+ attributeAssertions .add (equalTo (MESSAGING_OPERATION , "publish" ));
116113 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 ()));
122115 }
123116 }
124117
@@ -133,16 +126,16 @@ static SpanDataAssert s3(SpanDataAssert span, String spanName, String bucketName
133126 equalTo (stringKey ("aws.agent" ), "java-aws-sdk" ),
134127 satisfies (stringKey ("aws.endpoint" ), val -> val .isInstanceOf (String .class )),
135128 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 )),
144137 satisfies (
145- ServerAttributes . SERVER_PORT ,
138+ SERVER_PORT ,
146139 val ->
147140 val .satisfiesAnyOf (
148141 v -> val .isInstanceOf (Number .class ), v -> assertThat (v ).isNull ())));
@@ -154,19 +147,18 @@ static SpanDataAssert sns(SpanDataAssert span, String spanName, String topicArn)
154147 .hasAttributesSatisfyingExactly (
155148 equalTo (stringKey ("aws.agent" ), "java-aws-sdk" ),
156149 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 )),
157160 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 ,
170162 val ->
171163 val .satisfiesAnyOf (
172164 v -> val .isInstanceOf (Number .class ), v -> assertThat (v ).isNull ())));
0 commit comments