Create PekkoSchedulerTest.scala

Author: pjfanning

instrumentation/pekko/pekko-actor-1.0/javaagent/src/test/scala/io/opentelemetry/javaagent/instrumentation/pekkoactor/v1_0/PekkoSchedulerTest.scala

@@ -0,0 +1,45 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.pekkoactor.v1_0
+
+import io.opentelemetry.api.GlobalOpenTelemetry
+import io.opentelemetry.api.trace.Span
+import org.apache.pekko.actor.ActorSystem
+import org.apache.pekko.pattern.after
+import org.assertj.core.api.Assertions.assertThat
+import org.junit.jupiter.api.Test
+import scala.concurrent.{ Await, Future }
+import scala.concurrent.duration.DurationInt
+
+class PekkoSchedulerTest {
+
+  @Test
+  def checkThatSpanWorksWithPekkoScheduledEvents(): Unit = {
+    val system = ActorSystem("my-system")
+    implicit val executionContext = system.dispatcher
+    val tracer = GlobalOpenTelemetry.get.getTracer("test-tracer")
+    val initialSpan = tracer.spanBuilder("test").startSpan
+    try {
+      val scope = initialSpan.makeCurrent()
+      val futureResult = for {
+        result1 <- Future {
+          assertThat(Span.current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
+          assertThat(Span.current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
+          1
+        }
+        _ = assertThat(Span.current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
+        _ = assertThat(Span.current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
+        result2 <- after(200.millis, system.scheduler)(Future.successful(2))
+        _ = assertThat(Span.current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
+        _ = assertThat(Span.current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
+      } yield result1 + result2
+      assertThat(Await.result(futureResult, 5.seconds)).isEqualTo(3)
+    } finally {
+      system.terminate()
+      initialSpan.end()
+    }
+  }
+}