test: fix handshake timeout test failing on macOS

metarhia · Jul 21, 2017 · 426d675 · 426d675
1 parent 26e64dc
commit 426d675
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/node/connection-handshake.js b/test/node/connection-handshake.js
@@ -136,11 +136,11 @@ test.test(
     });
 
     const port = server.address().port;
-    net.connect(port, (error) => {
-      test.assertNot(error, 'must connect to server');
-      server.getClients()[0].on('close', () => {
-        test.pass('connection must be closed');
-      });
+    const connection = net.connect(port, error =>
+      test.assertNot(error, 'must connect to server')
+    );
+    connection.on('close', () => {
+      test.pass('connection must be closed');
     });
   }
 );