2025-11-05 17:04:23 -03:00

83 lines
2.1 KiB
JavaScript

import Connection from "../index.js";
test("disconnect", async () => {
const conn = new Connection();
const el = {};
const spy_closeStream = jest
.spyOn(conn, "_closeStream")
.mockImplementation(async () => el);
const spy_closeSocket = jest.spyOn(conn, "_closeSocket");
expect(await conn.disconnect()).toBe(el);
expect(spy_closeStream).toHaveBeenCalledTimes(1);
expect(spy_closeSocket).toHaveBeenCalledTimes(1);
});
test("disconnect with _closeStream rejection", async () => {
const conn = new Connection();
const spy_closeStream = jest
.spyOn(conn, "_closeStream")
.mockImplementation(() => {
return Promise.reject();
});
const spy_closeSocket = jest.spyOn(conn, "_closeSocket");
await conn.disconnect();
expect(spy_closeStream).toHaveBeenCalledTimes(1);
expect(spy_closeSocket).toHaveBeenCalledTimes(1);
});
test("disconnect with _closeSocket rejection", async () => {
const conn = new Connection();
const spy_closeStream = jest.spyOn(conn, "_closeStream");
const spy_closeSocket = jest
.spyOn(conn, "_closeSocket")
.mockImplementation(() => {
return Promise.reject();
});
await conn.disconnect();
expect(spy_closeStream).toHaveBeenCalledTimes(1);
expect(spy_closeSocket).toHaveBeenCalledTimes(1);
});
test("disconnect with _closeStream and _closeSocket rejections", async () => {
const conn = new Connection();
const spy_closeStream = jest
.spyOn(conn, "_closeStream")
.mockImplementation(() => {
return Promise.reject();
});
const spy_closeSocket = jest
.spyOn(conn, "_closeSocket")
.mockImplementation(() => {
return Promise.reject();
});
await conn.disconnect();
expect(spy_closeStream).toHaveBeenCalledTimes(1);
expect(spy_closeSocket).toHaveBeenCalledTimes(1);
});
test("resolves if socket property is undefined", async () => {
const conn = new Connection();
conn.footerElement = () => <foo />;
conn.socket = undefined;
await conn.disconnect();
expect().pass();
});
test("does not reject if connection is not established", async () => {
const conn = new Connection();
await conn.disconnect();
expect().pass();
});