dd-trace 3.20.0 → 3.22.0

package/packages/datadog-plugin-cypress/src/plugin.js

@@ -17,13 +17,16 @@ const {
   TEST_COMMAND,
   TEST_MODULE,
   TEST_SOURCE_START,
-  finishAllTraceSpans
+  finishAllTraceSpans,
+  getCoveredFilenamesFromCoverage,
+  getTestSuitePath,
+  addIntelligentTestRunnerSpanTags
 } = require('../../dd-trace/src/plugins/util/test')
+const { ORIGIN_KEY, COMPONENT } = require('../../dd-trace/src/constants')
+const log = require('../../dd-trace/src/log')
 
 const TEST_FRAMEWORK_NAME = 'cypress'
 
-const { ORIGIN_KEY, COMPONENT } = require('../../dd-trace/src/constants')
-
 const CYPRESS_STATUS_TO_TEST_STATUS = {
   passed: 'pass',
   failed: 'fail',
@@ -52,6 +55,13 @@ function getCypressVersion (details) {
   return ''
 }
 
+function getRootDir (details) {
+  if (details && details.config) {
+    return details.config.projectRoot || details.config.repoRoot || process.cwd()
+  }
+  return process.cwd()
+}
+
 function getCypressCommand (details) {
   if (!details) {
     return TEST_FRAMEWORK_NAME
@@ -79,10 +89,62 @@ function getSuiteStatus (suiteStats) {
   return 'pass'
 }
 
+function getItrConfig (tracer, testConfiguration) {
+  return new Promise(resolve => {
+    if (!tracer._tracer._exporter || !tracer._tracer._exporter.getItrConfiguration) {
+      return resolve({ err: new Error('CI Visibility was not initialized correctly') })
+    }
+
+    tracer._tracer._exporter.getItrConfiguration(testConfiguration, (err, itrConfig) => {
+      resolve({ err, itrConfig })
+    })
+  })
+}
+
+function getSkippableTests (isSuitesSkippingEnabled, tracer, testConfiguration) {
+  if (!isSuitesSkippingEnabled) {
+    return Promise.resolve({ skippableTests: [] })
+  }
+  return new Promise(resolve => {
+    if (!tracer._tracer._exporter || !tracer._tracer._exporter.getItrConfiguration) {
+      return resolve({ err: new Error('CI Visibility was not initialized correctly') })
+    }
+    tracer._tracer._exporter.getSkippableSuites(testConfiguration, (err, skippableTests) => {
+      resolve({
+        err,
+        skippableTests
+      })
+    })
+  })
+}
+
 module.exports = (on, config) => {
   const tracer = require('../../dd-trace')
   const testEnvironmentMetadata = getTestEnvironmentMetadata(TEST_FRAMEWORK_NAME)
 
+  const {
+    'git.repository_url': repositoryUrl,
+    'git.commit.sha': sha,
+    'os.version': osVersion,
+    'os.platform': osPlatform,
+    'os.architecture': osArchitecture,
+    'runtime.name': runtimeName,
+    'runtime.version': runtimeVersion,
+    'git.branch': branch
+  } = testEnvironmentMetadata
+
+  const testConfiguration = {
+    repositoryUrl,
+    sha,
+    osVersion,
+    osPlatform,
+    osArchitecture,
+    runtimeName,
+    runtimeVersion,
+    branch,
+    testLevel: 'test'
+  }
+
   const codeOwnersEntries = getCodeOwnersFileEntries()
 
   let activeSpan = null
@@ -91,31 +153,54 @@ module.exports = (on, config) => {
   let testSuiteSpan = null
   let command = null
   let frameworkVersion
+  let rootDir
+  let isSuitesSkippingEnabled = false
+  let isCodeCoverageEnabled = false
+  let testsToSkip = []
 
   on('before:run', (details) => {
-    const childOf = getTestParentSpan(tracer)
+    return getItrConfig(tracer, testConfiguration).then(({ err, itrConfig }) => {
+      if (err) {
+        log.error(err)
+      } else {
+        isSuitesSkippingEnabled = itrConfig.isSuitesSkippingEnabled
+        isCodeCoverageEnabled = itrConfig.isCodeCoverageEnabled
+      }
 
-    command = getCypressCommand(details)
-    frameworkVersion = getCypressVersion(details)
+      getSkippableTests(isSuitesSkippingEnabled, tracer, testConfiguration).then(({ err, skippableTests }) => {
+        if (err) {
+          log.error(err)
+        } else {
+          testsToSkip = skippableTests || []
+        }
 
-    const testSessionSpanMetadata = getTestSessionCommonTags(command, frameworkVersion, TEST_FRAMEWORK_NAME)
-    const testModuleSpanMetadata = getTestModuleCommonTags(command, frameworkVersion, TEST_FRAMEWORK_NAME)
+        const childOf = getTestParentSpan(tracer)
+        rootDir = getRootDir(details)
 
-    testSessionSpan = tracer.startSpan(`${TEST_FRAMEWORK_NAME}.test_session`, {
-      childOf,
-      tags: {
-        [COMPONENT]: TEST_FRAMEWORK_NAME,
-        ...testEnvironmentMetadata,
-        ...testSessionSpanMetadata
-      }
-    })
-    testModuleSpan = tracer.startSpan(`${TEST_FRAMEWORK_NAME}.test_module`, {
-      childOf: testSessionSpan,
-      tags: {
-        [COMPONENT]: TEST_FRAMEWORK_NAME,
-        ...testEnvironmentMetadata,
-        ...testModuleSpanMetadata
-      }
+        command = getCypressCommand(details)
+        frameworkVersion = getCypressVersion(details)
+
+        const testSessionSpanMetadata = getTestSessionCommonTags(command, frameworkVersion, TEST_FRAMEWORK_NAME)
+        const testModuleSpanMetadata = getTestModuleCommonTags(command, frameworkVersion, TEST_FRAMEWORK_NAME)
+
+        testSessionSpan = tracer.startSpan(`${TEST_FRAMEWORK_NAME}.test_session`, {
+          childOf,
+          tags: {
+            [COMPONENT]: TEST_FRAMEWORK_NAME,
+            ...testEnvironmentMetadata,
+            ...testSessionSpanMetadata
+          }
+        })
+        testModuleSpan = tracer.startSpan(`${TEST_FRAMEWORK_NAME}.test_module`, {
+          childOf: testSessionSpan,
+          tags: {
+            [COMPONENT]: TEST_FRAMEWORK_NAME,
+            ...testEnvironmentMetadata,
+            ...testModuleSpanMetadata
+          }
+        })
+        return details
+      })
     })
   })
 
@@ -125,6 +210,16 @@ module.exports = (on, config) => {
       testModuleSpan.setTag(TEST_STATUS, testStatus)
       testSessionSpan.setTag(TEST_STATUS, testStatus)
 
+      addIntelligentTestRunnerSpanTags(
+        testSessionSpan,
+        testModuleSpan,
+        {
+          isSuitesSkipped: !!testsToSkip.length,
+          isSuitesSkippingEnabled,
+          isCodeCoverageEnabled
+        }
+      )
+
       testModuleSpan.finish()
       testSessionSpan.finish()
 
@@ -132,9 +227,15 @@ module.exports = (on, config) => {
     }
 
     return new Promise(resolve => {
-      tracer._tracer._exporter._writer.flush(() => {
-        resolve(null)
-      })
+      if (tracer._tracer._exporter.flush) {
+        tracer._tracer._exporter.flush(() => {
+          resolve(null)
+        })
+      } else {
+        tracer._tracer._exporter._writer.flush(() => {
+          resolve(null)
+        })
+      }
     })
   })
   on('task', {
@@ -153,9 +254,9 @@ module.exports = (on, config) => {
       })
       return null
     },
-    'dd:testSuiteFinish': (suiteStats) => {
+    'dd:testSuiteFinish': (stats) => {
       if (testSuiteSpan) {
-        const status = getSuiteStatus(suiteStats)
+        const status = getSuiteStatus(stats)
         testSuiteSpan.setTag(TEST_STATUS, status)
         testSuiteSpan.finish()
         testSuiteSpan = null
@@ -164,6 +265,12 @@ module.exports = (on, config) => {
     },
     'dd:beforeEach': (test) => {
       const { testName, testSuite } = test
+      // skip test
+      if (testsToSkip.find(test => {
+        return testName === test.name && testSuite === test.suite
+      })) {
+        return { shouldSkip: true }
+      }
 
       const testSuiteTags = {
         [TEST_COMMAND]: command,
@@ -202,11 +309,24 @@ module.exports = (on, config) => {
           }
         })
       }
-      return activeSpan ? activeSpan.context().toTraceId() : null
+      return activeSpan ? { traceId: activeSpan.context().toTraceId() } : {}
     },
-    'dd:afterEach': (test) => {
+    'dd:afterEach': ({ test, coverage }) => {
       const { state, error, isRUMActive, testSourceLine } = test
       if (activeSpan) {
+        if (coverage && tracer._tracer._exporter.exportCoverage && isCodeCoverageEnabled) {
+          const coverageFiles = getCoveredFilenamesFromCoverage(coverage)
+          const relativeCoverageFiles = coverageFiles.map(file => getTestSuitePath(file, rootDir))
+          const { _traceId, _spanId } = testSuiteSpan.context()
+          const formattedCoverage = {
+            sessionId: _traceId,
+            suiteId: _spanId,
+            testId: activeSpan.context()._spanId,
+            files: relativeCoverageFiles
+          }
+          tracer._tracer._exporter.exportCoverage(formattedCoverage)
+        }
+
         activeSpan.setTag(TEST_STATUS, CYPRESS_STATUS_TO_TEST_STATUS[state])
         if (error) {
           activeSpan.setTag('error', error)

package/packages/datadog-plugin-cypress/src/support.js

@@ -1,10 +1,13 @@
 /* eslint-disable */
-beforeEach(() => {
+beforeEach(function () {
   cy.task('dd:beforeEach', {
     testName: Cypress.mocha.getRunner().suite.ctx.currentTest.fullTitle(),
     testSuite: Cypress.mocha.getRootSuite().file
-  }).then(traceId => {
+  }).then(({ traceId, shouldSkip }) => {
     Cypress.env('traceId', traceId)
+    if (shouldSkip) {
+      this.skip()
+    }
   })
 })
 
@@ -36,6 +39,6 @@ afterEach(() => {
     if (win.DD_RUM) {
       testInfo.isRUMActive = true
     }
-    cy.task('dd:afterEach', testInfo)
+    cy.task('dd:afterEach', { test: testInfo, coverage: win.__coverage__ })
   })
 })

package/packages/datadog-plugin-google-cloud-pubsub/src/client.js

@@ -4,15 +4,16 @@ const ClientPlugin = require('../../dd-trace/src/plugins/client')
 
 class GoogleCloudPubsubClientPlugin extends ClientPlugin {
   static get id () { return 'google-cloud-pubsub' }
+  static get type () { return 'messaging' }
   static get operation () { return 'request' }
 
   start ({ request, api, projectId }) {
     if (api === 'publish') return
 
-    this.startSpan('pubsub.request', {
-      service: this.config.service || `${this.tracer._service}-pubsub`,
+    this.startSpan(this.operationName(), {
+      service: this.config.service || this.serviceName(),
       resource: [api, request.name].filter(x => x).join(' '),
-      kind: 'client',
+      kind: this.constructor.kind,
       meta: {
         'pubsub.method': api,
         'gcloud.project_id': projectId

package/packages/datadog-plugin-google-cloud-pubsub/src/consumer.js

@@ -11,11 +11,9 @@ class GoogleCloudPubsubConsumerPlugin extends ConsumerPlugin {
     const topic = subscription.metadata && subscription.metadata.topic
     const childOf = this.tracer.extract('text_map', message.attributes) || null
 
-    this.startSpan('pubsub.receive', {
+    this.startSpan({
       childOf,
-      service: this.config.service,
       resource: topic,
-      kind: 'consumer',
       type: 'worker',
       meta: {
         'gcloud.project_id': subscription.pubsub.projectId,

package/packages/datadog-plugin-google-cloud-pubsub/src/producer.js

@@ -11,10 +11,8 @@ class GoogleCloudPubsubProducerPlugin extends ProducerPlugin {
 
     const messages = request.messages || []
     const topic = request.topic
-    const span = this.startSpan('pubsub.request', { // TODO: rename
-      service: this.config.service || `${this.tracer._service}-pubsub`,
+    const span = this.startSpan({ // TODO: rename
       resource: `${api} ${topic}`,
-      kind: 'producer',
       meta: {
         'gcloud.project_id': projectId,
         'pubsub.method': api, // TODO: remove

package/packages/datadog-plugin-http/src/client.js

@@ -1,6 +1,6 @@
 'use strict'
 
-const Plugin = require('../../dd-trace/src/plugins/plugin')
+const ClientPlugin = require('../../dd-trace/src/plugins/client')
 const { storage } = require('../../datadog-core')
 const tags = require('../../../ext/tags')
 const analyticsSampler = require('../../dd-trace/src/analytics_sampler')
@@ -9,96 +9,99 @@ const HTTP_HEADERS = formats.HTTP_HEADERS
 const urlFilter = require('../../dd-trace/src/plugins/util/urlfilter')
 const log = require('../../dd-trace/src/log')
 const url = require('url')
-const { COMPONENT, ERROR_MESSAGE, ERROR_TYPE, ERROR_STACK } = require('../../dd-trace/src/constants')
+const { CLIENT_PORT_KEY, COMPONENT, ERROR_MESSAGE, ERROR_TYPE, ERROR_STACK } = require('../../dd-trace/src/constants')
 
 const HTTP_STATUS_CODE = tags.HTTP_STATUS_CODE
 const HTTP_REQUEST_HEADERS = tags.HTTP_REQUEST_HEADERS
 const HTTP_RESPONSE_HEADERS = tags.HTTP_RESPONSE_HEADERS
 
-class HttpClientPlugin extends Plugin {
+class HttpClientPlugin extends ClientPlugin {
   static get id () {
     return 'http'
   }
 
-  constructor (...args) {
-    super(...args)
-
-    this.addSub('apm:http:client:request:start', ({ args, http }) => {
-      const store = storage.getStore()
-      const options = args.options
-      const agent = options.agent || options._defaultAgent || http.globalAgent
-      const protocol = options.protocol || agent.protocol || 'http:'
-      const hostname = options.hostname || options.host || 'localhost'
-      const host = options.port ? `${hostname}:${options.port}` : hostname
-      const path = options.path ? options.path.split(/[?#]/)[0] : '/'
-      const uri = `${protocol}//${host}${path}`
-      const allowed = this.config.filter(uri)
-
-      const method = (options.method || 'GET').toUpperCase()
-      const childOf = store && allowed ? store.span : null
-      const span = this.tracer.startSpan('http.request', {
-        childOf,
-        tags: {
-          [COMPONENT]: this.constructor.id,
-          'span.kind': 'client',
-          'service.name': getServiceName(this.tracer, this.config, options),
-          'resource.name': method,
-          'span.type': 'http',
-          'http.method': method,
-          'http.url': uri
-        }
-      })
+  addTraceSub (eventName, handler) {
+    this.addSub(`apm:${this.constructor.id}:client:${this.operation}:${eventName}`, handler)
+  }
 
-      // TODO: Figure out a better way to do this for any span.
-      if (!allowed) {
-        span._spanContext._trace.record = false
+  start ({ args, http }) {
+    const store = storage.getStore()
+    const options = args.options
+    const agent = options.agent || options._defaultAgent || http.globalAgent
+    const protocol = options.protocol || agent.protocol || 'http:'
+    const hostname = options.hostname || options.host || 'localhost'
+    const host = options.port ? `${hostname}:${options.port}` : hostname
+    const path = options.path ? options.path.split(/[?#]/)[0] : '/'
+    const uri = `${protocol}//${host}${path}`
+    const allowed = this.config.filter(uri)
+
+    const method = (options.method || 'GET').toUpperCase()
+    const childOf = store && allowed ? store.span : null
+    // TODO delegate to super.startspan
+    const span = this.startSpan('http.request', {
+      childOf,
+      meta: {
+        [COMPONENT]: this.constructor.id,
+        'span.kind': 'client',
+        'service.name': getServiceName(this.tracer, this.config, options),
+        'resource.name': method,
+        'span.type': 'http',
+        'http.method': method,
+        'http.url': uri,
+        'out.host': hostname
+      },
+      metrics: {
+        [CLIENT_PORT_KEY]: parseInt(options.port)
       }
+    })
 
-      if (!(hasAmazonSignature(options) || !this.config.propagationFilter(uri))) {
-        this.tracer.inject(span, HTTP_HEADERS, options.headers)
-      }
+    // TODO: Figure out a better way to do this for any span.
+    if (!allowed) {
+      span._spanContext._trace.record = false
+    }
 
-      analyticsSampler.sample(span, this.config.measured)
-      this.enter(span, store)
-    })
+    if (!(hasAmazonSignature(options) || !this.config.propagationFilter(uri))) {
+      this.tracer.inject(span, HTTP_HEADERS, options.headers)
+    }
 
-    this.addSub('apm:http:client:request:finish', ({ req, res }) => {
-      const span = storage.getStore().span
-      if (res) {
-        span.setTag(HTTP_STATUS_CODE, res.statusCode)
+    analyticsSampler.sample(span, this.config.measured)
+    this.enter(span, store)
+  }
 
-        if (!this.config.validateStatus(res.statusCode)) {
-          span.setTag('error', 1)
-        }
+  finish ({ req, res }) {
+    const span = storage.getStore().span
+    if (res) {
+      span.setTag(HTTP_STATUS_CODE, res.statusCode)
 
-        addResponseHeaders(res, span, this.config)
+      if (!this.config.validateStatus(res.statusCode)) {
+        span.setTag('error', 1)
       }
 
-      addRequestHeaders(req, span, this.config)
+      addResponseHeaders(res, span, this.config)
+    }
 
-      this.config.hooks.request(span, req, res)
-      span.finish()
-    })
+    addRequestHeaders(req, span, this.config)
 
-    this.addSub('apm:http:client:request:error', errorHandler)
+    this.config.hooks.request(span, req, res)
+    span.finish()
   }
 
-  configure (config) {
-    return super.configure(normalizeClientConfig(config))
-  }
-}
+  error (err) {
+    const span = storage.getStore().span
 
-function errorHandler (err) {
-  const span = storage.getStore().span
+    if (err) {
+      span.addTags({
+        [ERROR_TYPE]: err.name,
+        [ERROR_MESSAGE]: err.message || err.code,
+        [ERROR_STACK]: err.stack
+      })
+    } else {
+      span.setTag('error', 1)
+    }
+  }
 
-  if (err) {
-    span.addTags({
-      [ERROR_TYPE]: err.name,
-      [ERROR_MESSAGE]: err.message || err.code,
-      [ERROR_STACK]: err.stack
-    })
-  } else {
-    span.setTag('error', 1)
+  configure (config) {
+    return super.configure(normalizeClientConfig(config))
   }
 }
 

package/packages/datadog-plugin-http2/src/client.js

@@ -1,7 +1,7 @@
 'use strict'
 
 const { storage } = require('../../datadog-core')
-const Plugin = require('../../dd-trace/src/plugins/plugin')
+const ClientPlugin = require('../../dd-trace/src/plugins/client')
 
 const URL = require('url').URL
 const log = require('../../dd-trace/src/log')
@@ -24,7 +24,7 @@ const HTTP2_HEADER_PATH = ':path'
 const HTTP2_HEADER_STATUS = ':status'
 const HTTP2_METHOD_GET = 'GET'
 
-class Http2ClientPlugin extends Plugin {
+class Http2ClientPlugin extends ClientPlugin {
   static get id () {
     return 'http2'
   }
@@ -32,46 +32,6 @@ class Http2ClientPlugin extends Plugin {
   constructor (...args) {
     super(...args)
 
-    this.addSub('apm:http2:client:request:start', ({ authority, options, headers = {} }) => {
-      const sessionDetails = extractSessionDetails(authority, options)
-      const path = headers[HTTP2_HEADER_PATH] || '/'
-      const pathname = path.split(/[?#]/)[0]
-      const method = headers[HTTP2_HEADER_METHOD] || HTTP2_METHOD_GET
-      const uri = `${sessionDetails.protocol}//${sessionDetails.host}:${sessionDetails.port}${pathname}`
-      const allowed = this.config.filter(uri)
-
-      const store = storage.getStore()
-      const childOf = store && allowed ? store.span : null
-      const span = this.tracer.startSpan('http.request', {
-        childOf,
-        tags: {
-          [COMPONENT]: this.constructor.id,
-          [CLIENT_PORT_KEY]: parseInt(sessionDetails.port),
-          [SPAN_KIND]: CLIENT,
-          'service.name': getServiceName(this.tracer, this.config, sessionDetails),
-          'resource.name': method,
-          'span.type': 'http',
-          'http.method': method,
-          'http.url': uri
-        }
-      })
-
-      // TODO: Figure out a better way to do this for any span.
-      if (!allowed) {
-        span._spanContext._trace.record = false
-      }
-
-      addHeaderTags(span, headers, HTTP_REQUEST_HEADERS, this.config)
-
-      if (!hasAmazonSignature(headers, path)) {
-        this.tracer.inject(span, HTTP_HEADERS, headers)
-      }
-
-      analyticsSampler.sample(span, this.config.measured)
-
-      this.enter(span, store)
-    })
-
     this.addSub('apm:http2:client:response', (headers) => {
       const span = storage.getStore().span
       const status = headers && headers[HTTP2_HEADER_STATUS]
@@ -84,14 +44,58 @@ class Http2ClientPlugin extends Plugin {
 
       addHeaderTags(span, headers, HTTP_RESPONSE_HEADERS, this.config)
     })
+  }
 
-    this.addSub('apm:http2:client:request:finish', () => {
-      const span = storage.getStore().span
+  addTraceSub (eventName, handler) {
+    this.addSub(`apm:${this.constructor.id}:client:${this.operation}:${eventName}`, handler)
+  }
 
-      span.finish()
+  start ({ authority, options, headers = {} }) {
+    const sessionDetails = extractSessionDetails(authority, options)
+    const path = headers[HTTP2_HEADER_PATH] || '/'
+    const pathname = path.split(/[?#]/)[0]
+    const method = headers[HTTP2_HEADER_METHOD] || HTTP2_METHOD_GET
+    const uri = `${sessionDetails.protocol}//${sessionDetails.host}:${sessionDetails.port}${pathname}`
+    const allowed = this.config.filter(uri)
+
+    const store = storage.getStore()
+    const childOf = store && allowed ? store.span : null
+    const span = this.startSpan('http.request', {
+      childOf,
+      meta: {
+        [COMPONENT]: this.constructor.id,
+        [SPAN_KIND]: CLIENT,
+        'service.name': getServiceName(this.tracer, this.config, sessionDetails),
+        'resource.name': method,
+        'span.type': 'http',
+        'http.method': method,
+        'http.url': uri,
+        'out.host': sessionDetails.host
+      },
+      metrics: {
+        [CLIENT_PORT_KEY]: parseInt(sessionDetails.port)
+      }
     })
 
-    this.addSub('apm:http2:client:request:error', this.addError)
+    // TODO: Figure out a better way to do this for any span.
+    if (!allowed) {
+      span._spanContext._trace.record = false
+    }
+
+    addHeaderTags(span, headers, HTTP_REQUEST_HEADERS, this.config)
+
+    if (!hasAmazonSignature(headers, path)) {
+      this.tracer.inject(span, HTTP_HEADERS, headers)
+    }
+
+    analyticsSampler.sample(span, this.config.measured)
+
+    this.enter(span, store)
+  }
+
+  finish () {
+    const span = storage.getStore().span
+    span.finish()
   }
 
   configure (config) {

package/packages/datadog-plugin-jest/src/index.js

@@ -118,8 +118,8 @@ class JestPlugin extends CiPlugin {
 
     this.addSub('ci:jest:worker-report:coverage', data => {
       const formattedCoverages = JSON.parse(data).map(coverage => ({
-        traceId: id(coverage.traceId),
-        spanId: id(coverage.spanId),
+        sessionId: id(coverage.sessionId),
+        suiteId: id(coverage.suiteId),
         files: coverage.files
       }))
       formattedCoverages.forEach(formattedCoverage => {
@@ -150,9 +150,10 @@ class JestPlugin extends CiPlugin {
      * fetching the ITR config.
      */
     this.addSub('ci:jest:test-suite:code-coverage', (coverageFiles) => {
+      const { _traceId, _spanId } = this.testSuiteSpan.context()
       const formattedCoverage = {
-        traceId: this.testSuiteSpan.context()._traceId,
-        spanId: this.testSuiteSpan.context()._spanId,
+        sessionId: _traceId,
+        suiteId: _spanId,
         files: coverageFiles
       }
       this.tracer._exporter.exportCoverage(formattedCoverage)

package/packages/datadog-plugin-jest/src/util.js

@@ -1,3 +1,5 @@
+const { getTestSuitePath } = require('../../dd-trace/src/plugins/util/test')
+
 /**
  * There are two ways to call `test.each` in `jest`:
  * 1. With an array of arrays: https://jestjs.io/docs/api#1-testeachtablename-fn-timeout
@@ -45,4 +47,11 @@ function getJestTestName (test) {
   return titles.join(' ')
 }
 
-module.exports = { getFormattedJestTestParameters, getJestTestName }
+function getJestSuitesToRun (skippableSuites, originalTests, rootDir) {
+  return originalTests.filter(({ path: testPath }) => {
+    const relativePath = getTestSuitePath(testPath, rootDir)
+    return !skippableSuites.includes(relativePath)
+  })
+}
+
+module.exports = { getFormattedJestTestParameters, getJestTestName, getJestSuitesToRun }