|
5 | 5 |
|
6 | 6 | package io.opentelemetry.instrumentation.awssdk.v2_2; |
7 | 7 |
|
| 8 | +import static io.opentelemetry.api.common.AttributeKey.stringKey; |
8 | 9 | import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.assertThat; |
| 10 | +import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.equalTo; |
| 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.ServerAttributes.SERVER_ADDRESS; |
| 15 | +import static io.opentelemetry.semconv.ServerAttributes.SERVER_PORT; |
9 | 16 | import static io.opentelemetry.semconv.UrlAttributes.URL_FULL; |
| 17 | +import static io.opentelemetry.semconv.incubating.AwsIncubatingAttributes.AWS_REQUEST_ID; |
| 18 | +import static io.opentelemetry.semconv.incubating.RpcIncubatingAttributes.RPC_METHOD; |
| 19 | +import static io.opentelemetry.semconv.incubating.RpcIncubatingAttributes.RPC_SERVICE; |
| 20 | +import static io.opentelemetry.semconv.incubating.RpcIncubatingAttributes.RPC_SYSTEM; |
10 | 21 |
|
11 | 22 | import io.opentelemetry.api.trace.SpanKind; |
12 | 23 | import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension; |
|
32 | 43 | import software.amazon.awssdk.services.ses.model.SendEmailRequest; |
33 | 44 |
|
34 | 45 | @TestInstance(TestInstance.Lifecycle.PER_CLASS) |
| 46 | +@SuppressWarnings("deprecation") // using deprecated semconv |
35 | 47 | public abstract class AbstractQueryProtocolModelTest { |
36 | 48 | private final MockWebServerExtension server = new MockWebServerExtension(); |
37 | 49 |
|
@@ -90,21 +102,29 @@ void testClientWithQueryProtocolModel() { |
90 | 102 | trace.hasSpansSatisfyingExactly( |
91 | 103 | span -> { |
92 | 104 | span.hasKind(SpanKind.CLIENT); |
93 | | - span.hasAttributesSatisfying( |
94 | | - attributes -> { |
95 | | - assertThat(attributes) |
96 | | - .hasEntrySatisfying( |
97 | | - URL_FULL, |
98 | | - entry -> { |
99 | | - assertThat(entry) |
100 | | - .satisfies( |
101 | | - value -> { |
102 | | - URI uri = URI.create(value); |
103 | | - assertThat(uri.getQuery()).isNull(); |
104 | | - }); |
105 | | - }); |
106 | | - }); |
| 105 | + span.hasAttributesSatisfyingExactly( |
| 106 | + equalTo(SERVER_ADDRESS, "127.0.0.1"), |
| 107 | + equalTo(SERVER_PORT, server.httpPort()), |
| 108 | + equalTo(HTTP_REQUEST_METHOD, "POST"), |
| 109 | + equalTo(HTTP_RESPONSE_STATUS_CODE, 200), |
| 110 | + equalTo(RPC_SYSTEM, "aws-api"), |
| 111 | + equalTo(RPC_SERVICE, "Ses"), |
| 112 | + equalTo(RPC_METHOD, "SendEmail"), |
| 113 | + equalTo(AWS_REQUEST_ID, "UNKNOWN"), |
| 114 | + equalTo(stringKey("aws.agent"), isJavaagentTest() ? "java-aws-sdk" : null), |
| 115 | + satisfies( |
| 116 | + URL_FULL, |
| 117 | + val -> |
| 118 | + val.satisfies( |
| 119 | + value -> { |
| 120 | + URI uri = URI.create(value); |
| 121 | + assertThat(uri.getQuery()).isNull(); |
| 122 | + }))); |
107 | 123 | }); |
108 | 124 | }); |
109 | 125 | } |
| 126 | + |
| 127 | + private boolean isJavaagentTest() { |
| 128 | + return getClass().getName().startsWith("io.opentelemetry.javaagent."); |
| 129 | + } |
110 | 130 | } |
0 commit comments