From bc6fc52f643caf70e132da3d5b014ec4c1609d90 Mon Sep 17 00:00:00 2001
From: Lenvin Gonsalves <lenvingonsalves@gmail.com>
Date: Mon, 18 Jul 2022 15:24:01 +0530
Subject: [PATCH] test_runner: introduces test_concurrency flag

adds the test_concurrency flag,
if set to true, runs all the tests in parallel

Fixes: https://github.com/nodejs/node/issues/43837
---
 doc/api/cli.md                   |  8 ++++++++
 lib/internal/test_runner/test.js | 15 ++++++++++++---
 src/node_options.cc              |  3 +++
 src/node_options.h               |  1 +
 4 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/doc/api/cli.md b/doc/api/cli.md
index a2861024659af22..22a2427977605ab 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -1108,6 +1108,14 @@ Starts the Node.js command line test runner. This flag cannot be combined with
 `--check`, `--eval`, `--interactive`, or the inspector. See the documentation
 on [running tests from the command line][] for more details.
 
+### `--test-concurrency`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+Configures the test runner to run sub tests in parallel.
+
 ### `--test-only`
 
 <!-- YAML
diff --git a/lib/internal/test_runner/test.js b/lib/internal/test_runner/test.js
index 2e1ea1d7a3da5b3..704cbd7702ed78f 100644
--- a/lib/internal/test_runner/test.js
+++ b/lib/internal/test_runner/test.js
@@ -1,10 +1,12 @@
 'use strict';
 const {
   ArrayPrototypePush,
+  ArrayPrototypeReduce,
   ArrayPrototypeShift,
   ArrayPrototypeUnshift,
   FunctionPrototype,
   Number,
+  PromiseResolve,
   ReflectApply,
   SafeMap,
   PromiseRace,
@@ -41,8 +43,8 @@ const noop = FunctionPrototype;
 const isTestRunner = getOptionValue('--test');
 const testOnlyFlag = !isTestRunner && getOptionValue('--test-only');
 // TODO(cjihrig): Use uv_available_parallelism() once it lands.
-const rootConcurrency = isTestRunner ? cpus().length : 1;
-
+const rootConcurrency = isTestRunner ? cpus().length - 1 : 1;
+const subTestConcurrency = getOptionValue('--test-concurrency');
 
 function testTimeout(promise, timeout) {
   if (timeout === kDefaultTimeout) {
@@ -505,7 +507,14 @@ class Suite extends Test {
     this.parent.activeSubtests++;
     this.startTime = hrtime();
     const subtests = this.skipped || this.error ? [] : this.subtests;
-    await SafePromiseAll(subtests, (subtests) => subtests.start());
+    if (subTestConcurrency) {
+      await SafePromiseAll(subtests, (subtests) => subtests.start());
+    } else {
+      await testTimeout(ArrayPrototypeReduce(subtests, async (prev, subtest) => {
+        await prev;
+        await subtest.run();
+      }, PromiseResolve()), this.timeout);
+    }
     this.pass();
     this.postRun();
   }
diff --git a/src/node_options.cc b/src/node_options.cc
index 25e4caf98ad2106..75971dfc0aa3d76 100644
--- a/src/node_options.cc
+++ b/src/node_options.cc
@@ -525,6 +525,9 @@ EnvironmentOptionsParser::EnvironmentOptionsParser() {
   AddOption("--test",
             "launch test runner on startup",
             &EnvironmentOptions::test_runner);
+  AddOption("--test-concurrency",
+            "tests defined within a file run in parallel",
+            &EnvironmentOptions::test_concurrency);
   AddOption("--test-only",
             "run tests with 'only' option set",
             &EnvironmentOptions::test_only,
diff --git a/src/node_options.h b/src/node_options.h
index a4744fb9e5b8a58..f525bd1ecbad7e6 100644
--- a/src/node_options.h
+++ b/src/node_options.h
@@ -150,6 +150,7 @@ class EnvironmentOptions : public Options {
   std::string redirect_warnings;
   std::string diagnostic_dir;
   bool test_runner = false;
+  bool test_concurrency = false;
   bool test_only = false;
   bool test_udp_no_try_send = false;
   bool throw_deprecation = false;