[INS-1693] Add WS echo server for smoke tests (#5050)

Co-authored-by: Dimitri Mitropoulos <[email protected]>

Author: filfreire

packages/insomnia-smoke-test/package-lock.json

@@ -41,6 +41,7 @@
     "@types/oidc-provider": "^7.8.1",
     "@types/ramda": "^0.27.45",
     "@types/uuid": "^8.3.4",
+    "@types/ws": "^8.5.3",
     "concurrently": "^7.0.0",
     "cross-env": "^7.0.3",
     "execa": "^5.0.0",
@@ -57,6 +58,7 @@
     "ramda-adjunct": "^2.34.0",
     "ts-node": "^9.1.1",
     "uuid": "^8.3.2",
+    "ws": "^8.8.1",
     "xvfb-maybe": "^0.2.1"
   }
 }

packages/insomnia-smoke-test/package.json

@@ -7,6 +7,7 @@ import gitlabApi from './gitlab-api';
 import { root, schema } from './graphql';
 import { startGRPCServer } from './grpc';
 import { oauthRoutes } from './oauth';
+import { startWebsocketServer } from './websocket';
 
 const app = express();
 const port = 4010;
@@ -55,7 +56,9 @@ app.use('/graphql', graphqlHTTP({
 }));
 
 startGRPCServer(grpcPort).then(() => {
-  app.listen(port, () => {
+  const server = app.listen(port, () => {
     console.log(`Listening at http://localhost:${port}`);
+    console.log(`Listening at ws://localhost:${port}`);
   });
+  startWebsocketServer(server);
 });

packages/insomnia-smoke-test/server/index.ts

@@ -0,0 +1,28 @@
+import { Server } from 'http';
+import { WebSocketServer } from 'ws';
+
+/**
+ * Starts an echo WebSocket server that receives messages from a client and echoes them back.
+ */
+export function startWebsocketServer(server: Server) {
+  const wsServer = new WebSocketServer({ server });
+
+  wsServer.on('connection', ws => {
+    console.log('WebSocket connection was opened');
+
+    ws.on('message', (message, isBinary) => {
+      if (isBinary) {
+        ws.send(message);
+        return;
+      }
+
+      ws.send(message.toString());
+    });
+
+    ws.on('close', () => {
+      console.log('WebSocket connection was closed');
+    });
+  });
+
+  return wsServer;
+}