Skip to content

Commit f92be45

Browse files
MariovidoMariovido
authored
[test optimization] [SDTEST-1720] Change capabilities tagging to send version number (#5463)
1 parent faea891 commit f92be45

File tree

11 files changed

+137
-133
lines changed

11 files changed

+137
-133
lines changed

‎integration-tests/cucumber/cucumber.spec.js‎

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ const {
5050
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
5151
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
5252
DD_CAPABILITIES_AUTO_TEST_RETRIES,
53+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
54+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
55+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX,
5356
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
5457
TEST_HAS_FAILED_ALL_RETRIES,
5558
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED
@@ -2461,14 +2464,6 @@ versions.forEach(version => {
24612464

24622465
runModes.forEach((runMode) => {
24632466
it(`(${runMode}) adds capabilities to tests`, (done) => {
2464-
receiver.setSettings({
2465-
flaky_test_retries_enabled: true,
2466-
itr_enabled: false,
2467-
early_flake_detection: {
2468-
enabled: true
2469-
},
2470-
known_tests_enabled: false
2471-
})
24722467
const runCommand = runMode === 'parallel' ? parallelModeCommand : runTestsCommand
24732468

24742469
const receiverPromise = receiver
@@ -2480,10 +2475,13 @@ versions.forEach(version => {
24802475
if (runMode === 'parallel') {
24812476
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], undefined)
24822477
} else {
2483-
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], 'false')
2478+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], '1')
24842479
}
2485-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'false')
2486-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'true')
2480+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
2481+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
2482+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
2483+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
2484+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
24872485
// capabilities logic does not overwrite test session name
24882486
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
24892487
})

‎integration-tests/cypress/cypress.spec.js‎

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -42,13 +42,16 @@ const {
4242
TEST_MANAGEMENT_IS_QUARANTINED,
4343
TEST_MANAGEMENT_ENABLED,
4444
TEST_MANAGEMENT_IS_DISABLED,
45-
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
46-
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
47-
DD_CAPABILITIES_AUTO_TEST_RETRIES,
4845
TEST_NAME,
4946
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
5047
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
51-
TEST_HAS_FAILED_ALL_RETRIES
48+
TEST_HAS_FAILED_ALL_RETRIES,
49+
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
50+
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
51+
DD_CAPABILITIES_AUTO_TEST_RETRIES,
52+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
53+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
54+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX
5255
} = require('../../packages/dd-trace/src/plugins/util/test')
5356
const { DD_HOST_CPU_COUNT } = require('../../packages/dd-trace/src/plugins/util/env')
5457
const { ERROR_MESSAGE } = require('../../packages/dd-trace/src/constants')
@@ -2218,23 +2221,18 @@ moduleTypes.forEach(({
22182221

22192222
context('libraries capabilities', () => {
22202223
it('adds capabilities to tests', (done) => {
2221-
receiver.setSettings({
2222-
flaky_test_retries_enabled: false,
2223-
itr_enabled: false,
2224-
early_flake_detection: {
2225-
enabled: true
2226-
},
2227-
known_tests_enabled: true
2228-
})
22292224
const receiverPromise = receiver
22302225
.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('/api/v2/citestcycle'), payloads => {
22312226
const metadataDicts = payloads.flatMap(({ payload }) => payload.metadata)
22322227

22332228
assert.isNotEmpty(metadataDicts)
22342229
metadataDicts.forEach(metadata => {
2235-
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], 'false')
2236-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'true')
2237-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'false')
2230+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], '1')
2231+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
2232+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
2233+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
2234+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
2235+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
22382236
// capabilities logic does not overwrite test session name
22392237
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
22402238
})

‎integration-tests/jest/jest.spec.js‎

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,9 @@ const {
4747
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
4848
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
4949
DD_CAPABILITIES_AUTO_TEST_RETRIES,
50+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
51+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
52+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX,
5053
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
5154
TEST_HAS_FAILED_ALL_RETRIES,
5255
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED
@@ -3468,24 +3471,18 @@ describe('jest CommonJS', () => {
34683471

34693472
context('libraries capabilities', () => {
34703473
it('adds capabilities to tests', (done) => {
3471-
receiver.setSettings({
3472-
flaky_test_retries_enabled: true,
3473-
itr_enabled: false,
3474-
early_flake_detection: {
3475-
enabled: true
3476-
},
3477-
known_tests_enabled: true
3478-
})
3479-
34803474
const eventsPromise = receiver
34813475
.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('/api/v2/citestcycle'), payloads => {
34823476
const metadataDicts = payloads.flatMap(({ payload }) => payload.metadata)
34833477

34843478
assert.isNotEmpty(metadataDicts)
34853479
metadataDicts.forEach(metadata => {
3486-
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], 'false')
3487-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'true')
3488-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'true')
3480+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], '1')
3481+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
3482+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
3483+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
3484+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
3485+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
34893486
// capabilities logic does not overwrite test session name
34903487
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
34913488
})

‎integration-tests/mocha/mocha.spec.js‎

Lines changed: 32 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ const {
4949
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
5050
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
5151
DD_CAPABILITIES_AUTO_TEST_RETRIES,
52+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
53+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
54+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX,
5255
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
5356
TEST_HAS_FAILED_ALL_RETRIES,
5457
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED
@@ -3041,44 +3044,55 @@ describe('mocha CommonJS', function () {
30413044
})
30423045

30433046
context('libraries capabilities', () => {
3044-
it('adds capabilities to tests', (done) => {
3045-
receiver.setSettings({
3046-
flaky_test_retries_enabled: true,
3047-
itr_enabled: true,
3048-
early_flake_detection: {
3049-
enabled: true
3050-
},
3051-
known_tests_enabled: true
3052-
})
3053-
3054-
const eventsPromise = receiver.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('citestcycle'), (payloads) => {
3047+
const getTestAssertions = (isParallel) =>
3048+
receiver.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('citestcycle'), (payloads) => {
30553049
const metadataDicts = payloads.flatMap(({ payload }) => payload.metadata)
30563050

30573051
assert.isNotEmpty(metadataDicts)
30583052
metadataDicts.forEach(metadata => {
3059-
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], 'true')
3060-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'true')
3061-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'true')
3053+
if (isParallel) {
3054+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], undefined)
3055+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], undefined)
3056+
} else {
3057+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], '1')
3058+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
3059+
}
3060+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
3061+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
3062+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
3063+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
30623064
// capabilities logic does not overwrite test session name
30633065
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
30643066
})
30653067
})
30663068

