From 492d9f503f58f76eeaf5302f930a38ce3e4f02c3 Mon Sep 17 00:00:00 2001
From: PoleTransformer <you@example.com>
Date: Tue, 18 Feb 2025 16:29:19 -0800
Subject: [PATCH] one assert per testcase

---
 server/monitor-types/websocket-upgrade.js |  5 +--
 test/backend-test/test-websocket.js       | 45 ++++++++++++++++-------
 2 files changed, 33 insertions(+), 17 deletions(-)

diff --git a/server/monitor-types/websocket-upgrade.js b/server/monitor-types/websocket-upgrade.js
index 5d4842000..eb9a37143 100644
--- a/server/monitor-types/websocket-upgrade.js
+++ b/server/monitor-types/websocket-upgrade.js
@@ -11,9 +11,8 @@ class WebSocketMonitorType extends MonitorType {
     async check(monitor, heartbeat, _server) {
         const [ message, code ] = await this.attemptUpgrade(monitor);
         heartbeat.status = code === 1000 ? UP : DOWN;
-        //heartbeat.msg = message;
-        heartbeat.msg = code; //unit testing
-        console.log(message); //temporary test to pass eslint check
+        heartbeat.msg = message;
+        console.log(code, message); //temp unit testing
     }
 
     /**
diff --git a/test/backend-test/test-websocket.js b/test/backend-test/test-websocket.js
index e5d88580f..ee62eef26 100644
--- a/test/backend-test/test-websocket.js
+++ b/test/backend-test/test-websocket.js
@@ -18,9 +18,13 @@ describe("Websocket Test", {
             status: PENDING,
         };
 
+        const expected = {
+            msg: "Unexpected server response: 200",
+            status: DOWN,
+        };
+
         await websocketMonitor.check(monitor, heartbeat, {});
-        assert.strictEqual(heartbeat.status, DOWN);
-        assert.strictEqual(heartbeat.msg, undefined);
+        assert.deepStrictEqual(heartbeat, expected);
     });
 
     test("Secure Websocket", async () => {
@@ -36,14 +40,16 @@ describe("Websocket Test", {
             status: PENDING,
         };
 
+        const expected = {
+            msg: "101 - OK",
+            status: UP,
+        };
+
         await websocketMonitor.check(monitor, heartbeat, {});
-        assert.strictEqual(heartbeat.status, UP);
-        assert.strictEqual(heartbeat.msg, 1000);
+        assert.deepStrictEqual(heartbeat, expected);
     });
 
-    test("Insecure Websocket", {
-        skip: !!process.env.CI,
-    }, async () => {
+    test("Insecure Websocket", async () => {
         const websocketMonitor = new WebSocketMonitorType();
 
         const monitor = {
@@ -56,10 +62,13 @@ describe("Websocket Test", {
             status: PENDING,
         };
 
+        const expected = {
+            msg: "101 - OK",
+            status: UP,
+        };
+
         await websocketMonitor.check(monitor, heartbeat, {});
-        console.log("Insecure WS Test:", heartbeat.msg, heartbeat.status);
-        assert.strictEqual(heartbeat.status, UP);
-        assert.strictEqual(heartbeat.msg, 1000);
+        assert.deepStrictEqual(heartbeat, expected);
     });
 
     test("Test a non compliant WS server without ignore", async () => {
@@ -75,9 +84,13 @@ describe("Websocket Test", {
             status: PENDING,
         };
 
+        const expected = {
+            msg: "Invalid Sec-WebSocket-Accept header",
+            status: DOWN,
+        };
+
         await websocketMonitor.check(monitor, heartbeat, {});
-        assert.strictEqual(heartbeat.status, DOWN);
-        assert.strictEqual(heartbeat.msg, undefined);
+        assert.deepStrictEqual(heartbeat, expected);
     });
 
     test("Test a non compliant WS server with ignore", async () => {
@@ -93,8 +106,12 @@ describe("Websocket Test", {
             status: PENDING,
         };
 
+        const expected = {
+            msg: "101 - OK",
+            status: UP,
+        };
+
         await websocketMonitor.check(monitor, heartbeat, {});
-        assert.strictEqual(heartbeat.status, UP);
-        assert.strictEqual(heartbeat.msg, 1000);
+        assert.deepStrictEqual(heartbeat, expected);
     });
 });