update tests

This commit is contained in:
Your Name 2025-02-17 18:01:33 -08:00
parent 587699d7b3
commit dcb07a5e2e

View file

@ -1,76 +1,97 @@
const test = require("node:test"); const { describe, test } = require("node:test");
const assert = require("node:assert"); const assert = require("node:assert");
const { websocket } = require("../../server/monitor-types/websocket"); const { websocket } = require("../../server/monitor-types/websocket");
const { UP, DOWN, PENDING } = require("../../src/util"); const { UP, DOWN, PENDING } = require("../../src/util");
test("Test Websocket TLS", async () => { describe("Websocket Test", {
const websocketMonitor = new websocket(); }, () => {
test("Non Websocket Server", async () => {
const websocketMonitor = new websocket();
const monitor = { const monitor = {
wsurl: "wss://echo.websocket.org", wsurl: "wss://example.org",
wsIgnoreHeaders: false, wsIgnoreHeaders: false,
}; };
const heartbeat = { const heartbeat = {
msg: "", msg: "",
status: PENDING, status: PENDING,
}; };
await websocketMonitor.check(monitor, heartbeat, {}); await websocketMonitor.check(monitor, heartbeat, {});
assert.strictEqual(heartbeat.status, UP); assert.strictEqual(heartbeat.status, DOWN);
assert.strictEqual(heartbeat.msg, "101 - OK"); assert.strictEqual(heartbeat.msg, "Unexpected server response: 200");
}); });
test("Test Websocket non TLS", async () => { test("Secure Websocket", async () => {
const websocketMonitor = new websocket(); const websocketMonitor = new websocket();
const monitor = { const monitor = {
wsurl: "ws://ws.ifelse.io", wsurl: "wss://echo.websocket.org",
wsIgnoreHeaders: false, wsIgnoreHeaders: false,
}; };
const heartbeat = { const heartbeat = {
msg: "", msg: "",
status: PENDING, status: PENDING,
}; };
await websocketMonitor.check(monitor, heartbeat, {}); await websocketMonitor.check(monitor, heartbeat, {});
assert.strictEqual(heartbeat.status, UP); assert.strictEqual(heartbeat.status, UP);
assert.strictEqual(heartbeat.msg, "101 - OK"); assert.strictEqual(heartbeat.msg, "101 - OK");
}); });
test("Test a non compliant WS server without ignore", async () => { test("Insecure Websocket", async () => {
const websocketMonitor = new websocket(); const websocketMonitor = new websocket();
const monitor = { const monitor = {
wsurl: "wss://c.img-cdn.net/yE4s7KehTFyj/", wsurl: "ws://ws.ifelse.io",
wsIgnoreHeaders: false, wsIgnoreHeaders: false,
}; };
const heartbeat = { const heartbeat = {
msg: "", msg: "",
status: PENDING, status: PENDING,
}; };
await websocketMonitor.check(monitor, heartbeat, {}); await websocketMonitor.check(monitor, heartbeat, {});
assert.strictEqual(heartbeat.status, DOWN); assert.strictEqual(heartbeat.status, UP);
assert.strictEqual(heartbeat.msg, "Invalid Sec-WebSocket-Accept header"); assert.strictEqual(heartbeat.msg, "101 - OK");
}); });
test("Test a non compliant WS server with ignore", async () => { test("Test a non compliant WS server without ignore", async () => {
const websocketMonitor = new websocket(); const websocketMonitor = new websocket();
const monitor = { const monitor = {
wsurl: "wss://c.img-cdn.net/yE4s7KehTFyj/", wsurl: "wss://c.img-cdn.net/yE4s7KehTFyj/",
wsIgnoreHeaders: true, wsIgnoreHeaders: false,
}; };
const heartbeat = { const heartbeat = {
msg: "", msg: "",
status: PENDING, status: PENDING,
}; };
await websocketMonitor.check(monitor, heartbeat, {}); await websocketMonitor.check(monitor, heartbeat, {});
assert.strictEqual(heartbeat.status, UP); assert.strictEqual(heartbeat.status, DOWN);
assert.strictEqual(heartbeat.msg, "101 - OK"); assert.strictEqual(heartbeat.msg, "Invalid Sec-WebSocket-Accept header");
});
test("Test a non compliant WS server with ignore", async () => {
const websocketMonitor = new websocket();
const monitor = {
wsurl: "wss://c.img-cdn.net/yE4s7KehTFyj/",
wsIgnoreHeaders: true,
};
const heartbeat = {
msg: "",
status: PENDING,
};
await websocketMonitor.check(monitor, heartbeat, {});
assert.strictEqual(heartbeat.status, UP);
assert.strictEqual(heartbeat.msg, "101 - OK");
});
}); });