3069+
const runTest = (done, isParallel, extraEnvVars = {}) => {
3070+
const testAssertionsPromise = getTestAssertions(isParallel)
3071+
30673072
childProcess = exec(
30683073
runTestsWithCoverageCommand,
30693074
{
30703075
cwd,
30713076
env: {
30723077
...getCiVisAgentlessConfig(receiver.port),
3073-
DD_TEST_SESSION_NAME: 'my-test-session-name'
3078+
DD_TEST_SESSION_NAME: 'my-test-session-name',
3079+
...extraEnvVars
30743080
},
30753081
stdio: 'inherit'
30763082
}
30773083
)
30783084
childProcess.on('exit', () => {
3079-
eventsPromise.then(() => {
3080-
done()
3081-
}).catch(done)
3085+
testAssertionsPromise.then(() => done()).catch(done)
3086+
})
3087+
}
3088+
3089+
it('adds capabilities to tests', (done) => {
3090+
runTest(done, false)
3091+
})
3092+
3093+
it('adds capabilities to tests (parallel)', (done) => {
3094+
runTest(done, true, {
3095+
RUN_IN_PARALLEL: '1'
30823096
})
30833097
})
30843098
})

‎integration-tests/playwright/playwright.spec.js‎

Lines changed: 8 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,9 @@ const {
3434
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
3535
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
3636
DD_CAPABILITIES_AUTO_TEST_RETRIES,
37+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
38+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
39+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX,
3740
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
3841
TEST_HAS_FAILED_ALL_RETRIES,
3942
TEST_NAME,
@@ -1309,32 +1312,18 @@ versions.forEach((version) => {
13091312

13101313
context('libraries capabilities', () => {
13111314
it('adds capabilities to tests', (done) => {
1312-
receiver.setKnownTests(
1313-
{
1314-
playwright: {
1315-
'passing-test.js': [
1316-
'should work with passing tests'
1317-
]
1318-
}
1319-
}
1320-
)
1321-
receiver.setSettings({
1322-
flaky_test_retries_enabled: true,
1323-
itr_enabled: false,
1324-
early_flake_detection: {
1325-
enabled: true
1326-
},
1327-
known_tests_enabled: true
1328-
})
13291315
const eventsPromise = receiver
13301316
.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('/api/v2/citestcycle'), payloads => {
13311317
const metadataDicts = payloads.flatMap(({ payload }) => payload.metadata)
13321318

13331319
assert.isNotEmpty(metadataDicts)
13341320
metadataDicts.forEach(metadata => {
13351321
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], undefined)
1336-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'true')
1337-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'true')
1322+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
1323+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
1324+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
1325+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
1326+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
13381327
// capabilities logic does not overwrite test session name
13391328
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
13401329
})

‎integration-tests/vitest/vitest.spec.js‎

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,15 @@ const {
3535
TEST_MANAGEMENT_ENABLED,
3636
TEST_MANAGEMENT_IS_QUARANTINED,
3737
TEST_MANAGEMENT_IS_DISABLED,
38+
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
39+
TEST_HAS_FAILED_ALL_RETRIES,
40+
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
3841
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
3942
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
4043
DD_CAPABILITIES_AUTO_TEST_RETRIES,
41-
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
42-
TEST_HAS_FAILED_ALL_RETRIES,
43-
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED
44+
DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE,
45+
DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE,
46+
DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX
4447
} = require('../../packages/dd-trace/src/plugins/util/test')
4548
const { DD_HOST_CPU_COUNT } = require('../../packages/dd-trace/src/plugins/util/env')
4649

@@ -1782,24 +1785,18 @@ versions.forEach((version) => {
17821785

17831786
context('libraries capabilities', () => {
17841787
it('adds capabilities to tests', (done) => {
1785-
receiver.setSettings({
1786-
flaky_test_retries_enabled: false,
1787-
itr_enabled: true,
1788-
early_flake_detection: {
1789-
enabled: true
1790-
},
1791-
known_tests_enabled: true
1792-
})
1793-
17941788
const eventsPromise = receiver
17951789
.gatherPayloadsMaxTimeout(({ url }) => url.endsWith('/api/v2/citestcycle'), payloads => {
17961790
const metadataDicts = payloads.flatMap(({ payload }) => payload.metadata)
17971791

17981792
assert.isNotEmpty(metadataDicts)
17991793
metadataDicts.forEach(metadata => {
18001794
assert.equal(metadata.test[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS], undefined)
1801-
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], 'true')
1802-
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], 'false')
1795+
assert.equal(metadata.test[DD_CAPABILITIES_EARLY_FLAKE_DETECTION], '1')
1796+
assert.equal(metadata.test[DD_CAPABILITIES_AUTO_TEST_RETRIES], '1')
1797+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_QUARANTINE], '1')
1798+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_DISABLE], '1')
1799+
assert.equal(metadata.test[DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX], '2')
18031800
// capabilities logic does not overwrite test session name
18041801
assert.equal(metadata.test[TEST_SESSION_NAME], 'my-test-session-name')
18051802
})

‎packages/datadog-instrumentations/src/vitest.js‎

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -721,15 +721,11 @@ addHook({
721721
// From >=3.0.1, the first arguments changes from a string to an object containing the filepath
722722
const testSuiteAbsolutePath = testPaths[0]?.filepath || testPaths[0]
723723

724-
const { isEarlyFlakeDetectionEnabled, isFlakyTestRetriesEnabled } = getProvidedContext()
725-
726724
const testSuiteAsyncResource = new AsyncResource('bound-anonymous-fn')
727725
testSuiteAsyncResource.runInAsyncScope(() => {
728726
testSuiteStartCh.publish({
729727
testSuiteAbsolutePath,
730-
frameworkVersion,
731-
isFlakyTestRetriesEnabled,
732-
isEarlyFlakeDetectionEnabled
728+
frameworkVersion
733729
})
734730
})
735731
const startTestsResponse = await startTests.apply(this, arguments)

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

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,10 @@ const {
3737
TEST_MANAGEMENT_IS_QUARANTINED,
3838
TEST_MANAGEMENT_ENABLED,
3939
TEST_MANAGEMENT_IS_DISABLED,
40-
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
41-
DD_CAPABILITIES_AUTO_TEST_RETRIES,
42-
DD_CAPABILITIES_TEST_IMPACT_ANALYSIS,
4340
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
4441
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
45-
TEST_HAS_FAILED_ALL_RETRIES
42+
TEST_HAS_FAILED_ALL_RETRIES,
43+
getLibraryCapabilitiesTags
4644
} = require('../../dd-trace/src/plugins/util/test')
4745
const { isMarkedAsUnskippable } = require('../../datadog-plugin-jest/src/util')
4846
const { ORIGIN_KEY, COMPONENT } = require('../../dd-trace/src/constants')
@@ -482,11 +480,10 @@ class CypressPlugin {
482480
[TEST_SESSION_NAME]: testSessionName
483481
}
484482
}
483+
const libraryCapabilitiesTags = getLibraryCapabilitiesTags(this.constructor.id)
485484
metadataTags.test = {
486485
...metadataTags.test,
487-
[DD_CAPABILITIES_TEST_IMPACT_ANALYSIS]: this.isSuitesSkippingEnabled ? 'true' : 'false',
488-
[DD_CAPABILITIES_EARLY_FLAKE_DETECTION]: this.isEarlyFlakeDetectionEnabled ? 'true' : 'false',
489-
[DD_CAPABILITIES_AUTO_TEST_RETRIES]: this.isFlakyTestRetriesEnabled ? 'true' : 'false'
486+
...libraryCapabilitiesTags
490487
}
491488

492489
this.tracer._tracer._exporter.addMetadataTags(metadataTags)

‎packages/datadog-plugin-vitest/src/index.js‎

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,10 @@ const {
2222
TEST_MANAGEMENT_ENABLED,
2323
TEST_MANAGEMENT_IS_QUARANTINED,
2424
TEST_MANAGEMENT_IS_DISABLED,
25-
DD_CAPABILITIES_EARLY_FLAKE_DETECTION,
26-
DD_CAPABILITIES_AUTO_TEST_RETRIES,
2725
TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
2826
TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
29-
TEST_HAS_FAILED_ALL_RETRIES
27+
TEST_HAS_FAILED_ALL_RETRIES,
28+
getLibraryCapabilitiesTags
3029
} = require('../../dd-trace/src/plugins/util/test')
3130
const { COMPONENT } = require('../../dd-trace/src/constants')
3231
const {
@@ -238,9 +237,7 @@ class VitestPlugin extends CiPlugin {
238237

239238
this.addSub('ci:vitest:test-suite:start', ({
240239
testSuiteAbsolutePath,
241-
frameworkVersion,
242-
isFlakyTestRetriesEnabled,
243-
isEarlyFlakeDetectionEnabled
240+
frameworkVersion
244241
}) => {
245242
this.command = process.env.DD_CIVISIBILITY_TEST_COMMAND
246243
this.frameworkVersion = frameworkVersion
@@ -258,10 +255,10 @@ class VitestPlugin extends CiPlugin {
258255
}
259256
}
260257
if (this.tracer._exporter.addMetadataTags) {
258+
const libraryCapabilitiesTags = getLibraryCapabilitiesTags(this.constructor.id)
261259
metadataTags.test = {
262260
...metadataTags.test,
263-
[DD_CAPABILITIES_EARLY_FLAKE_DETECTION]: isEarlyFlakeDetectionEnabled ? 'true' : 'false',
264-
[DD_CAPABILITIES_AUTO_TEST_RETRIES]: isFlakyTestRetriesEnabled ? 'true' : 'false'
261+
...libraryCapabilitiesTags
265262
}
266263
this.tracer._exporter.addMetadataTags(metadataTags)
267264
}

0 commit comments

Comments
 (0)