Assign each worker a worker id in parallel mode.

This uses a new feature from the workerpool module, which allows
overriding process parameters for each process, which is why the
workerpool module has been updated to the latest version (6.2.0).

docs/index.md

@@ -1379,6 +1379,10 @@ Free-tier cloud CI services may not provide a suitable multi-core container or V
 
 It's unlikely (but not impossible) to see a performance gain from a [job count](#-jobs-count-j-count) _greater than_ the number of available CPU cores. That said, _play around with the job count_--there's no one-size-fits all, and the unique characteristics of your tests will determine the optimal number of jobs; it may even be that fewer is faster!
 
+### Parallel Mode Worker IDs
+
+Each process launched by parallel mode is assigned a unique id, from 0 for the first process to be launched, to N-1 for the Nth process. This worker id may be accessed in tests via the environment variable `MOCHA_WORKER_ID`. It can be used for example to assign a different database, service port, etc for each test process.
+
 ## Root Hook Plugins
 
 > _New in v8.0.0_

lib/nodejs/buffered-worker-pool.js

@@ -75,7 +75,23 @@ class BufferedWorkerPool {
       process.execArgv.join(' ')
     );
 
-    this.options = {...WORKER_POOL_DEFAULT_OPTS, opts, maxWorkers};
+    let counter = 0;
+    const onCreateWorker = ({forkOpts}) => {
+      return {
+        forkOpts: {
+          ...forkOpts,
+          // adds an incrental id to all workers, which can be useful to allocate resources for each process
+          env: {...process.env, MOCHA_WORKER_ID: counter++}
+        }
+      };
+    };
+
+    this.options = {
+      ...WORKER_POOL_DEFAULT_OPTS,
+      ...opts,
+      maxWorkers,
+      onCreateWorker
+    };
     this._pool = workerpool.pool(WORKER_PATH, this.options);
   }
 

package.json

@@ -72,7 +72,7 @@
     "strip-json-comments": "3.1.1",
     "supports-color": "8.1.1",
     "which": "2.0.2",
-    "workerpool": "6.1.5",
+    "workerpool": "6.2.0",
     "yargs": "16.2.0",
     "yargs-parser": "20.2.4",
     "yargs-unparser": "2.0.0"

test/integration/fixtures/parallel/testworkerid1.mjs

@@ -0,0 +1,7 @@
+import {describe, it} from '../../../../index.js';
+
+describe('test1', () => {
+  it('should fail', () => {
+    throw new Error(`MOCHA_WORKER_ID=${process.env.MOCHA_WORKER_ID}`);
+  });
+});

test/integration/fixtures/parallel/testworkerid2.mjs

@@ -0,0 +1,7 @@
+import {describe, it} from '../../../../index.js';
+
+describe('test2', () => {
+  it('should fail', () => {
+    throw new Error(`MOCHA_WORKER_ID=${process.env.MOCHA_WORKER_ID}`);
+  });
+});

test/integration/fixtures/parallel/testworkerid3.mjs

@@ -0,0 +1,7 @@
+import {describe, it} from '../../../../index.js';
+
+describe('test3', () => {
+  it('should fail', () => {
+    throw new Error(`MOCHA_WORKER_ID=${process.env.MOCHA_WORKER_ID}`);
+  });
+});

test/integration/parallel.spec.js

@@ -18,4 +18,42 @@ describe('parallel run', () => {
     assert.strictEqual(result.stats.failures, 1);
     assert.strictEqual(result.stats.passes, 2);
   });
+
+  it('should correctly set worker ids for each process', async () => {
+    const result = await runMochaJSONAsync('parallel/testworkerid3.mjs', [
+      '--parallel',
+      '--jobs',
+      '2',
+      require.resolve('./fixtures/parallel/testworkerid1.mjs'),
+      require.resolve('./fixtures/parallel/testworkerid2.mjs')
+    ]);
+    // test 1 always runs on worker 0
+    assert.strictEqual(
+      result.tests.filter(
+        test =>
+          test.fullTitle === 'test1 should fail' &&
+          test.err.message === 'MOCHA_WORKER_ID=0'
+      ).length,
+      1
+    );
+    // test 2 always runs on worker 1
+    assert.strictEqual(
+      result.tests.filter(
+        test =>
+          test.fullTitle === 'test2 should fail' &&
+          test.err.message === 'MOCHA_WORKER_ID=1'
+      ).length,
+      1
+    );
+    // test 3 may run on either worker 0 or 1 (as jobs = 2)
+    assert.strictEqual(
+      result.tests.filter(
+        test =>
+          test.fullTitle === 'test3 should fail' &&
+          (test.err.message === 'MOCHA_WORKER_ID=0' ||
+            test.err.message === 'MOCHA_WORKER_ID=1')
+      ).length,
+      1
+    );
+  });
 });