test(lesy-pilot-ui): add tests for socket class

Author: lokesh-coder

packages/plugins/lesy-plugin-pilot/src/__tests__/pilot.socket.test.ts

@@ -0,0 +1,43 @@
+// tslint:disable-next-line: import-name
+import WebSocket from "ws";
+import { WebSocketBus } from "../pilot.socket";
+
+describe("web socket", () => {
+  let wsServer: WebSocketBus;
+  let wsClient: WebSocket;
+
+  beforeAll((done) => {
+    wsServer = new WebSocketBus().startServer("localhost", 1234).init({
+      requestRunCommand: (payload) => {
+        return {
+          success: true,
+          data: payload,
+        };
+      },
+    });
+
+    wsServer["ws"].on("connection", () => wsClient.on("open", done));
+    wsClient = new WebSocket(`ws://localhost:1234`);
+  });
+
+  afterAll((done) => {
+    wsClient.close();
+    wsServer.close(done);
+  });
+
+  it("should exchange messages", (done) => {
+    wsClient.on("message", (m) => {
+      expect(m).toEqual(
+        JSON.stringify({
+          success: true,
+          data: "echo 123",
+        }),
+      );
+      done();
+    });
+
+    wsClient.send(
+      JSON.stringify({ REQUEST: "requestRunCommand", PAYLOAD: "echo 123" }),
+    );
+  });
+});

packages/plugins/lesy-plugin-pilot/src/pilot.command.ts

@@ -65,7 +65,7 @@ export default class PilotCommand {
       [currCtx.feature.pkg.name]: currCtx,
     };
     global.lesySelectedProject = currCtx.feature.pkg.name;
-    const projectsPaths = await this.fetchProjectPaths();
+    const projectsPaths = await this.fetchProjectPaths(); // todo: remove same project dup
     for (let i = 0; i < projectsPaths.length; i = i + 1) {
       const p = await require(projectsPaths[i]).default;
       global.lesyWorkspace[p.feature.pkg.name] = p.localState;

packages/plugins/lesy-plugin-pilot/src/pilot.socket.ts

@@ -37,8 +37,10 @@ export class WebSocketBus {
     return this.receiver.asObservable();
   }
 
-  close() {
-    this.ws.close();
+  close(cb?: (param: any) => void) {
+    this.sender.unsubscribe();
+    this.receiver.unsubscribe();
+    this.ws.close(cb);
   }
 
   private onConnectionOpen() {