1 files changed, 35 insertions, 2 deletions
diff --git a/tests/js/hero.mjs b/tests/js/hero.mjs
index 33403d0..aed9cfc 100644
--- a/tests/js/hero.mjs
+++ b/tests/js/hero.mjs
@@ -1483,6 +1483,8 @@ const test_handleRequest = async t => {
 				h1: "H1",
 				h2: "h2",
 			},
+			upgrade: false,
+			socket: null,
 		};
 
 		assert.deepEqual(
@@ -1502,9 +1504,40 @@ const test_handleRequest = async t => {
 				},
 				handler: fn,
 				ref: req,
-				upgrade: true,
-				socket: fn,
+				upgrade: false,
+				socket: null,
+			},
+		);
+	});
+
+	await t.test("upgrade request", async () => {
+		const socket  = () => {};
+		const handler = req => {
+			assert.equal(req.socket, socket);
+			return "handler ret";
+		};
+		const table = {
+			websocket: {
+				GET: {
+					api: {
+						socket: {
+							"": handler,
+						},
+					},
+				},
 			},
+			interceptors: [],
+		};
+		const req = {
+			method: "GET",
+			url: "/api/socket",
+			upgrade: true,
+			socket,
+		};
+
+		assert.deepEqual(
+			await handleRequest(table, req),
+			"handler ret",
 		);
 	});