From dfc05a7ce2093453da23991beb72c7e691729baa Mon Sep 17 00:00:00 2001 From: yassinedorbozgithub Date: Tue, 13 May 2025 08:55:03 +0100 Subject: [PATCH] fix(api): update WebsocketGateway unit tests --- api/src/websocket/websocket.gateway.spec.ts | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/api/src/websocket/websocket.gateway.spec.ts b/api/src/websocket/websocket.gateway.spec.ts index 5e6d41a7..6035c3eb 100644 --- a/api/src/websocket/websocket.gateway.spec.ts +++ b/api/src/websocket/websocket.gateway.spec.ts @@ -109,16 +109,10 @@ describe('WebsocketGateway', () => { describe('joinNotificationSockets', () => { it('should make socket1 and socket3 join the room MESSAGE', async () => { - messageRoomSockets.forEach((socket) => { - socket?.connect(); - }); + messageRoomSockets.forEach((socket) => socket.connect()); for (const socket of messageRoomSockets) { - await new Promise((resolve) => { - socket.on('connect', async () => { - resolve(); - }); - }); + await new Promise((resolve) => socket.on('connect', resolve)); } const serverSockets = await gateway.io.fetchSockets(); @@ -126,8 +120,8 @@ describe('WebsocketGateway', () => { expect(serverSockets.length).toBe(2); jest.spyOn(gateway, 'getNotificationSockets').mockResolvedValueOnce( - serverSockets.filter(({ handshake }) => { - const uuid = handshake.headers.uuid?.toString() || ''; + serverSockets.filter(({ handshake: { headers } }) => { + const uuid = headers.uuid?.toString() || ''; return validUuids.includes(uuid); }), @@ -146,7 +140,7 @@ describe('WebsocketGateway', () => { }); } - sockets.forEach((socket) => socket.disconnect()); + messageRoomSockets.forEach((socket) => socket.disconnect()); }); it('should throw an error when socket array is empty', async () => {