-
Notifications
You must be signed in to change notification settings - Fork 114
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
6d68c99
commit 8613b6c
Showing
4 changed files
with
246 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -8,3 +8,4 @@ tmp-* | |
| .eslintcache | ||
| .ncu | ||
| package-lock.json | ||
| .vscode | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,222 @@ | ||
| import _ from 'lodash'; | ||
| import { createWriteStream } from 'node:fs'; | ||
| import { appendFile, open, rename } from 'node:fs/promises'; | ||
| import { Transform } from 'node:stream'; | ||
| import { pipeline } from 'node:stream/promises'; | ||
|
|
||
| export async function markFlakyTests(aggregation) { | ||
| try { | ||
| const tests = getFlakyTests(aggregation); | ||
| // group tests by type (ex: parallel, pummel) | ||
| const groupedByType = _.groupBy(tests, ({ file }) => file.split('/', 1)); | ||
|
|
||
| for (const [type, failedTests] of Object.entries(groupedByType)) { | ||
| await editStatusFile(type, failedTests); | ||
| } | ||
| } catch (error) { | ||
| console.error(error); | ||
| } | ||
| }; | ||
|
|
||
| export function getFlakyTests(aggregation) { | ||
| const failedRuns = []; | ||
| const { JS_TEST_FAILURE } = aggregation; | ||
|
|
||
| for (const failedTest of JS_TEST_FAILURE) { | ||
| const { failures } = failedTest; | ||
| if (!failures) continue; | ||
| for (const failure of failures) { | ||
| const { builtOn, file } = failure; | ||
| if (!builtOn) continue; | ||
| const { system, architecture } = parseSystemArchitecture(builtOn); | ||
| failedRuns.push({ | ||
| builtOn, | ||
| file, | ||
| system, | ||
| architecture, | ||
| written: false | ||
| }); | ||
| } | ||
| } | ||
|
|
||
| return failedRuns; | ||
| } | ||
|
|
||
| function matchSystem(rawSystem) { | ||
| let system; | ||
| switch (true) { | ||
| case rawSystem.includes('container'): | ||
| system = 'docker'; | ||
| break; | ||
| case rawSystem.includes('win'): | ||
| system = 'win32'; | ||
| break; | ||
| case rawSystem.includes('fedora'): | ||
| case rawSystem.includes('ubuntu'): | ||
| case rawSystem.includes('rhel'): | ||
| case rawSystem.includes('debian'): | ||
| system = 'linux'; | ||
| break; | ||
| case rawSystem.includes('macos'): | ||
| system = 'macos'; | ||
| break; | ||
| case rawSystem.includes('solaris'): | ||
| case rawSystem.includes('smartos'): | ||
| system = 'solaris'; | ||
| break; | ||
| case rawSystem.includes('freebsd'): | ||
| system = 'freebsd'; | ||
| break; | ||
| case rawSystem.includes('aix72'): | ||
| system = 'aix'; | ||
| break; | ||
| default: | ||
| system = rawSystem; | ||
| break; | ||
| } | ||
|
|
||
| return system; | ||
| } | ||
|
|
||
| function matchArchitecture(rawArchitecture) { | ||
| let architecture; | ||
| switch (true) { | ||
| case rawArchitecture.includes('arm64'): | ||
| architecture = 'arm64'; | ||
| break; | ||
| case rawArchitecture.includes('arm'): | ||
| architecture = 'arm'; | ||
| break; | ||
| case rawArchitecture.includes('s390x'): | ||
| case rawArchitecture.includes('ppc64'): | ||
| architecture = 'ibm'; | ||
| break; | ||
| default: | ||
| architecture = rawArchitecture; | ||
| break; | ||
| } | ||
| return architecture; | ||
| } | ||
|
|
||
| function parseSystemArchitecture(builtOn) { | ||
| const buildInfos = builtOn.split('-'); | ||
| const rawArchitecture = buildInfos[buildInfos.length - 2]; // second last element is architecture | ||
| const rawSystem = buildInfos[buildInfos.length - 3]; // third last element is os | ||
|
|
||
| return { | ||
| architecture: matchArchitecture(rawArchitecture), | ||
| system: matchSystem(rawSystem) | ||
| }; | ||
| } | ||
|
|
||
| async function editStatusFile(type, failedTests) { | ||
| try { | ||
| const groupedByHeaders = _.groupBy(failedTests, (f) => createHeader(f.system, f.architecture)); | ||
| // assume the .status file exists | ||
| const fileName = `./test/${type}/${type}.status`; | ||
| const tmpFile = `${fileName}.tmp`; | ||
| const file = await open(fileName); | ||
|
|
||
| await pipeline( | ||
| file.readLines(), | ||
| new FlakyTestTransfrom(groupedByHeaders), | ||
| createWriteStream(tmpFile) // write output on a temp file | ||
| ); | ||
|
|
||
| // if the header was not present we append it at the end of the file | ||
| await appendTestsWithNewHeader(tmpFile, groupedByHeaders); | ||
|
|
||
| await rename(tmpFile, fileName); | ||
| } catch (error) { | ||
| // file might not exist or error was not parsable | ||
| console.error(error); | ||
| } | ||
| } | ||
|
|
||
| function createHeader(system, architecture) { | ||
| return `[$system==${system} && $arch==${architecture}]`; | ||
| } | ||
|
|
||
| function generateSkipFileLine(file) { | ||
| // take only filename without ex: /parallel/ | ||
| const filename = file.split('/')[1]; | ||
| return `${filename}: PASS, FLAKY`; | ||
| } | ||
|
|
||
| function appendTestsWithNewHeader(tmpFile, groupedByHeaders) { | ||
| const text = []; | ||
|
|
||
| for (const [header, failedTests] of Object.entries(groupedByHeaders)) { | ||
| // skip if there isnt at least one failedTest with written false or no failedTests | ||
| if (!failedTests?.length || !failedTests.some(f => f.written === false)) continue; | ||
|
|
||
| // add space on top of header | ||
| text.push('\n' + header); | ||
|
|
||
| // add test lines in a set to avoid duplicates | ||
| const newLines = new Set(); | ||
| for (const failedTest of failedTests) { | ||
| // skip tests we have already been written because we found the header | ||
| if (failedTest.written) continue; | ||
| newLines.add(generateSkipFileLine(failedTest.file)); | ||
| } | ||
| text.push(...newLines); | ||
| } | ||
|
|
||
| return appendFile(tmpFile, text.join('\n')); | ||
| } | ||
|
|
||
| class FlakyTestTransfrom extends Transform { | ||
| constructor(groupedByHeaders) { | ||
| super(); | ||
| this.groupedByHeaders = groupedByHeaders; | ||
| this.bufferedLines = []; | ||
| this.uniqueTests = new Set(); | ||
| } | ||
|
|
||
| _transform(chunk, _encoding, callback) { | ||
| const chunkStringified = chunk.toString(); | ||
|
|
||
| // keys of groupedByHeaders are [$system==win32 && $arch==arm64] | ||
| const failedTests = this.groupedByHeaders[chunkStringified]; | ||
|
|
||
| if (failedTests?.length) { | ||
| // when we hit a new header, push bufferedLines | ||
| this.push(this.bufferedLinesToString()); | ||
|
|
||
| // reset | ||
| this.bufferedLines = []; | ||
| this.uniqueTests = new Set(); | ||
|
|
||
| // header first | ||
| this.bufferedLines.push(chunkStringified); | ||
|
|
||
| for (const failedTest of failedTests) { | ||
| // set written to true because there was an existing header | ||
| failedTest.written = true; | ||
| const skipFileLine = generateSkipFileLine(failedTest.file); | ||
| this.bufferedLines.push(skipFileLine); | ||
| this.uniqueTests.add(skipFileLine); | ||
| } | ||
| // this is a non header line | ||
| // we check if its a test we have added already | ||
| } else if (!this.uniqueTests.has(chunkStringified)) { | ||
| this.bufferedLines.push(chunkStringified); | ||
| } | ||
|
|
||
| callback(); | ||
| } | ||
|
|
||
| bufferedLinesToString() { | ||
| if (this.bufferedLines.length > 0) { | ||
| return this.bufferedLines.join('\n') + '\n'; | ||
| } | ||
| return '\n'; | ||
| } | ||
|
|
||
| _flush(callback) { | ||
| // Push any remaining buffered lines | ||
| this.push(this.bufferedLinesToString()); | ||
| callback(); | ||
| } | ||
| } |