[test] Add test for 20585ee

Author: lpinca

test/WebSocketServer.test.js

@@ -238,7 +238,7 @@ describe('WebSocketServer', function () {
       });
     });
 
-    it('is updated when client closes the connection', function (done) {
+    it('is updated when client closes the connection (1/2)', function (done) {
       const wss = new WebSocketServer({ port: ++port }, () => {
         const ws = new WebSocket(`ws://localhost:${port}`);
 
@@ -253,6 +253,21 @@ describe('WebSocketServer', function () {
         });
       });
     });
+
+    it('is updated when client closes the connection (2/2)', function (done) {
+      const wss = new WebSocketServer({ port: ++port }, () => {
+        const ws = new WebSocket(`ws://localhost:${port}`);
+
+        ws.onopen = () => ws.close();
+      });
+
+      wss.on('connection', (ws) => {
+        ws.onclose = () => {
+          assert.strictEqual(wss.clients.size, 0);
+          wss.close(done);
+        };
+      });
+    });
   });
 
   describe('#options', function () {