--wip-- [skip ci]

Author: SimenB

packages/jest-runner/build/index.js

@@ -0,0 +1,220 @@
+'use strict';
+
+function _exit() {
+  const data = _interopRequireDefault(require('exit'));
+
+  _exit = function _exit() {
+    return data;
+  };
+
+  return data;
+}
+
+function _throat() {
+  const data = _interopRequireDefault(require('throat'));
+
+  _throat = function _throat() {
+    return data;
+  };
+
+  return data;
+}
+
+function _jestWorker() {
+  const data = _interopRequireDefault(require('jest-worker'));
+
+  _jestWorker = function _jestWorker() {
+    return data;
+  };
+
+  return data;
+}
+
+var _runTest = _interopRequireDefault(require('./runTest'));
+
+var _testWorker = require('./testWorker');
+
+function _interopRequireDefault(obj) {
+  return obj && obj.__esModule ? obj : {default: obj};
+}
+
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
+  try {
+    var info = gen[key](arg);
+    var value = info.value;
+  } catch (error) {
+    reject(error);
+    return;
+  }
+  if (info.done) {
+    resolve(value);
+  } else {
+    Promise.resolve(value).then(_next, _throw);
+  }
+}
+
+function _asyncToGenerator(fn) {
+  return function() {
+    var self = this,
+      args = arguments;
+    return new Promise(function(resolve, reject) {
+      var gen = fn.apply(self, args);
+      function _next(value) {
+        asyncGeneratorStep(gen, resolve, reject, _next, _throw, 'next', value);
+      }
+      function _throw(err) {
+        asyncGeneratorStep(gen, resolve, reject, _next, _throw, 'throw', err);
+      }
+      _next(undefined);
+    });
+  };
+}
+
+const TEST_WORKER_PATH = require.resolve('./testWorker');
+
+class TestRunner {
+  constructor(globalConfig, context) {
+    this._globalConfig = globalConfig;
+    this._context = context || {};
+  }
+
+  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*() {
+      process.env.JEST_WORKER_ID = '1';
+      const mutex = (0, _throat().default)(1);
+      return tests.reduce(
+        (promise, test) =>
+          mutex(() =>
+            promise
+              .then(
+                /*#__PURE__*/
+                _asyncToGenerator(function*() {
+                  if (watcher.isInterrupted()) {
+                    throw new CancelRun();
+                  }
+
+                  yield onStart(test);
+                  return (0,
+                  _runTest.default)(test.path, _this2._globalConfig, test.context.config, test.context.resolver, _this2._context);
+                })
+              )
+              .then(result => onResult(test, result))
+              .catch(err => onFailure(test, err))
+          ),
+        Promise.resolve()
+      );
+    })();
+  }
+
+  _createParallelTestRun(tests, watcher, onStart, onResult, onFailure) {
+    var _this3 = this;
+
+    return _asyncToGenerator(function*() {
+      const worker = new (_jestWorker()).default(TEST_WORKER_PATH, {
+        exposedMethods: ['worker'],
+        forkOptions: {
+          stdio: 'pipe'
+        },
+        maxRetries: 3,
+        numWorkers: _this3._globalConfig.maxWorkers
+      });
+      if (worker.getStdout()) worker.getStdout().pipe(process.stdout);
+      if (worker.getStderr()) worker.getStderr().pipe(process.stderr);
+      const mutex = (0, _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 = test =>
+        mutex(
+          /*#__PURE__*/
+          _asyncToGenerator(function*() {
+            if (watcher.isInterrupted()) {
+              return Promise.reject();
+            }
+
+            yield onStart(test);
+            return worker.worker({
+              config: test.context.config,
+              context: _this3._context,
+              globalConfig: _this3._globalConfig,
+              path: test.path,
+              serializableModuleMap: watcher.isWatchMode()
+                ? test.context.moduleMap.toJSON()
+                : null
+            });
+          })
+        );
+
+      const onError =
+        /*#__PURE__*/
+        (function() {
+          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().default)(1);
+            }
+          });
+
+          return function onError(_x, _x2) {
+            return _ref3.apply(this, arguments);
+          };
+        })();
+
+      const onInterrupt = new Promise((_, reject) => {
+        watcher.on('change', state => {
+          if (state.interrupted) {
+            reject(new CancelRun());
+          }
+        });
+      });
+      const runAllTests = Promise.all(
+        tests.map(test =>
+          runTestInWorker(test)
+            .then(testResult => onResult(test, testResult))
+            .catch(error => onError(error, test))
+        )
+      );
+
+      const cleanup = () => worker.end();
+
+      return Promise.race([runAllTests, onInterrupt]).then(cleanup, cleanup);
+    })();
+  }
+}
+
+class CancelRun extends Error {
+  constructor(message) {
+    super(message);
+    this.name = 'CancelRun';
+  }
+}
+
+module.exports = TestRunner;