dd-trace 5.54.0 → 5.56.0

package/packages/datadog-instrumentations/src/jest.js

@@ -1,6 +1,6 @@
 'use strict'
 
-const { addHook, channel, AsyncResource } = require('./helpers/instrument')
+const { addHook, channel } = require('./helpers/instrument')
 const shimmer = require('../../datadog-shimmer')
 const log = require('../../dd-trace/src/log')
 const {
@@ -83,14 +83,13 @@ let testManagementAttemptToFixRetries = 0
 let isImpactedTestsEnabled = false
 let modifiedTests = {}
 
-const sessionAsyncResource = new AsyncResource('bound-anonymous-fn')
-
 const testContexts = new WeakMap()
 const originalTestFns = new WeakMap()
 const originalHookFns = new WeakMap()
 const retriedTestsToNumAttempts = new Map()
 const newTestsTestStatuses = new Map()
 const attemptToFixRetriedTestsStatuses = new Map()
+const wrappedWorkers = new WeakSet()
 
 const BREAKPOINT_HIT_GRACE_PERIOD_MS = 200
 
@@ -427,6 +426,10 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
       if (event.name === 'add_test') {
         const originalTestName = this.getTestNameFromAddTestEvent(event, state)
 
+        if (event.failing) {
+          return
+        }
+
         const isSkipped = event.mode === 'todo' || event.mode === 'skip'
         if (this.isTestManagementTestsEnabled) {
           const isAttemptToFix = this.testManagementTestsForThisSuite?.attemptToFix?.includes(originalTestName)
@@ -581,22 +584,30 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
         }
       }
       if (event.name === 'test_skip' || event.name === 'test_todo') {
-        const asyncResource = new AsyncResource('bound-anonymous-fn')
-        asyncResource.runInAsyncScope(() => {
-          testSkippedCh.publish({
-            test: {
-              name: getJestTestName(event.test),
-              suite: this.testSuite,
-              testSourceFile: this.testSourceFile,
-              displayName: this.displayName,
-              frameworkVersion: jestVersion,
-              testStartLine: getTestLineStart(event.test.asyncError, this.testSuite)
-            },
-            isDisabled: this.testManagementTestsForThisSuite?.disabled?.includes(getJestTestName(event.test))
-          })
+        testSkippedCh.publish({
+          test: {
+            name: getJestTestName(event.test),
+            suite: this.testSuite,
+            testSourceFile: this.testSourceFile,
+            displayName: this.displayName,
+            frameworkVersion: jestVersion,
+            testStartLine: getTestLineStart(event.test.asyncError, this.testSuite)
+          },
+          isDisabled: this.testManagementTestsForThisSuite?.disabled?.includes(getJestTestName(event.test))
         })
       }
     }
+
+    teardown () {
+      if (this._globalProxy?.propertyToValue) {
+        for (const [key] of this._globalProxy.propertyToValue) {
+          if (typeof key === 'string' && key.startsWith('_dd')) {
+            this._globalProxy.propertyToValue.delete(key)
+          }
+        }
+      }
+      return super.teardown()
+    }
   }
 }
 
@@ -652,219 +663,210 @@ function getWrappedScheduleTests (scheduleTests, frameworkVersion) {
   }
 }
 
