fix: code

Author: alexander-akait

lib/servers/SockJSServer.js

@@ -74,12 +74,14 @@ module.exports = class SockJSServer extends BaseServer {
       });
     });
 
-    this.implementation.close = () => {
+    this.implementation.close = (callback) => {
       for (const client of this.clients) {
         client.close();
       }
 
       this.clients.clear();
+
+      callback();
     };
   }
 };

test/server/proxy-option.test.js

@@ -584,7 +584,7 @@ describe("proxy option", () => {
           ws = new WebSocket(`ws://localhost:${port3}/proxy3/socket`);
 
           ws.on("message", (message) => {
-            responseMessage = message;
+            responseMessage = message.toString();
             done();
           });
 
@@ -600,6 +600,10 @@ describe("proxy option", () => {
         afterAll(async () => {
           webSocketServer.close();
 
+          for (const client of webSocketServer.clients) {
+            client.terminate();
+          }
+
           await server.stop();
         });
       });

test/server/setupExitSignals-option.test.js

@@ -46,10 +46,12 @@ describe("setupExitSignals option", () => {
 
   it.each(signals)("should close and exit on %s", (signal, done) => {
     process.emit(signal);
-    process.nextTick(() => {
+
+    setTimeout(() => {
       expect(killSpy.mock.calls.length).toEqual(1);
+      expect(exitSpy.mock.calls.length).toEqual(1);
 
       done();
-    });
+    }, 1000);
   });
 });