fix(test-optimization): support older Jest and Mocha versions (#9695)

Author: juan-fernandez

integration-tests/ci-visibility/jest-flaky/parameterized-fails.js

@@ -1,12 +1,10 @@
 'use strict'
 
-const assert = require('node:assert/strict')
-
 describe('test-flaky-test-retries', () => {
   test.each([
     ['passing row', true],
     ['failing row', false],
   ])('preserves parameters between retries', (row, shouldPass) => {
-    assert.strictEqual(shouldPass, true, row)
+    expect(shouldPass).toBe(true)
   })
 })

integration-tests/ci-visibility/mocha-plugin-tests/top-level-it-mixed-only.js

@@ -4,5 +4,7 @@ it('top-level passing test', () => {})
 
 // eslint-disable-next-line mocha/no-exclusive-tests
 describe.only('an exclusive describe block', () => {
-  it('nested passing test', () => {})
+  // eslint-disable-next-line mocha/no-exclusive-tests
+  it.only('nested passing test', () => {})
+  it('nested non-exclusive test', () => {})
 })

packages/datadog-instrumentations/src/mocha/main.js

@@ -398,6 +398,57 @@ function isFailedTestReplayEnabled () {
   return config.isTestDynamicInstrumentationEnabled && config.isDiEnabled
 }
 
+/**
+ * @typedef {object} MochaSuite
+ * @property {MochaSuite[]} suites
+ * @property {import('mocha').Test[]} tests
+ * @property {MochaSuite[]} _onlySuites
+ * @property {import('mocha').Test[]} _onlyTests
+ */
+
+/**
+ * Mirrors Mocha 5's private exclusivity check.
+ *
+ * @param {MochaSuite} suite
+ * @returns {boolean}
+ */
+function hasOnly (suite) {
+  if (suite._onlyTests.length || suite._onlySuites.length) return true
+
+  for (const childSuite of suite.suites) {
+    if (hasOnly(childSuite)) return true
+  }
+  return false
+}
+
+/**
+ * Mirrors Mocha 5's private exclusivity filter.
+ *
+ * @param {MochaSuite} suite
+ * @returns {boolean}
+ */
+function filterOnly (suite) {
+  if (suite._onlyTests.length) {
+    suite.tests = suite._onlyTests
+    suite.suites = []
+  } else {
+    suite.tests = []
+
+    for (const onlySuite of suite._onlySuites) {
+      if (hasOnly(onlySuite)) filterOnly(onlySuite)
+    }
+
+    const filteredSuites = []
+    for (const childSuite of suite.suites) {
+      if (suite._onlySuites.includes(childSuite) || filterOnly(childSuite)) {
+        filteredSuites.push(childSuite)
+      }
+    }
+    suite.suites = filteredSuites
+  }
+  return suite.tests.length > 0 || suite.suites.length > 0
+}
+
 function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFinishRequest, localSuites) {
   const ctx = {
     isParallel,
@@ -431,8 +482,10 @@ function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFini
 
     // We remove the suites that we skip through ITR
     // Mocha normally applies exclusivity after this asynchronous configuration step.
-    if (runner.suite.hasOnly()) {
-      runner.suite.filterOnly()
+    if (typeof runner.suite.hasOnly === 'function') {
+      if (runner.suite.hasOnly()) runner.suite.filterOnly()
+    } else if (hasOnly(runner.suite)) {
+      filterOnly(runner.suite)
     }
     const numTestsToRun = runner.grepTotal(runner.suite)
     const filteredSuites = getFilteredSuites(runner.suite.suites)