// 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

})

})