Simplify MetricsTestBase#testConnectionLost

Signed-off-by: Thomas Segismont <[email protected]>

Author: tsegismont

vertx-sql-client/src/test/java/io/vertx/tests/sqlclient/tck/MetricsTestBase.java

@@ -33,7 +33,6 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
-import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Function;
@@ -185,13 +184,10 @@ private void awaitQueueSize(TestContext ctx, AtomicInteger queueSize, int num) t
   public void testConnectionLost(TestContext ctx) throws Exception {
     SqlConnectOptions connectOptions = connectOptions();
     ProxyServer proxy = ProxyServer.create(vertx, connectOptions.getPort(), connectOptions.getHost());
-    AtomicBoolean firstConnection = new AtomicBoolean(true);
+    AtomicReference<ProxyServer.Connection> firstConnection = new AtomicReference<>();
     proxy.proxyHandler(proxiedConn -> {
-      if (firstConnection.compareAndSet(true, false)) {
+      if (firstConnection.compareAndSet(null, proxiedConn)) {
         proxiedConn.connect();
-        vertx.eventBus().consumer("disconnect", msg -> {
-          proxiedConn.clientSocket().close();
-        });
       }
     });
     // Start proxy
@@ -232,7 +228,7 @@ public void dequeue(Object taskMetric) {
     }
     conn.closeHandler(v -> async.countDown());
     awaitQueueSize(ctx, queueSize, 16);
-    vertx.eventBus().send("disconnect", "boom");
+    firstConnection.get().clientSocket().close();
     async.await(20_000);
     ctx.assertEquals(0, queueSize.get());
   }