Add test for async

Author: trask

WORK_IN_PROGRESS.txt

@@ -0,0 +1,3 @@
+
+./gradlew :instrumentation:spring:spring-webmvc:spring-webmvc-3.1:javaagent:test --tests SpringBootBasedTest.deferredResult
+

instrumentation/spring/spring-webmvc/spring-webmvc-common/testing/src/main/java/io/opentelemetry/instrumentation/spring/webmvc/boot/AbstractSpringBootBasedTest.java

@@ -9,6 +9,7 @@
 import static io.opentelemetry.instrumentation.testing.junit.code.SemconvCodeStabilityUtil.codeFunctionSuffixAssertions;
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.AUTH_ERROR;
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.CAPTURE_HEADERS;
+import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.DEFERRED_RESULT;
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.EXCEPTION;
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.INDEXED_CHILD;
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.LOGIN;
@@ -144,6 +145,33 @@ void testCharacterEncodingOfTestPassword(String testPassword) {
                             .hasKind(SpanKind.INTERNAL)));
   }
 
+  @Test
+  void deferredResult() {
+    AggregatedHttpResponse response =
+        client.execute(request(DEFERRED_RESULT, "GET")).aggregate().join();
+
+    assertThat(response.status().code()).isEqualTo(DEFERRED_RESULT.getStatus());
+    assertThat(response.contentUtf8()).isEqualTo(DEFERRED_RESULT.getBody());
+
+    testing()
+        .waitAndAssertTraces(
+            trace ->
+                trace.hasSpansSatisfyingExactly(
+                    span -> {
+                      assertServerSpan(span, "GET", DEFERRED_RESULT, DEFERRED_RESULT.getStatus());
+                      span.hasNoParent();
+                    },
+                    span -> assertHandlerSpan(span, "GET", DEFERRED_RESULT).hasParent(trace.getSpan(0))),
+            trace ->
+                trace.hasSpansSatisfyingExactly(
+                    span ->
+                        span.hasName("deferred-result-child")
+                            .hasKind(SpanKind.INTERNAL)
+                            .hasNoParent()
+                            .hasTotalAttributeCount(0),
+                    span -> assertHandlerSpan(span, "GET", DEFERRED_RESULT).hasParent(trace.getSpan(0))));
+  }
+
   @Override
   protected List<Consumer<SpanDataAssert>> errorPageSpanAssertions(
       String method, ServerEndpoint endpoint) {
@@ -228,6 +256,8 @@ private static String getHandlerSpanName(ServerEndpoint endpoint) {
       return "TestController.captureHeaders";
     } else if (INDEXED_CHILD.equals(endpoint)) {
       return "TestController.indexedChild";
+    } else if (DEFERRED_RESULT.equals(endpoint)) {
+      return "TestController.deferredResult";
     }
     return "TestController." + endpoint.name().toLowerCase(Locale.ROOT);
   }

instrumentation/spring/spring-webmvc/spring-webmvc-common/testing/src/main/java/io/opentelemetry/instrumentation/spring/webmvc/boot/AppConfig.java

@@ -6,6 +6,8 @@
 package io.opentelemetry.instrumentation.spring.webmvc.boot;
 
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.scheduling.annotation.EnableAsync;
 
 @SpringBootApplication
+@EnableAsync
 public class AppConfig {}

instrumentation/spring/spring-webmvc/spring-webmvc-common/testing/src/main/java/io/opentelemetry/instrumentation/spring/webmvc/boot/TestBean.java

@@ -0,0 +1,32 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.spring.webmvc.boot;
+
+import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.DEFERRED_RESULT;
+
+import io.opentelemetry.api.GlobalOpenTelemetry;
+import io.opentelemetry.api.trace.Span;
+import io.opentelemetry.api.trace.Tracer;
+import io.opentelemetry.context.Scope;
+import org.springframework.scheduling.annotation.Async;
+import org.springframework.stereotype.Service;
+import org.springframework.web.context.request.async.DeferredResult;
+
+@Service
+public class TestBean {
+
+  private static final Tracer tracer = GlobalOpenTelemetry.getTracer("test");
+
+  @Async
+  public void asyncDependencyCall(DeferredResult<String> deferredResult) {
+    Span span = tracer.spanBuilder("deferred-result-child").startSpan();
+    try (Scope ignored = span.makeCurrent()) {
+      deferredResult.setResult(DEFERRED_RESULT.getBody());
+    } finally {
+      span.end();
+    }
+  }
+}

instrumentation/spring/spring-webmvc/spring-webmvc-common/testing/src/main/java/io/opentelemetry/instrumentation/spring/webmvc/boot/TestController.java

@@ -16,6 +16,7 @@
 import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.SUCCESS;
 
 import java.util.Objects;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
@@ -25,11 +26,14 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.context.request.async.DeferredResult;
 import org.springframework.web.servlet.view.RedirectView;
 
 @Controller
 public class TestController {
 
+  @Autowired private TestBean testBean;
+
   @RequestMapping("/basicsecured/endpoint")
   @ResponseBody
   String secureEndpoint() {
@@ -100,6 +104,14 @@ String indexedChild(@RequestParam("id") String id) {
         });
   }
 
+  @RequestMapping("/deferred-result")
+  @ResponseBody
+  DeferredResult<String> deferredResult() {
+    DeferredResult<String> deferredResult = new DeferredResult<>();
+    testBean.asyncDependencyCall(deferredResult);
+    return deferredResult;
+  }
+
   @ExceptionHandler
   ResponseEntity<String> handleException(Throwable throwable) {
     return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR.value())

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/junit/http/ServerEndpoint.java

@@ -48,6 +48,8 @@ public class ServerEndpoint {
       new ServerEndpoint("AUTH_ERROR", "basicsecured/endpoint", 401, null);
   public static final ServerEndpoint INDEXED_CHILD =
       new ServerEndpoint("INDEXED_CHILD", "child", 200, "success");
+  public static final ServerEndpoint DEFERRED_RESULT =
+      new ServerEndpoint("DEFERRED_RESULT", "deferred-result", 200, "deferred result");
 
   public static final String ID_ATTRIBUTE_NAME = "test.request.id";
   public static final String ID_PARAMETER_NAME = "id";