Revert "refactor: emit client connect in next tick during testing"

This reverts commit 5efb887.
ilijaNL · Feb 1, 2023 · c10d0bf · c10d0bf
1 parent 5efb887
commit c10d0bf
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/__tests__/utils/tservers.ts b/src/__tests__/utils/tservers.ts
@@ -218,7 +218,6 @@ export async function startWSTServer(
   const pendingClients: TServerClient[] = [];
   wsServer.on('connection', (client) => {
     pendingClients.push(toClient(client));
-    setTimeout(() => emitter.emit('client'), 0);
     client.once('close', () => {
       pendingCloses++;
       emitter.emit('close');
@@ -254,10 +253,10 @@ export async function startWSTServer(
           resolve();
         }
         if (pendingClients.length > 0) return done();
-        emitter.once('client', done);
+        wsServer.once('connection', done);
         if (expire)
           setTimeout(() => {
-            emitter.off('client', done); // expired
+            wsServer.off('connection', done); // expired
             resolve();
           }, expire);
       });
@@ -507,7 +506,6 @@ export async function startFastifyWSTServer(
     fastify.get(path, { websocket: true }, (connection, request) => {
       sockets.add(connection.socket);
       pendingClients.push(toClient(connection.socket));
-      setTimeout(() => emitter.emit('client'), 0);
       connection.socket.once('close', () => {
         sockets.delete(connection.socket);
         pendingCloses++;
@@ -586,10 +584,10 @@ export async function startFastifyWSTServer(
           resolve();
         }
         if (pendingClients.length > 0) return done();
-        emitter.once('client', done);
+        fastify.websocketServer.once('connection', done);
         if (expire)
           setTimeout(() => {
-            emitter.off('client', done); // expired
+            fastify.websocketServer.off('connection', done); // expired
             resolve();
           }, expire);
       });