test: update test-http-upgrade-client to use countdown

PR-URL: #17339 Reviewed-By: Jon Moss <[email protected]> Reviewed-By: Anatoli Papirovski <[email protected]>
nodejs · Dec 12, 2017 · c9a4f4f · c9a4f4f
1 parent 91d5416
commit c9a4f4f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/parallel/test-http-upgrade-client.js b/test/parallel/test-http-upgrade-client.js
@@ -29,6 +29,7 @@ const assert = require('assert');
 
 const http = require('http');
 const net = require('net');
+const Countdown = require('../common/countdown');
 
 // Create a TCP server
 const srv = net.createServer(function(c) {
@@ -60,7 +61,8 @@ srv.listen(0, '127.0.0.1', common.mustCall(function() {
       ['Origin', 'http://www.websocket.org']
     ]
   ];
-  let left = headers.length;
+  const countdown = new Countdown(headers.length, () => srv.close());
+
   headers.forEach(function(h) {
     const req = http.get({
       port: port,
@@ -87,8 +89,7 @@ srv.listen(0, '127.0.0.1', common.mustCall(function() {
       assert.deepStrictEqual(expectedHeaders, res.headers);
 
       socket.end();
-      if (--left === 0)
-        srv.close();
+      countdown.dec();
     }));
     req.on('close', common.mustCall(function() {
       assert.strictEqual(sawUpgrade, true);