[ci-visibility] Support mocha parallel mode (#4314)

DataDog · May 23, 2024 · ead2f41 · ead2f41
1 parent d1e1d01
commit ead2f41
Show file tree

Hide file tree

Showing 17 changed files with 1,196 additions and 730 deletions.
diff --git a/ci/init.js b/ci/init.js
@@ -4,6 +4,7 @@ const { isTrue } = require('../packages/dd-trace/src/util')
 
 const isJestWorker = !!process.env.JEST_WORKER_ID
 const isCucumberWorker = !!process.env.CUCUMBER_WORKER_ID
+const isMochaWorker = !!process.env.MOCHA_WORKER_ID
 
 const options = {
   startupLogs: false,
@@ -44,6 +45,12 @@ if (isCucumberWorker) {
   }
 }
 
+if (isMochaWorker) {
+  options.experimental = {
+    exporter: 'mocha_worker'
+  }
+}
+
 if (shouldInit) {
   tracer.init(options)
   tracer.use('fs', false)

diff --git a/ext/exporters.d.ts b/ext/exporters.d.ts
@@ -5,6 +5,7 @@ declare const exporters: {
   AGENT_PROXY: 'agent_proxy',
   JEST_WORKER: 'jest_worker',
   CUCUMBER_WORKER: 'cucumber_worker'
+  MOCHA_WORKER: 'mocha_worker'
 }
 
 export = exporters
diff --git a/ext/exporters.js b/ext/exporters.js
@@ -5,5 +5,6 @@ module.exports = {
   DATADOG: 'datadog',
   AGENT_PROXY: 'agent_proxy',
   JEST_WORKER: 'jest_worker',
-  CUCUMBER_WORKER: 'cucumber_worker'
+  CUCUMBER_WORKER: 'cucumber_worker',
+  MOCHA_WORKER: 'mocha_worker'
 }