test (websocket): speed up test/websocket/issue-2679.js (#4501)

Author: Uzlopak

test/websocket/issue-2679.js

@@ -1,20 +1,22 @@
 'use strict'
 
-const { test } = require('node:test')
+const { test, after } = require('node:test')
 const assert = require('node:assert')
 const { once } = require('node:events')
 const { WebSocketServer } = require('ws')
 const { WebSocket } = require('../..')
 
 test('Close without receiving code does not send an invalid payload', async () => {
   const server = new WebSocketServer({ port: 0 })
+  after(() => {
+    server.close()
+    return once(server, 'close')
+  })
 
   await once(server, 'listening')
 
   server.on('connection', (sock, request) => {
-    setTimeout(() => {
-      sock.close()
-    }, 3000)
+    sock.close()
   })
 
   server.on('error', (err) => assert.ifError(err))
@@ -23,7 +25,4 @@ test('Close without receiving code does not send an invalid payload', async () =
   await once(client, 'open')
 
   await once(client, 'close')
-
-  server.close()
-  await once(server, 'close')
 })