test: fix flaky tests

socketio · Jan 5, 2022 · c840bad · c840bad
1 parent f2b8de7
commit c840bad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/test/socket.io.ts b/test/socket.io.ts
@@ -1830,7 +1830,7 @@ describe("socket.io", () => {
           reconnectionDelay: 100,
         });
         clientSocket.once("connect", () => {
-          srv.close(() => {
+          sio.close(() => {
             clientSocket.io.on("reconnect", () => {
               clientSocket.emit("ev", "payload");
             });

diff --git a/test/uws.ts b/test/uws.ts
@@ -46,8 +46,10 @@ describe("socket.io with uWebSocket.js-based engine", () => {
     });
 
     const partialDone = createPartialDone(done, 4);
-    io.on("connection", partialDone);
-    io.of("/custom").on("connection", partialDone);
+    client.on("connect", partialDone);
+    clientWSOnly.on("connect", partialDone);
+    clientPollingOnly.on("connect", partialDone);
+    clientCustomNamespace.on("connect", partialDone);
   });
 
   afterEach(() => {