Fix TracingTest failures (#88)

This follows-up on sticky context removal in Vert.x core. Signed-off-by: Thomas Segismont <[email protected]>
vert-x3 · Jan 9, 2025 · 7c241ad · 7c241ad
1 parent ff5c5d0
commit 7c241ad
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/test/java/io/vertx/cassandra/tests/TracingTest.java b/src/test/java/io/vertx/cassandra/tests/TracingTest.java
@@ -91,9 +91,11 @@ public void receiveResponse(Context context, Object response, Object payload, Th
       }
     };
 
-    client.executeWithFullFetch(cql).onComplete(ctx.asyncAssertSuccess(rows -> {
-      async.countDown();
-    }));
+    expectedContext.runOnContext(v -> {
+      client.executeWithFullFetch(cql).onComplete(ctx.asyncAssertSuccess(rows -> {
+        async.countDown();
+      }));
+    });
   }
 
   @Test
@@ -126,7 +128,9 @@ public void receiveResponse(Context context, Object response, Object payload, Th
       }
     };
 
-    client.executeWithFullFetch(cql).onComplete(ctx.asyncAssertFailure(t -> async.countDown()));
+    expectedContext.runOnContext(v -> {
+      client.executeWithFullFetch(cql).onComplete(ctx.asyncAssertFailure(t -> async.countDown()));
+    });
   }
 
   @Test
@@ -161,11 +165,12 @@ public void receiveResponse(Context context, Object response, Object payload, Th
       }
     };
 
-    client.prepare(cql).onComplete(ctx.asyncAssertSuccess(ps -> {
-      client.execute(ps.bind("B")).onComplete(ctx.asyncAssertSuccess(rs -> {
-        async.countDown();
+    expectedContext.runOnContext(v -> {
+      client.prepare(cql).onComplete(ctx.asyncAssertSuccess(ps -> {
+        client.execute(ps.bind("B")).onComplete(ctx.asyncAssertSuccess(rs -> {
+          async.countDown();
+        }));
       }));
-    }));
-
+    });
   }
 }