feat(experimentally): WebsocketServer implementation and tests (#2056)

Author: knagaitsev

lib/servers/WebsocketServer.js

@@ -1,8 +1,34 @@
 'use strict';
 
-// const ws = require('ws');
+/* eslint-disable
+  class-methods-use-this
+*/
+const ws = require('ws');
 const BaseServer = require('./BaseServer');
 
-// ws server implementation under construction
-// will need changes in the client as well to function
-module.exports = class WebsocketServer extends BaseServer {};
+module.exports = class WebsocketServer extends BaseServer {
+  constructor(server) {
+    super(server);
+    this.wsServer = new ws.Server({
+      server: this.server.listeningApp,
+      path: this.server.sockPath,
+    });
+
+    this.wsServer.on('error', (err) => {
+      this.server.log.error(err.message);
+    });
+  }
+
+  send(connection, message) {
+    connection.send(message);
+  }
+
+  close(connection) {
+    connection.close();
+  }
+
+  // f should return the resulting connection
+  onConnection(f) {
+    this.wsServer.on('connection', f);
+  }
+};

test/ports-map.js

@@ -36,6 +36,7 @@ const portsList = {
   'stats-option': 1,
   ProvidePlugin: 1,
   WebsocketClient: 1,
+  WebsocketServer: 1,
 };
 
 let startPort = 8079;

test/server/servers/WebsocketServer.test.js

@@ -0,0 +1,67 @@
+'use strict';
+
+const http = require('http');
+const express = require('express');
+const ws = require('ws');
+const WebsocketServer = require('../../../lib/servers/WebsocketServer');
+const port = require('../../ports-map').WebsocketServer;
+
+describe('WebsocketServer', () => {
+  let socketServer;
+  let listeningApp;
+
+  beforeAll((done) => {
+    // eslint-disable-next-line new-cap
+    const app = new express();
+
+    listeningApp = http.createServer(app);
+    listeningApp.listen(port, 'localhost', () => {
+      const server = {
+        log: {
+          error: () => {},
+          debug: () => {},
+        },
+        sockPath: '/ws-server',
+        listeningApp,
+      };
+
+      socketServer = new WebsocketServer(server);
+
+      done();
+    });
+  });
+
+  describe('server', () => {
+    it('should recieve connection, send message, and close client', (done) => {
+      const data = [];
+
+      socketServer.onConnection((connection) => {
+        data.push('open');
+        socketServer.send(connection, 'hello world');
+        setTimeout(() => {
+          socketServer.close(connection);
+        }, 1000);
+      });
+
+      // eslint-disable-next-line new-cap
+      const client = new ws(`http://localhost:${port}/ws-server`);
+
+      client.onmessage = (e) => {
+        data.push(e.data);
+      };
+
+      client.onclose = () => {
+        data.push('close');
+      };
+
+      setTimeout(() => {
+        expect(data).toMatchSnapshot();
+        done();
+      }, 3000);
+    });
+  });
+
+  afterAll((done) => {
+    listeningApp.close(done);
+  });
+});

test/server/servers/__snapshots__/WebsocketServer.test.js.snap

@@ -0,0 +1,9 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`WebsocketServer server should recieve connection, send message, and close client 1`] = `
+Array [
+  "open",
+  "hello world",
+  "close",
+]
+`;