diff --git a/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/build.gradle.kts b/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/build.gradle.kts index a833ca05cb..bb7321bb62 100644 --- a/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/build.gradle.kts +++ b/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/build.gradle.kts @@ -11,7 +11,3 @@ dependencies { testImplementation("org.assertj:assertj-core") } - -tasks.withType().configureEach { - jvmArgs("-Dio.opentelemetry.context.enableStrictContext=false") -} diff --git a/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/src/test/java/TracingRequestHandlerTest.java b/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/src/test/java/TracingRequestHandlerTest.java index 821cb3d81e..0f796d8788 100644 --- a/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/src/test/java/TracingRequestHandlerTest.java +++ b/instrumentation/aws-sdk/aws-sdk-1.11/javaagent-unit-tests/src/test/java/TracingRequestHandlerTest.java @@ -25,7 +25,13 @@ class TracingRequestHandlerTest { } private static Request request() { - Request request = new DefaultRequest<>(new SendMessageRequest(), "test"); + // Using a subclass of SendMessageRequest because for SendMessageRequest instrumentation + // creates PRODUCER span, for others CLIENT span. We need to use CLIENT spans for + // runWithClientSpan in shouldNotSetScopeAndNotFailIfClientSpanAlreadyPresent to work. + class CustomSendMessageRequest extends SendMessageRequest {} + + Request request = + new DefaultRequest<>(new CustomSendMessageRequest(), "test"); request.setEndpoint(URI.create("http://test.uri")); return request; } @@ -61,5 +67,8 @@ class TracingRequestHandlerTest { // then - no exception and scope not set assertThat(request.getHandlerContext(TracingRequestHandler.SCOPE)).isNotNull(); underTest.afterResponse(request, response(request)); + + // cleanup + underTest.afterError(request, null, null); } }