Fix intermittent timeout in SockJSRawTransportTest

Use await() for WebSocket connect instead of an async callback to ensure
the frame handler is set before any frames are processed.

Signed-off-by: Thomas Segismont <tsegismont@gmail.com>

Author: tsegismont

vertx-web/src/test/java/io/vertx/ext/web/tests/handler/sockjs/SockJSRawTransportTest.java

@@ -16,6 +16,7 @@
 package io.vertx.ext.web.tests.handler.sockjs;
 
 import io.vertx.core.buffer.Buffer;
+import io.vertx.core.http.WebSocket;
 import io.vertx.core.http.WebSocketConnectOptions;
 import io.vertx.ext.web.handler.sockjs.SockJSHandlerOptions;
 import io.vertx.junit5.Checkpoint;
@@ -136,21 +137,20 @@ private void testWrite(boolean text, Checkpoint checkpoint) throws Exception {
       });
     };
     startServers(new SockJSHandlerOptions());
-    wsClient.connect("/test/websocket").onComplete(TestUtils.onSuccess(ws -> {
-      ws.frameHandler(frame -> {
-        if (frame.isClose()) {
-          //
+    WebSocket ws = wsClient.connect("/test/websocket").await();
+    ws.frameHandler(frame -> {
+      if (frame.isClose()) {
+        //
+      } else {
+        if (text) {
+          assertTrue(frame.isText());
+          assertEquals(expected, frame.textData());
         } else {
-          if (text) {
-            assertTrue(frame.isText());
-            assertEquals(expected, frame.textData());
-          } else {
-            assertTrue(frame.isBinary());
-            assertEquals(Buffer.buffer(expected), frame.binaryData());
-          }
-          ws.end();
+          assertTrue(frame.isBinary());
+          assertEquals(Buffer.buffer(expected), frame.binaryData());
         }
-      });
-    }));
+        ws.end();
+      }
+    });
   }
 }