165 lines
9.3 KiB
JavaScript
165 lines
9.3 KiB
JavaScript
|
'use strict';var _exit;
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
function _load_exit() {return _exit = _interopRequireDefault(require('exit'));}var _run_test;
|
||
|
function _load_run_test() {return _run_test = _interopRequireDefault(require('./run_test'));}var _throat;
|
||
|
function _load_throat() {return _throat = _interopRequireDefault(require('throat'));}var _jestWorker;
|
||
|
function _load_jestWorker() {return _jestWorker = _interopRequireDefault(require('jest-worker'));}function _interopRequireDefault(obj) {return obj && obj.__esModule ? obj : { default: obj };}function _asyncToGenerator(fn) {return function () {var gen = fn.apply(this, arguments);return new Promise(function (resolve, reject) {function step(key, arg) {try {var info = gen[key](arg);var value = info.value;} catch (error) {reject(error);return;}if (info.done) {resolve(value);} else {return Promise.resolve(value).then(function (value) {step("next", value);}, function (err) {step("throw", err);});}}return step("next");});};} /**
|
||
|
* Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
|
||
|
*
|
||
|
* This source code is licensed under the MIT license found in the
|
||
|
* LICENSE file in the root directory of this source tree.
|
||
|
*
|
||
|
*
|
||
|
*/const TEST_WORKER_PATH = require.resolve('./test_worker');class TestRunner {
|
||
|
|
||
|
constructor(globalConfig) {
|
||
|
this._globalConfig = globalConfig;
|
||
|
}
|
||
|
|
||
|
runTests(
|
||
|
tests,
|
||
|
watcher,
|
||
|
onStart,
|
||
|
onResult,
|
||
|
onFailure,
|
||
|
options)
|
||
|
{var _this = this;return _asyncToGenerator(function* () {
|
||
|
return yield options.serial ?
|
||
|
_this._createInBandTestRun(tests, watcher, onStart, onResult, onFailure) :
|
||
|
_this._createParallelTestRun(
|
||
|
tests,
|
||
|
watcher,
|
||
|
onStart,
|
||
|
onResult,
|
||
|
onFailure);})();
|
||
|
|
||
|
}
|
||
|
|
||
|
_createInBandTestRun(
|
||
|
tests,
|
||
|
watcher,
|
||
|
onStart,
|
||
|
onResult,
|
||
|
onFailure)
|
||
|
{var _this2 = this;return _asyncToGenerator(function* () {
|
||
|
const mutex = (0, (_throat || _load_throat()).default)(1);
|
||
|
return tests.reduce(
|
||
|
function (promise, test) {return (
|
||
|
mutex(function () {return (
|
||
|
promise.
|
||
|
then(_asyncToGenerator(function* () {
|
||
|
if (watcher.isInterrupted()) {
|
||
|
throw new CancelRun();
|
||
|
}
|
||
|
|
||
|
yield onStart(test);
|
||
|
return (0, (_run_test || _load_run_test()).default)(
|
||
|
test.path,
|
||
|
_this2._globalConfig,
|
||
|
test.context.config,
|
||
|
test.context.resolver);
|
||
|
|
||
|
})).
|
||
|
then(function (result) {return onResult(test, result);}).
|
||
|
catch(function (err) {return onFailure(test, err);}));}));},
|
||
|
|
||
|
Promise.resolve());})();
|
||
|
|
||
|
}
|
||
|
|
||
|
_createParallelTestRun(
|
||
|
tests,
|
||
|
watcher,
|
||
|
onStart,
|
||
|
onResult,
|
||
|
onFailure)
|
||
|
{var _this3 = this;return _asyncToGenerator(function* () {
|
||
|
// $FlowFixMe: class object is augmented with worker when instantiating.
|
||
|
const worker = new (_jestWorker || _load_jestWorker()).default(TEST_WORKER_PATH, {
|
||
|
exposedMethods: ['worker'],
|
||
|
forkOptions: { stdio: 'inherit' },
|
||
|
maxRetries: 3,
|
||
|
numWorkers: _this3._globalConfig.maxWorkers });
|
||
|
|
||
|
|
||
|
const mutex = (0, (_throat || _load_throat()).default)(_this3._globalConfig.maxWorkers);
|
||
|
|
||
|
// Send test suites to workers continuously instead of all at once to track
|
||
|
// the start time of individual tests.
|
||
|
const runTestInWorker = function (test) {return (
|
||
|
mutex(_asyncToGenerator(function* () {
|
||
|
if (watcher.isInterrupted()) {
|
||
|
return Promise.reject();
|
||
|
}
|
||
|
|
||
|
yield onStart(test);
|
||
|
|
||
|
return worker.worker({
|
||
|
config: test.context.config,
|
||
|
globalConfig: _this3._globalConfig,
|
||
|
path: test.path,
|
||
|
rawModuleMap: watcher.isWatchMode() ?
|
||
|
test.context.moduleMap.getRawModuleMap() :
|
||
|
null });
|
||
|
|
||
|
})));};
|
||
|
|
||
|
const onError = (() => {var _ref3 = _asyncToGenerator(function* (err, test) {
|
||
|
yield onFailure(test, err);
|
||
|
if (err.type === 'ProcessTerminatedError') {
|
||
|
console.error(
|
||
|
'A worker process has quit unexpectedly! ' +
|
||
|
'Most likely this is an initialization error.');
|
||
|
|
||
|
(0, (_exit || _load_exit()).default)(1);
|
||
|
}
|
||
|
});return function onError(_x, _x2) {return _ref3.apply(this, arguments);};})();
|
||
|
|
||
|
const onInterrupt = new Promise(function (_, reject) {
|
||
|
watcher.on('change', function (state) {
|
||
|
if (state.interrupted) {
|
||
|
reject(new CancelRun());
|
||
|
}
|
||
|
});
|
||
|
});
|
||
|
|
||
|
const runAllTests = Promise.all(
|
||
|
tests.map(function (test) {return (
|
||
|
runTestInWorker(test).
|
||
|
then(function (testResult) {return onResult(test, testResult);}).
|
||
|
catch(function (error) {return onError(error, test);}));}));
|
||
|
|
||
|
|
||
|
|
||
|
const cleanup = function () {return worker.end();};
|
||
|
return Promise.race([runAllTests, onInterrupt]).then(cleanup, cleanup);})();
|
||
|
}}
|
||
|
|
||
|
|
||
|
class CancelRun extends Error {
|
||
|
constructor(message) {
|
||
|
super(message);
|
||
|
this.name = 'CancelRun';
|
||
|
}}
|
||
|
|
||
|
|
||
|
module.exports = TestRunner;
|