test: add common.mustCall

nodejs · Jul 16, 2017 · 2a86f01 · 2a86f01
1 parent 5d3609d
commit 2a86f01
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 12 deletions.
diff --git a/test/parallel/test-http-server-keep-alive-timeout.js b/test/parallel/test-http-server-keep-alive-timeout.js
@@ -21,13 +21,15 @@ function run() {
 
 test(function serverEndKeepAliveTimeoutWithPipeline(cb) {
   let requestCount = 0;
-  process.on('exit', () => {
-    assert.strictEqual(requestCount, 3);
-  });
   const server = http.createServer((req, res) => {
     requestCount++;
     res.end();
   });
+  server.listen(0, common.mustCall(() => {
+    process.on('exit', () => {
+      assert.strictEqual(requestCount, 3);
+    });
+  }));
   server.setTimeout(500, common.mustCall((socket) => {
     // End this test and call `run()` for the next test (if any).
     socket.destroy();
@@ -50,12 +52,14 @@ test(function serverEndKeepAliveTimeoutWithPipeline(cb) {
 
 test(function serverNoEndKeepAliveTimeoutWithPipeline(cb) {
   let requestCount = 0;
-  process.on('exit', () => {
-    assert.strictEqual(requestCount, 3);
-  });
   const server = http.createServer((req, res) => {
     requestCount++;
   });
+  server.listen(0, common.mustCall(() => {
+    process.on('exit', () => {
+      assert.strictEqual(requestCount, 3);
+    });
+  }));
   server.setTimeout(500, common.mustCall((socket) => {
     // End this test and call `run()` for the next test (if any).
     socket.destroy();

diff --git a/test/parallel/test-https-server-keep-alive-timeout.js b/test/parallel/test-https-server-keep-alive-timeout.js
@@ -30,13 +30,15 @@ function run() {
 
 test(function serverKeepAliveTimeoutWithPipeline(cb) {
   let requestCount = 0;
-  process.on('exit', function() {
-    assert.strictEqual(requestCount, 3);
-  });
   const server = https.createServer(serverOptions, (req, res) => {
     requestCount++;
     res.end();
   });
+  server.listen(0, common.mustCall(() => {
+    process.on('exit', function() {
+      assert.strictEqual(requestCount, 3);
+    });
+  }));
   server.setTimeout(500, common.mustCall((socket) => {
     // End this test and call `run()` for the next test (if any).
     socket.destroy();
@@ -60,12 +62,14 @@ test(function serverKeepAliveTimeoutWithPipeline(cb) {
 
 test(function serverNoEndKeepAliveTimeoutWithPipeline(cb) {
   let requestCount = 0;
-  process.on('exit', () => {
-    assert.strictEqual(requestCount, 3);
-  });
   const server = https.createServer(serverOptions, (req, res) => {
     requestCount++;
   });
+  server.listen(0, common.mustCall(() => {
+    process.on('exit', function() {
+      assert.strictEqual(requestCount, 3);
+    });
+  }));
   server.setTimeout(500, common.mustCall((socket) => {
     // End this test and call `run()` for the next test (if any).
     socket.destroy();