From 87d5f5ee3f25bfac7cda30e223ac1f12e162d434 Mon Sep 17 00:00:00 2001
From: Bill Church <billchurch@users.noreply.github.com>
Date: Thu, 22 Aug 2024 02:08:26 +0000
Subject: [PATCH] chore: socket.js test

---
 tests/socket.test.js | 95 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100644 tests/socket.test.js

diff --git a/tests/socket.test.js b/tests/socket.test.js
new file mode 100644
index 0000000..eb7b0bf
--- /dev/null
+++ b/tests/socket.test.js
@@ -0,0 +1,95 @@
+// server
+// tests/socket.test.js
+
+const EventEmitter = require("events")
+const socketHandler = require("../app/socket")
+// const WebSSH2Socket = require("../app/socket")
+
+jest.mock("../app/ssh")
+
+describe("socketHandler", () => {
+  let io
+  let socket
+  let config
+
+  beforeEach(() => {
+    socket = new EventEmitter()
+    socket.id = "test-socket-id"
+    socket.handshake = {
+      session: {}
+    }
+    socket.emit = jest.fn()
+
+    io = {
+      on: jest.fn(function(event, callback) {
+        if (event === "connection") {
+          callback(socket)
+        }
+      })
+    }
+
+    config = {
+      ssh: {
+        term: "xterm-color"
+      },
+      options: {
+        allowreauth: true
+      }
+    }
+
+    socketHandler(io, config)
+  })
+
+  afterEach(() => {
+    jest.clearAllMocks()
+  })
+
+  test("should set up connection listener on io", () => {
+    expect(io.on).toHaveBeenCalledWith("connection", expect.any(Function))
+  })
+
+  test("should set up authenticate event listener on socket", () => {
+    expect(socket.listeners("authenticate")).toHaveLength(1)
+  })
+
+  test("should set up terminal event listener on socket", () => {
+    expect(socket.listeners("terminal")).toHaveLength(1)
+  })
+
+  test("should set up disconnect event listener on socket", () => {
+    expect(socket.listeners("disconnect")).toHaveLength(1)
+  })
+
+  test("should emit request_auth when not authenticated", () => {
+    expect(socket.emit).toHaveBeenCalledWith("authentication", {
+      action: "request_auth"
+    })
+  })
+
+  test("should handle authenticate event", () => {
+    const creds = {
+      username: "testuser",
+      password: "testpass",
+      host: "testhost",
+      port: 22
+    }
+    socket.emit("authenticate", creds)
+    // build out later
+  })
+
+  test("should handle terminal event", () => {
+    const terminalData = {
+      term: "xterm",
+      rows: 24,
+      cols: 80
+    }
+    socket.emit("terminal", terminalData)
+    // build out later
+  })
+
+  test("should handle disconnect event", () => {
+    const reason = "test-reason"
+    socket.emit("disconnect", reason)
+    // build out later
+  })
+})