[test] Fix nits

websockets · Mar 10, 2023 · 23acf8c · 23acf8c
1 parent cd89e07
commit 23acf8c
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/test/websocket.test.js b/test/websocket.test.js
@@ -3860,16 +3860,18 @@ describe('WebSocket', () => {
 
     it('honors the `finishRequest` option', (done) => {
       const wss = new WebSocket.Server({ port: 0 }, () => {
-        const ws = new WebSocket(`ws://localhost:${wss.address().port}`, {
-          finishRequest(request, websocket) {
-            process.nextTick(() => {
-              assert.strictEqual(request, ws._req);
-              assert.strictEqual(websocket, ws);
-            });
-            request.on('socket', (socket) => {
+        const host = `localhost:${wss.address().port}`;
+        const ws = new WebSocket(`ws://${host}`, {
+          finishRequest(req, ws) {
+            assert.ok(req instanceof http.ClientRequest);
+            assert.strictEqual(req.getHeader('host'), host);
+            assert.ok(ws instanceof WebSocket);
+            assert.strictEqual(req, ws._req);
+
+            req.on('socket', (socket) => {
               socket.on('connect', () => {
-                request.setHeader('Cookie', 'foo=bar');
-                request.end();
+                req.setHeader('Cookie', 'foo=bar');
+                req.end();
               });
             });
           }