Fix flaky hibernate reactive test (#9338)

Author: laurit

instrumentation/hibernate/hibernate-reactive-1.0/javaagent/build.gradle.kts

@@ -38,6 +38,7 @@ testing {
           implementation("org.hibernate.reactive:hibernate-reactive-core:1.0.0.Final")
           implementation("io.vertx:vertx-pg-client:4.1.5")
         }
+        compileOnly("io.vertx:vertx-codegen:4.1.5")
       }
     }
 

instrumentation/hibernate/hibernate-reactive-1.0/javaagent/src/hibernateReactive1Test/java/io/opentelemetry/javaagent/instrumentation/hibernate/reactive/v1_0/HibernateReactiveTest.java

@@ -18,7 +18,9 @@
 import io.opentelemetry.api.trace.SpanKind;
 import io.opentelemetry.instrumentation.testing.junit.AgentInstrumentationExtension;
 import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
+import io.vertx.core.Vertx;
 import java.time.Duration;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import javax.persistence.EntityManagerFactory;
 import javax.persistence.Persistence;
@@ -117,23 +119,28 @@ void testMutiny() {
 
   @Test
   void testStage() throws Exception {
-    testing
-        .runWithSpan(
-            "parent",
-            () ->
-                stageSessionFactory
-                    .withSession(
-                        session -> {
-                          if (!Span.current().getSpanContext().isValid()) {
-                            throw new IllegalStateException("missing parent span");
-                          }
-
-                          return session
-                              .find(Value.class, 1L)
-                              .thenAccept(value -> testing.runWithSpan("callback", () -> {}));
-                        })
-                    .toCompletableFuture())
-        .get(30, TimeUnit.SECONDS);
+    CountDownLatch latch = new CountDownLatch(1);
+    testing.runWithSpan(
+        "parent",
+        () ->
+            Vertx.vertx()
+                .getOrCreateContext()
+                .runOnContext(
+                    event ->
+                        stageSessionFactory
+                            .withSession(
+                                session -> {
+                                  if (!Span.current().getSpanContext().isValid()) {
+                                    throw new IllegalStateException("missing parent span");
+                                  }
+
+                                  return session
+                                      .find(Value.class, 1L)
+                                      .thenAccept(
+                                          value -> testing.runWithSpan("callback", () -> {}));
+                                })
+                            .thenAccept(unused -> latch.countDown())));
+    latch.await(30, TimeUnit.SECONDS);
 
     assertTrace();
   }