-
Notifications
You must be signed in to change notification settings - Fork 51
/
index.ts
94 lines (78 loc) · 2.83 KB
/
index.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import {spawn} from 'child_process'
import {getParser} from './parsers'
import parseOptions from './parse-options'
import Logger from './logger'
function filterArgs (protractorArgs: string[]) {
protractorArgs = protractorArgs.filter((arg) => !/^--(suite|specs)=/.test(arg));
['--suite', '--specs'].forEach((item) => {
let index = protractorArgs.indexOf(item)
if (index !== -1) {
protractorArgs.splice(index, 2)
}
})
return protractorArgs
}
function noop (status: number, output?: string) {}
function flake (options: object | undefined = {}, callback: Function | undefined = noop) {
let testAttempt = 1
let parsedOptions = parseOptions(options)
let parser = getParser(parsedOptions.parser)
let logger = new Logger(parsedOptions.color)
function handleTestEnd (status: number, output = '') {
if (status === 0) {
callback(status)
} else {
if (++testAttempt <= parsedOptions.maxAttempts) {
logger.log('info', `\nUsing ${parser.name} to parse output\n`)
let failedSpecs = parser.parse(output)
logger.log('info', `Re-running tests: test attempt ${testAttempt}\n`)
if (parsedOptions.protractorRetryConfig) {
logger.log('info', `Using provided protractorRetryConfig: ${parsedOptions.protractorRetryConfig}\n`)
}
if (failedSpecs.length === 0) {
logger.log('info', '\nTests failed but no specs were found. All specs will be run again.\n\n')
} else {
logger.log('info', 'Re-running the following test files:\n')
logger.log('info', failedSpecs.join('\n') + '\n')
}
return startProtractor(failedSpecs, true)
}
callback(status, output)
}
}
function startProtractor (specFiles: string[] = [], retry = false) {
let output = ''
let protractorArgs = [parsedOptions.protractorPath].concat(parsedOptions.protractorArgs)
protractorArgs.push('--params.flake.iteration', testAttempt.toString())
if (retry) {
protractorArgs.push('--params.flake.retry', true.toString())
}
if (specFiles.length) {
protractorArgs = filterArgs(protractorArgs)
protractorArgs.push('--specs', specFiles.join(','))
}
if (parsedOptions.protractorRetryConfig && retry) {
protractorArgs.push(parsedOptions.protractorRetryConfig)
}
let protractor = spawn(
parsedOptions.nodeBin,
protractorArgs,
parsedOptions.protractorSpawnOptions
)
protractor.stdout.on('data', (buffer) => {
let text = buffer.toString()
logger.protractor(text)
output = output + text
})
protractor.stderr.on('data', (buffer) => {
let text = buffer.toString()
logger.protractor(text)
output = output + text
})
protractor.on('exit', function (status) {
handleTestEnd(status, output)
})
}
startProtractor()
}
export = flake