add tests

Author: pjfanning

instrumentation/pekko/pekko-http-1.0/javaagent/src/test/scala/io/opentelemetry/javaagent/instrumentation/pekkohttp/v1_0/PekkoHttpServerInstrumentationTestAsync.scala

@@ -6,14 +6,15 @@
 package io.opentelemetry.javaagent.instrumentation.pekkohttp.v1_0
 
 import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension
-import io.opentelemetry.instrumentation.testing.junit.http.{
-  HttpServerInstrumentationExtension,
-  HttpServerTestOptions
-}
+import io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.SUCCESS
+import io.opentelemetry.instrumentation.testing.junit.http.{HttpServerInstrumentationExtension, HttpServerTestOptions}
+import io.opentelemetry.testing.internal.armeria.common.{AggregatedHttpRequest, AggregatedHttpResponse}
+import org.assertj.core.api.Assertions.assertThat
+import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.extension.RegisterExtension
 
 class PekkoHttpServerInstrumentationTestAsync
-    extends AbstractHttpServerInstrumentationTest {
+  extends AbstractHttpServerInstrumentationTest {
 
   @RegisterExtension val extension: InstrumentationExtension =
     HttpServerInstrumentationExtension.forAgent()
@@ -27,9 +28,24 @@ class PekkoHttpServerInstrumentationTestAsync
     PekkoHttpTestAsyncWebServer.stop()
 
   override protected def configure(
-      options: HttpServerTestOptions
-  ): Unit = {
+                                    options: HttpServerTestOptions
+                                  ): Unit = {
     super.configure(options)
     options.setTestHttpPipelining(false)
   }
+
+  @Test
+  def successfulPostRequestWithParent(): Unit = {
+    val method = "POST"
+    val traceId = "00000000000000000000000000000123"
+    val parentId = "0000000000000456"
+    val aggregatedHttpRequest = AggregatedHttpRequest.of( // intentionally sending mixed-case "tracePARENT" to make sure that TextMapGetters are
+      // not case-sensitive
+      request(SUCCESS, method).headers.toBuilder.set("tracePARENT", "00-" + traceId + "-" + parentId + "-01").build)
+    val response = client.execute(aggregatedHttpRequest).aggregate.join
+    assertThat(response.status.code).isEqualTo(SUCCESS.getStatus)
+    assertThat(response.contentUtf8).isEqualTo(SUCCESS.getBody)
+    val spanId = assertResponseHasCustomizedHeaders(response, SUCCESS, traceId)
+    assertTheTraces(1, traceId, parentId, spanId, "POST", SUCCESS)
+  }
 }

instrumentation/pekko/pekko-http-1.0/javaagent/src/test/scala/io/opentelemetry/javaagent/instrumentation/pekkohttp/v1_0/PekkoHttpTestAsyncWebServer.scala

@@ -5,19 +5,20 @@
 
 package io.opentelemetry.javaagent.instrumentation.pekkohttp.v1_0
 
+import io.opentelemetry.context.Context
+import io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint._
+import io.opentelemetry.instrumentation.testing.junit.http.{AbstractHttpServerTest, ServerEndpoint}
 import org.apache.pekko.actor.ActorSystem
 import org.apache.pekko.http.scaladsl.Http
 import org.apache.pekko.http.scaladsl.Http.ServerBinding
-import org.apache.pekko.http.scaladsl.model.HttpMethods.GET
+import org.apache.pekko.http.scaladsl.model.HttpMethods.{GET, POST}
+import org.apache.pekko.http.scaladsl.model.StatusCodes.OK
 import org.apache.pekko.http.scaladsl.model._
+import org.apache.pekko.pattern.after
 import org.apache.pekko.stream.ActorMaterializer
-import io.opentelemetry.instrumentation.testing.junit.http.{
-  AbstractHttpServerTest,
-  ServerEndpoint
-}
-import io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint._
 
 import java.util.function.Supplier
+import scala.concurrent.duration.DurationInt
 import scala.concurrent.{Await, ExecutionContextExecutor, Future}
 
 object PekkoHttpTestAsyncWebServer {
@@ -57,6 +58,17 @@ object PekkoHttpTestAsyncWebServer {
           }
         )
       }
+    case HttpRequest(POST, uri: Uri, _, _, _) =>
+      val endpoint = ServerEndpoint.forPath(uri.path.toString())
+      AbstractHttpServerTest.controller(endpoint, () =>
+        for {
+          _ <- Future.successful(2)
+          firstContext = Context.current()
+          _ <- after(1.second, system.scheduler)(Future.successful(1))
+          secondContext = Context.current()
+          _ = assert(firstContext eq secondContext)
+        } yield HttpResponse(status = OK, entity = endpoint.getBody)
+      )
   }
 
   private var binding: ServerBinding = _