Add tests on Websocket manager

server/test/websockets/index.test.js

@@ -0,0 +1,61 @@
+const EventEmitter = require('events');
+const WebSocket = require('ws');
+const { assert, fake } = require('sinon');
+const WebsocketManager = require('../../api/websockets');
+
+describe('Websockets', () => {
+  it('should send message to all users', () => {
+    const wss = new EventEmitter();
+    const gladys = {
+      event: new EventEmitter(),
+    };
+    const websocketManager = new WebsocketManager(wss, gladys);
+    const client = {
+      readyState: WebSocket.OPEN,
+      send: fake.returns(0),
+    };
+    const message = {
+      type: 'test',
+      payload: 'test',
+    };
+    websocketManager.userConnected({ id: 'aa0eaee9-5b90-4287-841a-0237f9d75832' }, client);
+    websocketManager.sendMessageAllUsers(message);
+    assert.calledWith(client.send, JSON.stringify(message));
+  });
+  it('should not send message to all users', () => {
+    const wss = new EventEmitter();
+    const gladys = {
+      event: new EventEmitter(),
+    };
+    const websocketManager = new WebsocketManager(wss, gladys);
+    const client = {
+      readyState: WebSocket.CLOSED,
+      send: fake.returns(0),
+    };
+    const message = {
+      type: 'test',
+      payload: 'test',
+    };
+    websocketManager.userConnected({ id: 'aa0eaee9-5b90-4287-841a-0237f9d75832' }, client);
+    websocketManager.sendMessageAllUsers(message);
+    assert.notCalled(client.send);
+  });
+  it('should send message to one users', () => {
+    const wss = new EventEmitter();
+    const gladys = {
+      event: new EventEmitter(),
+    };
+    const websocketManager = new WebsocketManager(wss, gladys);
+    const client = {
+      readyState: WebSocket.OPEN,
+      send: fake.returns(0),
+    };
+    const message = {
+      type: 'test',
+      payload: 'test',
+    };
+    websocketManager.userConnected({ id: 'aa0eaee9-5b90-4287-841a-0237f9d75832' }, client);
+    websocketManager.sendMessageUser({ ...message, userId: 'aa0eaee9-5b90-4287-841a-0237f9d75832' });
+    assert.calledWith(client.send, JSON.stringify(message));
+  });
+});