Update PekkoHttpTestAsyncWebServer.scala

Author: pjfanning

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

@@ -5,7 +5,7 @@
 
 package io.opentelemetry.javaagent.instrumentation.pekkohttp.v1_0
 
-import io.opentelemetry.context.Context
+import io.opentelemetry.api.trace.Span
 import io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint._
 import io.opentelemetry.instrumentation.testing.junit.http.{AbstractHttpServerTest, ServerEndpoint}
 import org.apache.pekko.actor.ActorSystem
@@ -63,10 +63,10 @@ object PekkoHttpTestAsyncWebServer {
       AbstractHttpServerTest.controller(endpoint, () =>
         for {
           _ <- Future.successful(2)
-          firstContext = Context.current()
+          firstTraceId = Span.current().getSpanContext().getTraceId()
           _ <- after(1.second, system.scheduler)(Future.successful(1))
-          secondContext = Context.current()
-          _ = assert(firstContext eq secondContext)
+          secondTraceId = Span.current().getSpanContext().getTraceId()
+          _ = require(firstTraceId eq secondTraceId, s"TraceId $firstTraceId did not match $secondTraceId")
         } yield HttpResponse(status = OK, entity = endpoint.getBody)
       )
   }