gitlab-org--gitlab-foss/scripts/frontend/test.js
2019-06-19 16:05:46 +00:00

123 lines
3.3 KiB
JavaScript
Executable file

#!/usr/bin/env node
const { spawn } = require('child_process');
const { EOL } = require('os');
const program = require('commander');
const chalk = require('chalk');
const SUCCESS_CODE = 0;
const JEST_ROUTE = 'spec/frontend';
const KARMA_ROUTE = 'spec/javascripts';
const COMMON_ARGS = ['--colors'];
const jestArgs = [...COMMON_ARGS, '--passWithNoTests'];
const karmaArgs = [...COMMON_ARGS, '--no-fail-on-empty-test-suite'];
program
.usage('[options] <file ...>')
.option('-p, --parallel', 'Run tests suites in parallel')
.option(
'-w, --watch',
'Rerun tests when files change (tests will be run in parallel if this enabled)',
)
.parse(process.argv);
const shouldParallelize = program.parallel || program.watch;
const isSuccess = code => code === SUCCESS_CODE;
const combineExitCodes = codes => {
const firstFail = codes.find(x => !isSuccess(x));
return firstFail === undefined ? SUCCESS_CODE : firstFail;
};
const skipIfFail = fn => code => (isSuccess(code) ? fn() : code);
const endWithEOL = str => (str[str.length - 1] === '\n' ? str : `${str}${EOL}`);
const runTests = paths => {
if (shouldParallelize) {
return Promise.all([runJest(paths), runKarma(paths)]).then(combineExitCodes);
} else {
return runJest(paths).then(skipIfFail(() => runKarma(paths)));
}
};
const spawnYarnScript = (cmd, args) => {
return new Promise((resolve, reject) => {
const proc = spawn('yarn', ['run', cmd, ...args]);
const output = data => {
const text = data
.toString()
.split(/\r?\n/g)
.map((line, idx, { length }) =>
idx === length - 1 && !line ? line : `${chalk.gray(cmd)}: ${line}`,
)
.join(EOL);
return endWithEOL(text);
};
proc.stdout.on('data', data => {
process.stdout.write(output(data));
});
proc.stderr.on('data', data => {
process.stderr.write(output(data));
});
proc.on('close', code => {
process.stdout.write(output(`exited with code ${code}`));
// We resolve even on a failure code because a `reject` would cause
// Promise.all to reject immediately (without waiting for other promises)
// to finish.
resolve(code);
});
});
};
const runJest = args => {
return spawnYarnScript('jest', [...jestArgs, ...toJestArgs(args)]);
};
const runKarma = args => {
return spawnYarnScript('karma', [...karmaArgs, ...toKarmaArgs(args)]);
};
const replacePath = to => path =>
path
.replace(JEST_ROUTE, to)
.replace(KARMA_ROUTE, to)
.replace('app/assets/javascripts', to);
const replacePathForJest = replacePath(JEST_ROUTE);
const replacePathForKarma = replacePath(KARMA_ROUTE);
const toJestArgs = paths => paths.map(replacePathForJest);
const toKarmaArgs = paths =>
paths.reduce((acc, path) => acc.concat('-f', replacePathForKarma(path)), []);
const main = paths => {
if (program.watch) {
jestArgs.push('--watch');
karmaArgs.push('--single-run', 'false', '--auto-watch');
}
runTests(paths).then(code => {
console.log('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~');
if (isSuccess(code)) {
console.log(chalk.bgGreen(chalk.black('All tests passed :)')));
} else {
console.log(chalk.bgRed(chalk.white(`Some tests failed :(`)));
}
console.log('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~');
if (!isSuccess(code)) {
process.exit(code);
}
});
};
main(program.args);