feat(kotlin): Add failing tests

Author: PerfectSlayer

dd-java-agent/instrumentation/kotlin-coroutines/coroutines-1.3/src/test/kotlin/KotlinCoroutineTests.kt

@@ -68,6 +68,11 @@ class KotlinCoroutineTests(dispatcher: CoroutineDispatcher) : CoreKotlinCoroutin
     return super.traceAfterTimeout()
   }
 
+  @Trace
+  override fun traceAfterDelay(): Int {
+    return super.traceAfterDelay()
+  }
+
   @Trace
   override fun tracedChild(opName: String) {
     super.tracedChild(opName)

dd-java-agent/instrumentation/kotlin-coroutines/coroutines-1.5/src/test/kotlin/KotlinCoroutineTests.kt

@@ -89,6 +89,11 @@ class KotlinCoroutineTests(dispatcher: CoroutineDispatcher) : CoreKotlinCoroutin
     return super.traceAfterTimeout()
   }
 
+  @Trace
+  override fun traceAfterDelay(): Int {
+    return super.traceAfterDelay()
+  }
+
   @Trace
   override fun tracedChild(opName: String) {
     super.tracedChild(opName)

dd-java-agent/instrumentation/kotlin-coroutines/src/testFixtures/groovy/datadog/trace/instrumentation/kotlin/coroutines/AbstractKotlinCoroutineInstrumentationTest.groovy

@@ -376,7 +376,7 @@ abstract class AbstractKotlinCoroutineInstrumentationTest<T extends CoreKotlinCo
     assertTraces(1) {
       trace(expectedNumberOfSpans, true) {
         span(5) {
-          operationName "top-level"
+          operationName "trace.annotation"
           parent()
         }
         span(0) {
@@ -406,6 +406,46 @@ abstract class AbstractKotlinCoroutineInstrumentationTest<T extends CoreKotlinCo
     [dispatcherName, dispatcher] << dispatchersToTest
   }
 
+  @Ignore("Not working: disconnected trace")
+  def "kotlin trace consistent after delay"() {
+    setup:
+    CoreKotlinCoroutineTests kotlinTest = getCoreKotlinCoroutineTestsInstance(dispatcher)
+    int expectedNumberOfSpans = kotlinTest.traceAfterDelay()
+
+    expect:
+    assertTraces(1) {
+      trace(expectedNumberOfSpans, true) {
+        span(5) {
+          operationName "trace.annotation"
+          parent()
+        }
+        span(1) {
+          operationName "before-process"
+          childOf span(5)
+        }
+        span(2) {
+          operationName "process-url-a"
+          childOf span(5)
+        }
+        span(3) {
+          operationName "process-url-b"
+          childOf span(5)
+        }
+        span(4) {
+          operationName "process-url-c"
+          childOf span(5)
+        }
+        span(0) {
+          operationName "after-process"
+          childOf span(5)
+        }
+      }
+    }
+
+    where:
+    [dispatcherName, dispatcher] << dispatchersToTest
+  }
+
   protected static DDSpan findSpan(List<DDSpan> trace, String opName) {
     for (DDSpan span : trace) {
       if (span.getOperationName() == opName) {

dd-java-agent/instrumentation/kotlin-coroutines/src/testFixtures/kotlin/datadog/trace/instrumentation/kotlin/coroutines/CoreKotlinCoroutineTests.kt

@@ -335,6 +335,39 @@ abstract class CoreKotlinCoroutineTests(private val dispatcher: CoroutineDispatc
     6
   }
 
+  @Trace
+  open fun traceAfterDelay(): Int = runTest {
+    tracedChild("before-process")
+
+    val inputs = listOf("a", "b", "c")
+
+    coroutineScope {
+      inputs.map { data ->
+        async {
+          upload(data)
+        }
+      }.awaitAll().map {
+        tracedChild("process-$it")
+        encrypt(it)
+      }
+    }
+
+    // FIXME: This span is detached
+    tracedChild("after-process")
+
+    6
+  }
+
+  private suspend fun upload(data: String): String {
+    delay(100)
+    return "url-$data"
+  }
+
+  private suspend fun encrypt(message: String): String {
+    delay(100)
+    return "encrypted-$message"
+  }
+
   @Trace
   protected open fun tracedChild(opName: String) {
     activeSpan().setSpanName(opName)