-addHook({
-  name: '@jest/core',
-  file: 'build/TestScheduler.js',
-  versions: ['>=27.0.0']
-}, (testSchedulerPackage, frameworkVersion) => {
-  const oldCreateTestScheduler = testSchedulerPackage.createTestScheduler
-  const newCreateTestScheduler = async function () {
-    if (!isSuitesSkippingEnabled || hasFilteredSkippableSuites) {
-      return oldCreateTestScheduler.apply(this, arguments)
-    }
-    // If suite skipping is enabled and has not filtered skippable suites yet, we'll attempt to do it
-    const scheduler = await oldCreateTestScheduler.apply(this, arguments)
-    shimmer.wrap(scheduler, 'scheduleTests', scheduleTests => getWrappedScheduleTests(scheduleTests, frameworkVersion))
-    return scheduler
-  }
-  testSchedulerPackage.createTestScheduler = newCreateTestScheduler
-  return testSchedulerPackage
-})
+function searchSourceWrapper (searchSourcePackage, frameworkVersion) {
+  const SearchSource = searchSourcePackage.default ?? searchSourcePackage
 
-addHook({
-  name: '@jest/core',
-  file: 'build/TestScheduler.js',
-  versions: ['>=24.8.0 <27.0.0']
-}, (testSchedulerPackage, frameworkVersion) => {
-  shimmer.wrap(
-    testSchedulerPackage.default.prototype,
-    'scheduleTests', scheduleTests => getWrappedScheduleTests(scheduleTests, frameworkVersion)
-  )
-  return testSchedulerPackage
-})
+  shimmer.wrap(SearchSource.prototype, 'getTestPaths', getTestPaths => async function () {
+    const testPaths = await getTestPaths.apply(this, arguments)
+    const [{ rootDir, shard }] = arguments
 
-addHook({
-  name: '@jest/test-sequencer',
-  versions: ['>=28']
-}, (sequencerPackage, frameworkVersion) => {
-  shimmer.wrap(sequencerPackage.default.prototype, 'shard', shard => function () {
-    const shardedTests = shard.apply(this, arguments)
+    if (isKnownTestsEnabled) {
+      const projectSuites = testPaths.tests.map(test => getTestSuitePath(test.path, test.context.config.rootDir))
+      const isFaulty =
+        getIsFaultyEarlyFlakeDetection(projectSuites, knownTests?.jest || {}, earlyFlakeDetectionFaultyThreshold)
+      if (isFaulty) {
+        log.error('Early flake detection is disabled because the number of new suites is too high.')
+        isEarlyFlakeDetectionEnabled = false
+        isKnownTestsEnabled = false
+        const testEnvironmentOptions = testPaths.tests[0]?.context?.config?.testEnvironmentOptions
+        // Project config is shared among all tests, so we can modify it here
+        if (testEnvironmentOptions) {
+          testEnvironmentOptions._ddIsEarlyFlakeDetectionEnabled = false
+          testEnvironmentOptions._ddIsKnownTestsEnabled = false
+        }
+        isEarlyFlakeDetectionFaulty = true
+      }
+    }
 
-    if (!shardedTests.length || !isSuitesSkippingEnabled || !skippableSuites.length) {
-      return shardedTests
+    if (shard?.shardCount > 1 || !isSuitesSkippingEnabled || !skippableSuites.length) {
+      // If the user is using jest sharding, we want to apply the filtering of tests in the shard process.
+      // The reason for this is the following:
+      // The tests for different shards are likely being run in different CI jobs so
+      // the requests to the skippable endpoint might be done at different times and their responses might be different.
+      // If the skippable endpoint is returning different suites and we filter the list of tests here,
+      // the base list of tests that is used for sharding might be different,
+      // causing the shards to potentially run the same suite.
+      return testPaths
     }
-    const [test] = shardedTests
-    const rootDir = test?.context?.config?.rootDir
+    const { tests } = testPaths
 
-    return applySuiteSkipping(shardedTests, rootDir, frameworkVersion)
+    const suitesToRun = applySuiteSkipping(tests, rootDir, frameworkVersion)
+    return { ...testPaths, tests: suitesToRun }
   })
-  return sequencerPackage
-})
 
-function cliWrapper (cli, jestVersion) {
-  shimmer.wrap(cli, 'runCLI', runCLI => async function () {
-    let onDone
-    const configurationPromise = new Promise((resolve) => {
-      onDone = resolve
-    })
-    if (!libraryConfigurationCh.hasSubscribers) {
-      return runCLI.apply(this, arguments)
+  return searchSourcePackage
+}
+
+function getCliWrapper (isNewJestVersion) {
+  return function cliWrapper (cli, jestVersion) {
+    if (isNewJestVersion) {
+      cli = shimmer.wrap(
+        cli,
+        'SearchSource',
+        searchSource => searchSourceWrapper(searchSource, jestVersion),
+        { replaceGetter: true }
+      )
     }
+    return shimmer.wrap(cli, 'runCLI', runCLI => async function () {
+      let onDone
+      const configurationPromise = new Promise((resolve) => {
+        onDone = resolve
+      })
+      if (!libraryConfigurationCh.hasSubscribers) {
+        return runCLI.apply(this, arguments)
+      }
 
-    sessionAsyncResource.runInAsyncScope(() => {
       libraryConfigurationCh.publish({ onDone })
-    })
 
-    try {
-      const { err, libraryConfig } = await configurationPromise
-      if (!err) {
-        isCodeCoverageEnabled = libraryConfig.isCodeCoverageEnabled
-        isSuitesSkippingEnabled = libraryConfig.isSuitesSkippingEnabled
-        isEarlyFlakeDetectionEnabled = libraryConfig.isEarlyFlakeDetectionEnabled
-        earlyFlakeDetectionNumRetries = libraryConfig.earlyFlakeDetectionNumRetries
-        earlyFlakeDetectionFaultyThreshold = libraryConfig.earlyFlakeDetectionFaultyThreshold
-        isKnownTestsEnabled = libraryConfig.isKnownTestsEnabled
-        isTestManagementTestsEnabled = libraryConfig.isTestManagementEnabled
-        testManagementAttemptToFixRetries = libraryConfig.testManagementAttemptToFixRetries
-        isImpactedTestsEnabled = libraryConfig.isImpactedTestsEnabled
+      try {
+        const { err, libraryConfig } = await configurationPromise
+        if (!err) {
+          isCodeCoverageEnabled = libraryConfig.isCodeCoverageEnabled
+          isSuitesSkippingEnabled = libraryConfig.isSuitesSkippingEnabled
+          isEarlyFlakeDetectionEnabled = libraryConfig.isEarlyFlakeDetectionEnabled
+          earlyFlakeDetectionNumRetries = libraryConfig.earlyFlakeDetectionNumRetries
+          earlyFlakeDetectionFaultyThreshold = libraryConfig.earlyFlakeDetectionFaultyThreshold
+          isKnownTestsEnabled = libraryConfig.isKnownTestsEnabled
+          isTestManagementTestsEnabled = libraryConfig.isTestManagementEnabled
+          testManagementAttemptToFixRetries = libraryConfig.testManagementAttemptToFixRetries
+          isImpactedTestsEnabled = libraryConfig.isImpactedTestsEnabled
+        }
+      } catch (err) {
+        log.error('Jest library configuration error', err)
       }
-    } catch (err) {
-      log.error('Jest library configuration error', err)
-    }
 
-    if (isKnownTestsEnabled) {
-      const knownTestsPromise = new Promise((resolve) => {
-        onDone = resolve
-      })
+      if (isKnownTestsEnabled) {
+        const knownTestsPromise = new Promise((resolve) => {
+          onDone = resolve
+        })
 
-      sessionAsyncResource.runInAsyncScope(() => {
         knownTestsCh.publish({ onDone })
-      })
 
-      try {
-        const { err, knownTests: receivedKnownTests } = await knownTestsPromise
-        if (err) {
-          // We disable EFD if there has been an error in the known tests request
-          isEarlyFlakeDetectionEnabled = false
-          isKnownTestsEnabled = false
-        } else {
-          knownTests = receivedKnownTests
+        try {
+          const { err, knownTests: receivedKnownTests } = await knownTestsPromise
+          if (err) {
+            // We disable EFD if there has been an error in the known tests request
+            isEarlyFlakeDetectionEnabled = false
+            isKnownTestsEnabled = false
+          } else {
+            knownTests = receivedKnownTests
+          }
+        } catch (err) {
+          log.error('Jest known tests error', err)
         }
-      } catch (err) {
-        log.error('Jest known tests error', err)
       }
-    }
 
-    if (isSuitesSkippingEnabled) {
-      const skippableSuitesPromise = new Promise((resolve) => {
-        onDone = resolve
-      })
+      if (isSuitesSkippingEnabled) {
+        const skippableSuitesPromise = new Promise((resolve) => {
+          onDone = resolve
+        })
 
-      sessionAsyncResource.runInAsyncScope(() => {
         skippableSuitesCh.publish({ onDone })
-      })
 
-      try {
-        const { err, skippableSuites: receivedSkippableSuites } = await skippableSuitesPromise
-        if (!err) {
-          skippableSuites = receivedSkippableSuites
+        try {
+          const { err, skippableSuites: receivedSkippableSuites } = await skippableSuitesPromise
+          if (!err) {
+            skippableSuites = receivedSkippableSuites
+          }
+        } catch (err) {
+          log.error('Jest test-suite skippable error', err)
         }
-      } catch (err) {
-        log.error('Jest test-suite skippable error', err)
       }
-    }
 
-    if (isTestManagementTestsEnabled) {
-      const testManagementTestsPromise = new Promise((resolve) => {
-        onDone = resolve
-      })
+      if (isTestManagementTestsEnabled) {
+        const testManagementTestsPromise = new Promise((resolve) => {
+          onDone = resolve
+        })
 
-      sessionAsyncResource.runInAsyncScope(() => {
         testManagementTestsCh.publish({ onDone })
-      })
 
-      try {
-        const { err, testManagementTests: receivedTestManagementTests } = await testManagementTestsPromise
-        if (!err) {
-          testManagementTests = receivedTestManagementTests
+        try {
+          const { err, testManagementTests: receivedTestManagementTests } = await testManagementTestsPromise
+          if (!err) {
+            testManagementTests = receivedTestManagementTests
+          }
+        } catch (err) {
+          log.error('Jest test management tests error', err)
         }
-      } catch (err) {
-        log.error('Jest test management tests error', err)
       }
-    }
 
-    if (isImpactedTestsEnabled) {
-      const impactedTestsPromise = new Promise((resolve) => {
-        onDone = resolve
-      })
+      if (isImpactedTestsEnabled) {
+        const impactedTestsPromise = new Promise((resolve) => {
+          onDone = resolve
+        })
 
-      sessionAsyncResource.runInAsyncScope(() => {
         impactedTestsCh.publish({ onDone })
-      })
 
-      try {
-        const { err, modifiedTests: receivedModifiedTests } = await impactedTestsPromise
-        if (!err) {
-          modifiedTests = receivedModifiedTests
+        try {
+          const { err, modifiedTests: receivedModifiedTests } = await impactedTestsPromise
+          if (!err) {
+            modifiedTests = receivedModifiedTests
+          }
+        } catch (err) {
+          log.error('Jest impacted tests error', err)
         }
-      } catch (err) {
-        log.error('Jest impacted tests error', err)
       }
-    }
 
-    const processArgv = process.argv.slice(2).join(' ')
-    sessionAsyncResource.runInAsyncScope(() => {
+      const processArgv = process.argv.slice(2).join(' ')
       testSessionStartCh.publish({ command: `jest ${processArgv}`, frameworkVersion: jestVersion })
-    })
 
-    const result = await runCLI.apply(this, arguments)
+      const result = await runCLI.apply(this, arguments)
 
-    const {
-      results: {
-        success,
-        coverageMap,
-        numFailedTestSuites,
-        numFailedTests,
-        numTotalTests,
-        numTotalTestSuites
-      }
-    } = result
+      const {
+        results: {
+          success,
+          coverageMap,
+          numFailedTestSuites,
+          numFailedTests,
+          numTotalTests,
+          numTotalTestSuites
+        }
+      } = result
 
-    let testCodeCoverageLinesTotal
+      let testCodeCoverageLinesTotal
 
-    if (isUserCodeCoverageEnabled) {
-      try {
-        const { pct, total } = coverageMap.getCoverageSummary().lines
-        testCodeCoverageLinesTotal = total === 0 ? 0 : pct
-      } catch {
-        // ignore errors
+      if (isUserCodeCoverageEnabled) {
+        try {
+          const { pct, total } = coverageMap.getCoverageSummary().lines
+          testCodeCoverageLinesTotal = total === 0 ? 0 : pct
+        } catch {
+          // ignore errors
+        }
       }
-    }
-    let status, error
+      let status, error
 
-    if (success) {
-      status = numTotalTests === 0 && numTotalTestSuites === 0 ? 'skip' : 'pass'
-    } else {
-      status = 'fail'
-      error = new Error(`Failed test suites: ${numFailedTestSuites}. Failed tests: ${numFailedTests}`)
-    }
-    let timeoutId
-
-    // Pass the resolve callback to defer it to DC listener
-    const flushPromise = new Promise((resolve) => {
-      onDone = () => {
-        clearTimeout(timeoutId)
-        resolve()
+      if (success) {
+        status = numTotalTests === 0 && numTotalTestSuites === 0 ? 'skip' : 'pass'
+      } else {
+        status = 'fail'
+        error = new Error(`Failed test suites: ${numFailedTestSuites}. Failed tests: ${numFailedTests}`)
       }
-    })
+      let timeoutId
 
-    const timeoutPromise = new Promise((resolve) => {
-      timeoutId = setTimeout(() => {
-        resolve('timeout')
-      }, FLUSH_TIMEOUT).unref()
-    })
+      // Pass the resolve callback to defer it to DC listener
+      const flushPromise = new Promise((resolve) => {
+        onDone = () => {
+          clearTimeout(timeoutId)
+          resolve()
+        }
+      })
+
+      const timeoutPromise = new Promise((resolve) => {
+        timeoutId = setTimeout(() => {
+          resolve('timeout')
+        }, FLUSH_TIMEOUT).unref()
+      })
 
-    sessionAsyncResource.runInAsyncScope(() => {
       testSessionFinishCh.publish({
         status,
         isSuitesSkipped,
@@ -880,84 +882,86 @@ function cliWrapper (cli, jestVersion) {
         isTestManagementTestsEnabled,
         onDone
       })
-    })
-    const waitingResult = await Promise.race([flushPromise, timeoutPromise])
 
-    if (waitingResult === 'timeout') {
-      log.error('Timeout waiting for the tracer to flush')
-    }
+      const waitingResult = await Promise.race([flushPromise, timeoutPromise])
 
-    numSkippedSuites = 0
-
-    /**
-     * If Early Flake Detection (EFD) is enabled the logic is as follows:
-     * - If all attempts for a test are failing, the test has failed and we will let the test process fail.
-     * - If just a single attempt passes, we will prevent the test process from failing.
-     * The rationale behind is the following: you may still be able to block your CI pipeline by gating
-     * on flakiness (the test will be considered flaky), but you may choose to unblock the pipeline too.
-     */
-
-    if (isEarlyFlakeDetectionEnabled) {
-      let numFailedTestsToIgnore = 0
-      for (const testStatuses of newTestsTestStatuses.values()) {
-        const { pass, fail } = getTestStats(testStatuses)
-        if (pass > 0) { // as long as one passes, we'll consider the test passed
-          numFailedTestsToIgnore += fail
-        }
+      if (waitingResult === 'timeout') {
+        log.error('Timeout waiting for the tracer to flush')
       }
-      // If every test that failed was an EFD retry, we'll consider the suite passed
-      if (numFailedTestsToIgnore !== 0 && result.results.numFailedTests === numFailedTestsToIgnore) {
-        result.results.success = true
+
+      numSkippedSuites = 0
+
+      /**
+       * If Early Flake Detection (EFD) is enabled the logic is as follows:
+       * - If all attempts for a test are failing, the test has failed and we will let the test process fail.
+       * - If just a single attempt passes, we will prevent the test process from failing.
+       * The rationale behind is the following: you may still be able to block your CI pipeline by gating
+       * on flakiness (the test will be considered flaky), but you may choose to unblock the pipeline too.
+       */
+
+      if (isEarlyFlakeDetectionEnabled) {
+        let numFailedTestsToIgnore = 0
+        for (const testStatuses of newTestsTestStatuses.values()) {
+          const { pass, fail } = getTestStats(testStatuses)
+          if (pass > 0) { // as long as one passes, we'll consider the test passed
+            numFailedTestsToIgnore += fail
+          }
+        }
+        // If every test that failed was an EFD retry, we'll consider the suite passed
+        if (numFailedTestsToIgnore !== 0 && result.results.numFailedTests === numFailedTestsToIgnore) {
+          result.results.success = true
+        }
       }
-    }
 
-    if (isTestManagementTestsEnabled) {
-      const failedTests = result
-        .results
-        .testResults.flatMap(({ testResults, testFilePath: testSuiteAbsolutePath }) => (
-          testResults.map(({ fullName: testName, status }) => (
-            { testName, testSuiteAbsolutePath, status }
+      if (isTestManagementTestsEnabled) {
+        const failedTests = result
+          .results
+          .testResults.flatMap(({ testResults, testFilePath: testSuiteAbsolutePath }) => (
+            testResults.map(({ fullName: testName, status }) => (
+              { testName, testSuiteAbsolutePath, status }
+            ))
           ))
-        ))
-        .filter(({ status }) => status === 'failed')
-
-      let numFailedQuarantinedTests = 0
-      let numFailedQuarantinedOrDisabledAttemptedToFixTests = 0
-
-      for (const { testName, testSuiteAbsolutePath } of failedTests) {
-        const testSuite = getTestSuitePath(testSuiteAbsolutePath, result.globalConfig.rootDir)
-        const originalName = removeAttemptToFixStringFromTestName(testName)
-        const testManagementTest = testManagementTests
-          ?.jest
-          ?.suites
-          ?.[testSuite]
-          ?.tests
-          ?.[originalName]
-          ?.properties
-        // This uses `attempt_to_fix` because this is always the main process and it's not formatted in camelCase
-        if (testManagementTest?.attempt_to_fix && (testManagementTest?.quarantined || testManagementTest?.disabled)) {
-          numFailedQuarantinedOrDisabledAttemptedToFixTests++
-        } else if (testManagementTest?.quarantined) {
-          numFailedQuarantinedTests++
+          .filter(({ status }) => status === 'failed')
+
+        let numFailedQuarantinedTests = 0
+        let numFailedQuarantinedOrDisabledAttemptedToFixTests = 0
+
+        for (const { testName, testSuiteAbsolutePath } of failedTests) {
+          const testSuite = getTestSuitePath(testSuiteAbsolutePath, result.globalConfig.rootDir)
+          const originalName = removeAttemptToFixStringFromTestName(testName)
+          const testManagementTest = testManagementTests
+            ?.jest
+            ?.suites
+            ?.[testSuite]
+            ?.tests
+            ?.[originalName]
+            ?.properties
+          // This uses `attempt_to_fix` because this is always the main process and it's not formatted in camelCase
+          if (testManagementTest?.attempt_to_fix && (testManagementTest?.quarantined || testManagementTest?.disabled)) {
+            numFailedQuarantinedOrDisabledAttemptedToFixTests++
+          } else if (testManagementTest?.quarantined) {
+            numFailedQuarantinedTests++
+          }
         }
-      }
 
-      // If every test that failed was quarantined, we'll consider the suite passed
-      // Note that if a test is attempted to fix,
-      // it's considered quarantined both if it's disabled and if it's quarantined (it'll run but its status is ignored)
-      if (
-        (numFailedQuarantinedOrDisabledAttemptedToFixTests !== 0 || numFailedQuarantinedTests !== 0) &&
-        result.results.numFailedTests ===
-          numFailedQuarantinedTests + numFailedQuarantinedOrDisabledAttemptedToFixTests
-      ) {
-        result.results.success = true
+        // If every test that failed was quarantined, we'll consider the suite passed
+        // Note that if a test is attempted to fix,
+        // it's considered quarantined both if it's disabled and if it's quarantined
+        // (it'll run but its status is ignored)
+        if (
+          (numFailedQuarantinedOrDisabledAttemptedToFixTests !== 0 || numFailedQuarantinedTests !== 0) &&
+          result.results.numFailedTests ===
+            numFailedQuarantinedTests + numFailedQuarantinedOrDisabledAttemptedToFixTests
+        ) {
+          result.results.success = true
+        }
       }
-    }
-
-    return result
-  })
 
-  return cli
+      return result
+    }, {
+      replaceGetter: true
+    })
+  }
 }
 
 function coverageReporterWrapper (coverageReporter) {
@@ -981,6 +985,55 @@ function coverageReporterWrapper (coverageReporter) {
   return coverageReporter
 }
 
+addHook({
+  name: '@jest/core',
+  file: 'build/TestScheduler.js',
+  versions: ['>=27.0.0']
+}, (testSchedulerPackage, frameworkVersion) => {
+  const oldCreateTestScheduler = testSchedulerPackage.createTestScheduler
+  const newCreateTestScheduler = async function () {
+    if (!isSuitesSkippingEnabled || hasFilteredSkippableSuites) {
+      return oldCreateTestScheduler.apply(this, arguments)
+    }
+    // If suite skipping is enabled and has not filtered skippable suites yet, we'll attempt to do it
+    const scheduler = await oldCreateTestScheduler.apply(this, arguments)
+    shimmer.wrap(scheduler, 'scheduleTests', scheduleTests => getWrappedScheduleTests(scheduleTests, frameworkVersion))
+    return scheduler
+  }
+  testSchedulerPackage.createTestScheduler = newCreateTestScheduler
+  return testSchedulerPackage
+})
+
+addHook({
+  name: '@jest/core',
+  file: 'build/TestScheduler.js',
+  versions: ['>=24.8.0 <27.0.0']
+}, (testSchedulerPackage, frameworkVersion) => {
+  shimmer.wrap(
+    testSchedulerPackage.default.prototype,
+    'scheduleTests', scheduleTests => getWrappedScheduleTests(scheduleTests, frameworkVersion)
+  )
+  return testSchedulerPackage
+})
+
+addHook({
+  name: '@jest/test-sequencer',
+  versions: ['>=28']
+}, (sequencerPackage, frameworkVersion) => {
+  shimmer.wrap(sequencerPackage.default.prototype, 'shard', shard => function () {
+    const shardedTests = shard.apply(this, arguments)
+
+    if (!shardedTests.length || !isSuitesSkippingEnabled || !skippableSuites.length) {
+      return shardedTests
+    }
+    const [test] = shardedTests
+    const rootDir = test?.context?.config?.rootDir
+
+    return applySuiteSkipping(shardedTests, rootDir, frameworkVersion)
+  })
+  return sequencerPackage
+})
+
 addHook({
   name: '@jest/reporters',
   file: 'build/coverage_reporter.js',
@@ -993,11 +1046,23 @@ addHook({
   versions: ['>=26.6.2']
 }, coverageReporterWrapper)
 
+addHook({
+  name: '@jest/reporters',
+  versions: ['>=30.0.0']
+}, (reporters) => {
+  return shimmer.wrap(reporters, 'CoverageReporter', coverageReporterWrapper, { replaceGetter: true })
+})
+
 addHook({
   name: '@jest/core',
   file: 'build/cli/index.js',
-  versions: ['>=24.8.0']
-}, cliWrapper)
+  versions: ['>=24.8.0 <30.0.0']
+}, getCliWrapper(false))
+
+addHook({
+  name: '@jest/core',
+  versions: ['>=30.0.0']
+}, getCliWrapper(true))
 
 function jestAdapterWrapper (jestAdapter, jestVersion) {
   const adapter = jestAdapter.default ?? jestAdapter
@@ -1006,45 +1071,40 @@ function jestAdapterWrapper (jestAdapter, jestVersion) {
     if (!environment) {
       return adapter.apply(this, arguments)
     }
-    const asyncResource = new AsyncResource('bound-anonymous-fn')
-    return asyncResource.runInAsyncScope(() => {
-      testSuiteStartCh.publish({
-        testSuite: environment.testSuite,
-        testEnvironmentOptions: environment.testEnvironmentOptions,
-        testSourceFile: environment.testSourceFile,
-        displayName: environment.displayName,
-        frameworkVersion: jestVersion
-      })
-      return adapter.apply(this, arguments).then(suiteResults => {
-        const { numFailingTests, skipped, failureMessage: errorMessage } = suiteResults
-        let status = 'pass'
-        if (skipped) {
-          status = 'skipped'
-        } else if (numFailingTests !== 0) {
-          status = 'fail'
-        }
+    testSuiteStartCh.publish({
+      testSuite: environment.testSuite,
+      testEnvironmentOptions: environment.testEnvironmentOptions,
+      testSourceFile: environment.testSourceFile,
+      displayName: environment.displayName,
+      frameworkVersion: jestVersion
+    })
+    return adapter.apply(this, arguments).then(suiteResults => {
+      const { numFailingTests, skipped, failureMessage: errorMessage } = suiteResults
+      let status = 'pass'
+      if (skipped) {
+        status = 'skipped'
+      } else if (numFailingTests !== 0) {
+        status = 'fail'
+      }
 
-        /**
-         * Child processes do not each request ITR configuration, so the jest's parent process
-         * needs to pass them the configuration. This is done via _ddTestCodeCoverageEnabled, which
-         * controls whether coverage is reported.
-        */
-        if (environment.testEnvironmentOptions?._ddTestCodeCoverageEnabled) {
-          const root = environment.repositoryRoot || environment.rootDir
+      /**
+       * Child processes do not each request ITR configuration, so the jest's parent process
+       * needs to pass them the configuration. This is done via _ddTestCodeCoverageEnabled, which
+       * controls whether coverage is reported.
+      */
+      if (environment.testEnvironmentOptions?._ddTestCodeCoverageEnabled) {
+        const root = environment.repositoryRoot || environment.rootDir
 
-          const coverageFiles = getCoveredFilenamesFromCoverage(environment.global.__coverage__)
-            .map(filename => getTestSuitePath(filename, root))
+        const coverageFiles = getCoveredFilenamesFromCoverage(environment.global.__coverage__)
+          .map(filename => getTestSuitePath(filename, root))
 
-          asyncResource.runInAsyncScope(() => {
-            testSuiteCodeCoverageCh.publish({ coverageFiles, testSuite: environment.testSourceFile })
-          })
-        }
-        testSuiteFinishCh.publish({ status, errorMessage })
-        return suiteResults
-      }).catch(error => {
-        testSuiteFinishCh.publish({ status: 'fail', error })
-        throw error
-      })
+        testSuiteCodeCoverageCh.publish({ coverageFiles, testSuite: environment.testSourceFile })
+      }
+      testSuiteFinishCh.publish({ status, errorMessage })
+      return suiteResults
+    }).catch(error => {
+      testSuiteFinishCh.publish({ status: 'fail', error })
+      throw error
     })
   })
   if (jestAdapter.default) {
@@ -1056,6 +1116,12 @@ function jestAdapterWrapper (jestAdapter, jestVersion) {
   return jestAdapter
 }
 
+addHook({
+  name: 'jest-circus',
+  file: 'build/runner.js',
+  versions: ['>=30.0.0']
+}, jestAdapterWrapper)
+
 addHook({
   name: 'jest-circus',
   file: 'build/legacy-code-todo-rewrite/jestAdapter.js',
@@ -1064,9 +1130,7 @@ addHook({
 
 function configureTestEnvironment (readConfigsResult) {
   const { configs } = readConfigsResult
-  sessionAsyncResource.runInAsyncScope(() => {
-    testSessionConfigurationCh.publish(configs.map(config => config.testEnvironmentOptions))
-  })
+  testSessionConfigurationCh.publish(configs.map(config => config.testEnvironmentOptions))
   // We can't directly use isCodeCoverageEnabled when reporting coverage in `jestAdapterWrapper`
   // because `jestAdapterWrapper` runs in a different process. We have to go through `testEnvironmentOptions`
   configs.forEach(config => {
@@ -1102,21 +1166,19 @@ function configureTestEnvironment (readConfigsResult) {
 }
 
 function jestConfigAsyncWrapper (jestConfig) {
-  shimmer.wrap(jestConfig, 'readConfigs', readConfigs => async function () {
+  return shimmer.wrap(jestConfig, 'readConfigs', readConfigs => async function () {
     const readConfigsResult = await readConfigs.apply(this, arguments)
     configureTestEnvironment(readConfigsResult)
     return readConfigsResult
   })
-  return jestConfig
 }
 
 function jestConfigSyncWrapper (jestConfig) {
-  shimmer.wrap(jestConfig, 'readConfigs', readConfigs => function () {
+  return shimmer.wrap(jestConfig, 'readConfigs', readConfigs => function () {
     const readConfigsResult = readConfigs.apply(this, arguments)
     configureTestEnvironment(readConfigsResult)
     return readConfigsResult
   })
-  return jestConfig
 }
 
 addHook({
@@ -1167,51 +1229,9 @@ addHook({
  */
 addHook({
   name: '@jest/core',
-  versions: ['>=24.8.0'],
+  versions: ['>=24.8.0 <30.0.0'],
   file: 'build/SearchSource.js'
-}, (searchSourcePackage, frameworkVersion) => {
-  const SearchSource = searchSourcePackage.default ?? searchSourcePackage
-
-  shimmer.wrap(SearchSource.prototype, 'getTestPaths', getTestPaths => async function () {
-    const testPaths = await getTestPaths.apply(this, arguments)
-    const [{ rootDir, shard }] = arguments
-
-    if (isKnownTestsEnabled) {
-      const projectSuites = testPaths.tests.map(test => getTestSuitePath(test.path, test.context.config.rootDir))
-      const isFaulty =
-        getIsFaultyEarlyFlakeDetection(projectSuites, knownTests?.jest || {}, earlyFlakeDetectionFaultyThreshold)
-      if (isFaulty) {
-        log.error('Early flake detection is disabled because the number of new suites is too high.')
-        isEarlyFlakeDetectionEnabled = false
-        isKnownTestsEnabled = false
-        const testEnvironmentOptions = testPaths.tests[0]?.context?.config?.testEnvironmentOptions
-        // Project config is shared among all tests, so we can modify it here
-        if (testEnvironmentOptions) {
-          testEnvironmentOptions._ddIsEarlyFlakeDetectionEnabled = false
-          testEnvironmentOptions._ddIsKnownTestsEnabled = false
-        }
-        isEarlyFlakeDetectionFaulty = true
-      }
-    }
-
-    if (shard?.shardCount > 1 || !isSuitesSkippingEnabled || !skippableSuites.length) {
-      // If the user is using jest sharding, we want to apply the filtering of tests in the shard process.
-      // The reason for this is the following:
-      // The tests for different shards are likely being run in different CI jobs so
-      // the requests to the skippable endpoint might be done at different times and their responses might be different.
-      // If the skippable endpoint is returning different suites and we filter the list of tests here,
-      // the base list of tests that is used for sharding might be different,
-      // causing the shards to potentially run the same suite.
-      return testPaths
-    }
-    const { tests } = testPaths
-
-    const suitesToRun = applySuiteSkipping(tests, rootDir, frameworkVersion)
-    return { ...testPaths, tests: suitesToRun }
-  })
-
-  return searchSourcePackage
-})
+}, searchSourceWrapper)
 
 // from 25.1.0 on, readConfigs becomes async
 addHook({
@@ -1259,19 +1279,27 @@ addHook({
   return runtimePackage
 })
 
-/*
-* This hook does three things:
-* - Pass known tests to the workers.
-* - Pass test management tests to the workers.
-* - Receive trace, coverage and logs payloads from the workers.
-*/
-addHook({
-  name: 'jest-worker',
-  versions: ['>=24.9.0'],
-  file: 'build/workers/ChildProcessWorker.js'
-}, (childProcessWorker) => {
-  const ChildProcessWorker = childProcessWorker.default
-  shimmer.wrap(ChildProcessWorker.prototype, 'send', send => function (request) {
+function onMessageWrapper (onMessage) {
+  return function () {
+    const [code, data] = arguments[0]
+    if (code === JEST_WORKER_TRACE_PAYLOAD_CODE) { // datadog trace payload
+      workerReportTraceCh.publish(data)
+      return
+    }
+    if (code === JEST_WORKER_COVERAGE_PAYLOAD_CODE) { // datadog coverage payload
+      workerReportCoverageCh.publish(data)
+      return
+    }
+    if (code === JEST_WORKER_LOGS_PAYLOAD_CODE) { // datadog logs payload
+      workerReportLogsCh.publish(data)
+      return
+    }
+    return onMessage.apply(this, arguments)
+  }
+}
+
+function sendWrapper (send) {
+  return function (request) {
     if (!isKnownTestsEnabled && !isTestManagementTestsEnabled && !isImpactedTestsEnabled) {
       return send.apply(this, arguments)
     }
@@ -1309,30 +1337,47 @@ addHook({
         }
       }
     }
-
     return send.apply(this, arguments)
-  })
-  shimmer.wrap(ChildProcessWorker.prototype, '_onMessage', _onMessage => function () {
-    const [code, data] = arguments[0]
-    if (code === JEST_WORKER_TRACE_PAYLOAD_CODE) { // datadog trace payload
-      sessionAsyncResource.runInAsyncScope(() => {
-        workerReportTraceCh.publish(data)
-      })
-      return
-    }
-    if (code === JEST_WORKER_COVERAGE_PAYLOAD_CODE) { // datadog coverage payload
-      sessionAsyncResource.runInAsyncScope(() => {
-        workerReportCoverageCh.publish(data)
-      })
-      return
-    }
-    if (code === JEST_WORKER_LOGS_PAYLOAD_CODE) { // datadog logs payload
-      sessionAsyncResource.runInAsyncScope(() => {
-        workerReportLogsCh.publish(data)
-      })
-      return
-    }
-    return _onMessage.apply(this, arguments)
-  })
+  }
+}
+
+function enqueueWrapper (enqueue) {
+  return function () {
+    shimmer.wrap(arguments[0], 'onStart', onStart => function (worker) {
+      if (worker && !wrappedWorkers.has(worker)) {
+        shimmer.wrap(worker._child, 'send', sendWrapper)
+        shimmer.wrap(worker, '_onMessage', onMessageWrapper)
+        worker._child.on('message', worker._onMessage.bind(worker))
+        wrappedWorkers.add(worker)
+      }
+      return onStart.apply(this, arguments)
+    })
+    return enqueue.apply(this, arguments)
+  }
+}
+
+/*
+* This hook does three things:
+* - Pass known tests to the workers.
+* - Pass test management tests to the workers.
+* - Receive trace, coverage and logs payloads from the workers.
+*/
+addHook({
+  name: 'jest-worker',
+  versions: ['>=24.9.0 <30.0.0'],
+  file: 'build/workers/ChildProcessWorker.js'
+}, (childProcessWorker) => {
+  const ChildProcessWorker = childProcessWorker.default
+  shimmer.wrap(ChildProcessWorker.prototype, 'send', sendWrapper)
+  shimmer.wrap(ChildProcessWorker.prototype, '_onMessage', onMessageWrapper)
   return childProcessWorker
 })
+
+addHook({
+  name: 'jest-worker',
+  versions: ['>=30.0.0']
+}, (jestWorkerPackage) => {
+  shimmer.wrap(jestWorkerPackage.FifoQueue.prototype, 'enqueue', enqueueWrapper)
+  shimmer.wrap(jestWorkerPackage.PriorityQueue.prototype, 'enqueue', enqueueWrapper)
+  return jestWorkerPackage
+})