creevey 0.8.1-sb7.1 → 0.8.1-sb7.3

package/lib/cjs/server/storybook/providers/nodejs.js

@@ -0,0 +1,239 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.extractStoriesData = extractStoriesData;
+exports.loadStories = void 0;
+
+var _path = _interopRequireDefault(require("path"));
+
+var _cluster = _interopRequireDefault(require("cluster"));
+
+var _chokidar = _interopRequireDefault(require("chokidar"));
+
+var _types = require("../../../types");
+
+var _utils = require("../../utils");
+
+var _messages = require("../../messages");
+
+var _helpers = require("../helpers");
+
+var _logger = require("../../logger");
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
+
+async function initStorybookEnvironment() {
+  // @ts-expect-error There is no @types/global-jsdom package
+  (await Promise.resolve().then(() => _interopRequireWildcard(require('global-jsdom')))).default(undefined, {
+    url: 'http://localhost'
+  }); // NOTE Cutoff `jsdom` part from userAgent, because storybook check enviroment and create events channel if runs in browser
+  // https://github.com/storybookjs/storybook/blob/v5.2.8/lib/core/src/client/preview/start.js#L98
+  // Example: "Mozilla/5.0 (linux) AppleWebKit/537.36 (KHTML, like Gecko) jsdom/15.2.1"
+
+  Object.defineProperty(window.navigator, 'userAgent', {
+    value: window.navigator.userAgent.split(' ').filter(token => !token.startsWith('jsdom')).join(' ')
+  }); // TODO Look at creevey debug flag
+
+  const {
+    logger
+  } = await (0, _helpers.importStorybookClientLogger)(); // NOTE: Disable duplication warnings for >=6.2 storybook
+
+  if (_cluster.default.isWorker) logger.warn = _types.noop; // NOTE: disable logger for 5.x storybook
+
+  logger.debug = _types.noop;
+  return Promise.resolve().then(() => _interopRequireWildcard(require('../entry')));
+}
+
+function watchStories(channel, watcher, initialFiles) {
+  const watchingFiles = initialFiles;
+  let storiesByFiles = new Map();
+  (0, _messages.subscribeOn)('shutdown', () => void watcher.close());
+  watcher.add(Array.from(watchingFiles));
+  watcher.on('change', filePath => storiesByFiles.set(_path.default.isAbsolute(filePath) ? filePath : `./${filePath.replace(/\\/g, '/')}`, []));
+  watcher.on('unlink', filePath => storiesByFiles.set(_path.default.isAbsolute(filePath) ? filePath : `./${filePath.replace(/\\/g, '/')}`, []));
+  return data => {
+    const stories = data.stories;
+    const files = new Set(Object.values(stories).map(story => story.parameters.fileName));
+    const addedFiles = Array.from(files).filter(filePath => !watchingFiles.has(filePath));
+    const removedFiles = Array.from(watchingFiles).filter(filePath => !files.has(filePath));
+    watcher.add(addedFiles);
+    addedFiles.forEach(filePath => {
+      watchingFiles.add(filePath);
+      storiesByFiles.set(filePath, []);
+    });
+    removedFiles.forEach(filePath => watchingFiles.delete(filePath));
+    Object.values(stories).forEach(story => {
+      var _storiesByFiles$get;
+
+      return (_storiesByFiles$get = storiesByFiles.get(story.parameters.fileName)) === null || _storiesByFiles$get === void 0 ? void 0 : _storiesByFiles$get.push(story);
+    });
+    channel.emit('storiesUpdated', storiesByFiles);
+    storiesByFiles = new Map();
+  };
+}
+
+function loadStoriesFromBundle(watch) {
+  const bundlePath = _path.default.join((0, _utils.getCreeveyCache)(), 'storybook/main.js');
+
+  if (watch) {
+    (0, _messages.subscribeOn)('webpack', message => {
+      if (message.type != 'rebuild succeeded') return;
+      Object.values(global.__CREEVEY_HMR_DATA__).filter(({
+        callback
+      }) => callback).forEach(({
+        data,
+        callback
+      }) => callback(data));
+      delete require.cache[bundlePath];
+      Promise.resolve(`${bundlePath}`).then(s => _interopRequireWildcard(require(s)));
+    });
+  }
+
+  Promise.resolve(`${bundlePath}`).then(s => _interopRequireWildcard(require(s)));
+}
+
+async function loadStoriesDirectly(config, {
+  watcher,
+  debug
+}) {
+  const {
+    toRequireContext,
+    normalizeStoriesEntry
+  } = await (0, _helpers.importStorybookCoreCommon)();
+  const {
+    addParameters,
+    configure
+  } = await Promise.resolve().then(() => _interopRequireWildcard(require('../entry')));
+  const requireContext = await (await Promise.resolve().then(() => _interopRequireWildcard(require('../../loaders/babel/register')))).default(config, debug);
+
+  const preview = (() => {
+    try {
+      return require.resolve(`${config.storybookDir}/preview`);
+    } catch (_) {
+      /* noop */
+    }
+  })();
+
+  const {
+    stories
+  } = await (0, _helpers.importStorybookConfig)();
+  const contexts = stories.map(entry => {
+    const normalizedEntry = normalizeStoriesEntry(entry, {
+      configDir: config.storybookDir,
+      workingDir: process.cwd()
+    });
+    const {
+      path: storiesPath,
+      recursive,
+      match
+    } = toRequireContext(normalizedEntry);
+    watcher === null || watcher === void 0 ? void 0 : watcher.add(_path.default.resolve(config.storybookDir, storiesPath));
+    return () => requireContext(storiesPath, recursive, new RegExp(match));
+  });
+
+  let disposeCallback = data => void data;
+
+  Object.assign(module, {
+    hot: {
+      data: {},
+
+      accept() {
+        /* noop */
+      },
+
+      dispose(callback) {
+        disposeCallback = callback;
+      }
+
+    }
+  });
+
+  async function startStorybook() {
+    if (preview) {
+      const {
+        parameters,
+        globals,
+        globalTypes
+      } = await Promise.resolve(`${preview}`).then(s => _interopRequireWildcard(require(s)));
+      if (parameters) addParameters(parameters);
+      if (globals) addParameters({
+        globals
+      });
+      if (globalTypes) addParameters({
+        globalTypes
+      });
+    }
+
+    try {
+      configure(contexts.map(ctx => ctx()), module, false);
+    } catch (error) {
+      if (_cluster.default.isPrimary) _logger.logger.error(error);
+    }
+  }
+
+  watcher === null || watcher === void 0 ? void 0 : watcher.add(config.storybookDir);
+  watcher === null || watcher === void 0 ? void 0 : watcher.on('all', (_event, filename) => {
+    var _module$hot;
+
+    disposeCallback((_module$hot = module.hot) === null || _module$hot === void 0 ? void 0 : _module$hot.data);
+    delete require.cache[filename];
+    void startStorybook();
+  });
+  void startStorybook();
+} // TODO Do we need to support multiple storybooks here?
+
+
+const loadStories = async (config, {
+  watch,
+  debug
+}, storiesListener) => {
+  const storybookApi = await initStorybookEnvironment();
+  const Events = await (0, _helpers.importStorybookCoreEvents)();
+  const {
+    channel
+  } = storybookApi;
+  channel.removeAllListeners(Events.CURRENT_STORY_WAS_SET);
+  channel.on('storiesUpdated', storiesListener);
+  let watcher;
+  if (watch) watcher = _chokidar.default.watch([], {
+    ignoreInitial: true
+  });
+  const loadPromise = new Promise(resolve => {
+    channel.once(Events.SET_STORIES, data => {
+      const stories = data.stories;
+      const files = new Set(Object.values(stories).map(story => story.parameters.fileName));
+      if (watcher) channel.on(Events.SET_STORIES, watchStories(channel, watcher, files));
+      resolve(stories);
+    });
+  });
+  if (config.useWebpackToExtractTests) loadStoriesFromBundle(watch);else void loadStoriesDirectly(config, {
+    watcher,
+    debug
+  });
+  return loadPromise;
+};
+
+exports.loadStories = loadStories;
+
+async function extractStoriesData(config, {
+  watch,
+  debug
+}) {
+  const storybookApi = await initStorybookEnvironment();
+  const Events = await (0, _helpers.importStorybookCoreEvents)();
+  const {
+    channel
+  } = storybookApi;
+  channel.removeAllListeners(Events.CURRENT_STORY_WAS_SET);
+  const loadPromise = new Promise(resolve => channel.once(Events.SET_STORIES, resolve));
+  if (config.useWebpackToExtractTests) loadStoriesFromBundle(watch);else void loadStoriesDirectly(config, {
+    debug
+  });
+  return loadPromise;
+}

package/lib/cjs/server/update.js

@@ -0,0 +1,79 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = update;
+
+var _path = _interopRequireDefault(require("path"));
+
+var _fs = _interopRequireWildcard(require("fs"));
+
+var _micromatch = _interopRequireDefault(require("micromatch"));
+
+var _types = require("../types");
+
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+function tryToLoadTestsData(filename) {
+  try {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    return require(filename);
+  } catch (_) {
+    /* noop */
+  }
+}
+
+const actualRegex = /^(.*)-actual-(\d+)\.png$/i;
+
+function approve(dirents, srcPath, dstPath, testPaths, isMatch) {
+  dirents.filter(dirent => dirent.isFile()).map(dirent => actualRegex.exec(dirent.name)).filter(_types.isDefined).filter(([fileName, imageName]) => !testPaths || testPaths.find(([token]) => token == imageName) && isMatch(_path.default.join(srcPath, fileName))).reduce((images, [, imageName, retry]) => Number(retry) > (images.get(imageName) ?? -1) ? images.set(imageName, Number(retry)) : images, new Map()).forEach((retry, imageName) => {
+    (0, _fs.mkdirSync)(dstPath, {
+      recursive: true
+    });
+
+    _fs.default.copyFileSync(_path.default.join(srcPath, `${imageName}-actual-${retry}.png`), _path.default.join(dstPath, `${imageName}.png`));
+  });
+}
+
+function traverse(srcPath, dstPath, testPaths, isMatch) {
+  const dirents = _fs.default.readdirSync(srcPath, {
+    withFileTypes: true
+  });
+
+  approve(dirents, srcPath, dstPath, testPaths, isMatch);
+  dirents.filter(dirent => dirent.isDirectory()).map(dirent => [dirent.name, testPaths === null || testPaths === void 0 ? void 0 : testPaths.map(([token, ...restPath]) => token == dirent.name ? restPath : null).filter(_types.isDefined)]).filter(([, paths]) => !paths || paths.length > 0).forEach(([dirname, paths]) => traverse(_path.default.join(srcPath, dirname), _path.default.join(dstPath, dirname), paths, isMatch));
+}
+
+function update(config, grepPattern) {
+  const {
+    reportDir,
+    screenDir
+  } = config;
+  const isMatch = grepPattern ? _micromatch.default.matcher(grepPattern, {
+    contains: true
+  }) : () => true;
+  const testsMeta = tryToLoadTestsData(`${reportDir}/tests.json`);
+  const testsReport = tryToLoadTestsData(`${reportDir}/data`);
+  let testPaths = null;
+
+  if (testsMeta && testsReport) {
+    testPaths = Object.values(testsMeta).filter(_types.isDefined).filter(({
+      id
+    }) => {
+      var _testsReport$id;
+
+      return ((_testsReport$id = testsReport[id]) === null || _testsReport$id === void 0 ? void 0 : _testsReport$id.status) == 'failed';
+    }).map(({
+      storyPath,
+      testName,
+      browser
+    }) => [...storyPath, ...(testName ? [testName] : []), browser]);
+  }
+
+  traverse(reportDir, screenDir, testPaths, value => isMatch(_path.default.relative(reportDir, value)));
+}

package/lib/cjs/server/utils.js

@@ -0,0 +1,169 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.extensions = exports.downloadBinary = exports.LOCALHOST_REGEXP = void 0;
+exports.getCreeveyCache = getCreeveyCache;
+exports.isShuttingDown = exports.isInsideDocker = void 0;
+exports.removeProps = removeProps;
+exports.runSequence = runSequence;
+exports.shouldSkip = shouldSkip;
+exports.shouldSkipByOption = shouldSkipByOption;
+exports.shutdown = shutdown;
+exports.shutdownWorkers = shutdownWorkers;
+exports.skipOptionKeys = void 0;
+exports.testsToImages = testsToImages;
+
+var _fs = require("fs");
+
+var _cluster = _interopRequireDefault(require("cluster"));
+
+var _types = require("../types");
+
+var _messages = require("./messages");
+
+var _findCacheDir = _interopRequireDefault(require("find-cache-dir"));
+
+var _https = require("https");
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+const isShuttingDown = {
+  current: false
+};
+exports.isShuttingDown = isShuttingDown;
+const LOCALHOST_REGEXP = /(localhost|127\.0\.0\.1)/i;
+exports.LOCALHOST_REGEXP = LOCALHOST_REGEXP;
+const extensions = ['.js', '.jsx', '.ts', '.tsx'];
+exports.extensions = extensions;
+const skipOptionKeys = ['in', 'kinds', 'stories', 'tests', 'reason'];
+exports.skipOptionKeys = skipOptionKeys;
+
+function matchBy(pattern, value) {
+  return typeof pattern == 'string' && pattern == value || Array.isArray(pattern) && pattern.includes(value) || pattern instanceof RegExp && pattern.test(value) || !(0, _types.isDefined)(pattern);
+}
+
+function shouldSkip(browser, meta, skipOptions, test) {
+  if (typeof skipOptions != 'object') {
+    return skipOptions;
+  }
+
+  for (const skipKey in skipOptions) {
+    const reason = shouldSkipByOption(browser, meta, skipOptions[skipKey], skipKey, test);
+    if (reason) return reason;
+  }
+
+  return false;
+}
+
+function shouldSkipByOption(browser, meta, skipOption, reason, test) {
+  if (Array.isArray(skipOption)) {
+    for (const skip of skipOption) {
+      const result = shouldSkipByOption(browser, meta, skip, reason, test);
+      if (result) return result;
+    }
+
+    return false;
+  }
+
+  const {
+    in: browsers,
+    kinds,
+    stories,
+    tests
+  } = skipOption;
+  const {
+    kind,
+    story
+  } = meta;
+  const skipByBrowser = matchBy(browsers, browser);
+  const skipByKind = matchBy(kinds, kind);
+  const skipByStory = matchBy(stories, story);
+  const skipByTest = !(0, _types.isDefined)(test) || matchBy(tests, test);
+  return skipByBrowser && skipByKind && skipByStory && skipByTest && reason;
+}
+
+async function shutdownWorkers() {
+  isShuttingDown.current = true;
+  await Promise.all(Object.values(_cluster.default.workers ?? {}).filter(_types.isDefined).filter(worker => worker.isConnected()).map(worker => new Promise(resolve => {
+    const timeout = setTimeout(() => worker.kill(), 10000);
+    worker.on('exit', () => {
+      clearTimeout(timeout);
+      resolve();
+    });
+    (0, _messages.sendShutdownMessage)(worker);
+  })));
+  (0, _messages.emitShutdownMessage)();
+}
+
+function shutdown() {
+  // eslint-disable-next-line no-process-exit
+  process.exit();
+}
+
+function getCreeveyCache() {
+  return (0, _findCacheDir.default)({
+    name: 'creevey',
+    cwd: __dirname
+  });
+}
+
+async function runSequence(seq, predicate) {
+  for (const fn of seq) {
+    if (predicate()) await fn();
+  }
+}
+
+function testsToImages(tests) {
+  return new Set([].concat(...tests.filter(_types.isDefined).map(({
+    browser,
+    testName,
+    storyPath,
+    results
+  }) => {
+    var _results$slice$;
+
+    return Object.keys((results === null || results === void 0 ? void 0 : (_results$slice$ = results.slice(-1)[0]) === null || _results$slice$ === void 0 ? void 0 : _results$slice$.images) ?? {}).map(image => `${[...storyPath, testName, browser, browser == image ? undefined : image].filter(_types.isDefined).join('/')}.png`);
+  })));
+} // https://tuhrig.de/how-to-know-you-are-inside-a-docker-container/
+
+
+const isInsideDocker = (0, _fs.existsSync)('/proc/1/cgroup') && /docker/.test((0, _fs.readFileSync)('/proc/1/cgroup', 'utf8'));
+exports.isInsideDocker = isInsideDocker;
+
+const downloadBinary = (downloadUrl, destination) => new Promise((resolve, reject) => (0, _https.get)(downloadUrl, response => {
+  if (response.statusCode == 302) {
+    const {
+      location
+    } = response.headers;
+    if (!location) return reject(new Error(`Couldn't download selenoid. Status code: ${response.statusCode ?? 'UNKNOWN'}`));
+    return resolve(downloadBinary(location, destination));
+  }
+
+  if (response.statusCode != 200) return reject(new Error(`Couldn't download selenoid. Status code: ${response.statusCode ?? 'UNKNOWN'}`));
+  const fileStream = (0, _fs.createWriteStream)(destination);
+  response.pipe(fileStream);
+  fileStream.on('finish', () => {
+    fileStream.close();
+    resolve();
+  });
+  fileStream.on('error', error => {
+    (0, _fs.unlink)(destination, _types.noop);
+    reject(error);
+  });
+}));
+
+exports.downloadBinary = downloadBinary;
+
+function removeProps(obj, propPath) {
+  const [prop, ...restPath] = propPath;
+
+  if (restPath.length > 0) {
+    if (typeof prop == 'string') obj[prop] && removeProps(obj[prop], restPath);
+    if ((0, _types.isFunction)(prop)) Object.keys(obj).filter(prop).forEach(key => obj[key] && removeProps(obj[key], restPath));
+  } else {
+    if (typeof prop == 'string') delete obj[prop];
+    if ((0, _types.isFunction)(prop)) Object.keys(obj).filter(prop).forEach(key => delete obj[key]);
+  }
+}

package/lib/cjs/server/worker/chai-image.js

@@ -0,0 +1,142 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = _default;
+
+var _pngjs = require("pngjs");
+
+var _pixelmatch = _interopRequireDefault(require("pixelmatch"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+function normalizeImageSize(image, width, height) {
+  const normalizedImage = Buffer.alloc(4 * width * height);
+
+  for (let y = 0; y < height; y++) {
+    for (let x = 0; x < width; x++) {
+      const i = (y * width + x) * 4;
+
+      if (x < image.width && y < image.height) {
+        const j = (y * image.width + x) * 4;
+        normalizedImage[i + 0] = image.data[j + 0];
+        normalizedImage[i + 1] = image.data[j + 1];
+        normalizedImage[i + 2] = image.data[j + 2];
+        normalizedImage[i + 3] = image.data[j + 3];
+      } else {
+        normalizedImage[i + 0] = 0;
+        normalizedImage[i + 1] = 0;
+        normalizedImage[i + 2] = 0;
+        normalizedImage[i + 3] = 0;
+      }
+    }
+  }
+
+  return normalizedImage;
+}
+
+function hasDiffPixels(diff) {
+  for (let i = 0; i < diff.length; i += 4) {
+    if (diff[i + 0] == 255 && diff[i + 1] == 0 && diff[i + 2] == 0 && diff[i + 3] == 255) return true;
+  }
+
+  return false;
+}
+
+function compareImages(expect, actual, diffOptions) {
+  const expectImage = _pngjs.PNG.sync.read(expect);
+
+  const actualImage = _pngjs.PNG.sync.read(actual);
+
+  const width = Math.max(actualImage.width, expectImage.width);
+  const height = Math.max(actualImage.height, expectImage.height);
+  const diffImage = new _pngjs.PNG({
+    width,
+    height
+  });
+  let actualImageData = actualImage.data;
+
+  if (actualImage.width < width || actualImage.height < height) {
+    actualImageData = normalizeImageSize(actualImage, width, height);
+  }
+
+  let expectImageData = expectImage.data;
+
+  if (expectImage.width < width || expectImage.height < height) {
+    expectImageData = normalizeImageSize(expectImage, width, height);
+  }
+
+  (0, _pixelmatch.default)(expectImageData, actualImageData, diffImage.data, width, height, diffOptions);
+  return {
+    isEqual: !hasDiffPixels(diffImage.data),
+    diff: _pngjs.PNG.sync.write(diffImage)
+  };
+}
+
+function _default(getExpected, diffOptions) {
+  return function chaiImage({
+    Assertion
+  }, utils) {
+    async function assertImage(actual, imageName) {
+      let onCompare = () => Promise.resolve();
+
+      let expected = await getExpected(imageName);
+      if (!(expected instanceof Buffer) && expected != null) ({
+        expected,
+        onCompare
+      } = expected);
+
+      if (expected == null) {
+        await onCompare(actual);
+        return imageName ? `Expected image '${imageName}' does not exists` : 'Expected image does not exists';
+      }
+
+      if (actual.equals(expected)) return await onCompare(actual);
+      const {
+        isEqual,
+        diff
+      } = compareImages(expected, actual, diffOptions);
+      if (isEqual) return await onCompare(actual);
+      await onCompare(actual, expected, diff);
+      return imageName ? `Expected image '${imageName}' to match` : 'Expected image to match';
+    }
+
+    utils.addMethod(Assertion.prototype, 'matchImage', async function matchImage(imageName) {
+      const actual = utils.flag(this, 'object');
+      const errorMessage = await assertImage(typeof actual == 'string' ? Buffer.from(actual, 'base64') : actual, imageName);
+
+      if (errorMessage) {
+        throw createImageError(imageName ? {
+          [imageName]: errorMessage
+        } : errorMessage);
+      }
+    });
+    utils.addMethod(Assertion.prototype, 'matchImages', async function matchImages() {
+      const errors = {};
+      await Promise.all(Object.entries(utils.flag(this, 'object')).map(async ([imageName, imageOrBase64]) => {
+        let errorMessage;
+
+        try {
+          errorMessage = await assertImage(typeof imageOrBase64 == 'string' ? Buffer.from(imageOrBase64, 'base64') : imageOrBase64, imageName);
+        } catch (error) {
+          errorMessage = error.stack;
+        }
+
+        if (errorMessage) {
+          errors[imageName] = errorMessage;
+        }
+      }));
+
+      if (Object.keys(errors).length > 0) {
+        throw createImageError(errors);
+      }
+    });
+  };
+}
+
+function createImageError(imageErrors) {
+  const error = new Error('Expected image to match');
+  error.images = imageErrors;
+  return error;
+}

package/lib/cjs/server/worker/helpers.js

@@ -0,0 +1,69 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.addTestsFromStories = addTestsFromStories;
+
+var _mocha = require("mocha");
+
+var _types = require("../../types");
+
+var _stories = require("../stories");
+
+function findOrCreateSuite(name, parent) {
+  const suite = parent.suites.find(({
+    title
+  }) => title == name) || new _mocha.Suite(name, parent.ctx);
+
+  if (!suite.parent) {
+    suite.parent = parent;
+    parent.addSuite(suite);
+  }
+
+  return suite;
+}
+
+function createTest(name, fn, skip = false) {
+  const test = new _mocha.Test(name, skip ? undefined : fn);
+  test.pending = Boolean(skip); // NOTE Can't define skip reason in mocha https://github.com/mochajs/mocha/issues/2026
+
+  test.skipReason = skip;
+  return test;
+}
+
+function addTest(rootSuite, test) {
+  const [testName, ...suitePath] = [...test.storyPath, test.testName].reverse().filter(_types.isDefined);
+  const suite = suitePath.reduceRight((subSuite, suiteName) => findOrCreateSuite(suiteName, subSuite), rootSuite);
+  const mochaTest = createTest(testName, test.fn, test.skip);
+  suite.addTest(mochaTest);
+  mochaTest.ctx = Object.setPrototypeOf({
+    id: test.id,
+    story: test.story
+  }, suite.ctx);
+  return mochaTest;
+}
+
+function removeTestOrSuite(testOrSuite) {
+  const {
+    parent
+  } = testOrSuite;
+  if (!parent) return;
+  if (testOrSuite instanceof _mocha.Test) parent.tests = parent.tests.filter(test => test != testOrSuite);
+  if (testOrSuite instanceof _mocha.Suite) parent.suites = parent.suites.filter(suite => suite != testOrSuite);
+  if (parent.tests.length == 0 && parent.suites.length == 0) removeTestOrSuite(parent);
+}
+
+async function addTestsFromStories(rootSuite, config, {
+  browser,
+  ...options
+}) {
+  const mochaTestsById = new Map();
+  const tests = await (0, _stories.loadTestsFromStories)([browser], listener => config.storiesProvider(config, options, listener), testsDiff => Object.entries(testsDiff).forEach(([id, newTest]) => {
+    const oldTest = mochaTestsById.get(id);
+    mochaTestsById.delete(id);
+    if (oldTest) removeTestOrSuite(oldTest);
+    if (newTest) mochaTestsById.set(id, addTest(rootSuite, newTest));
+  }));
+  Object.values(tests).filter(_types.isDefined).forEach(test => mochaTestsById.set(test.id, addTest(rootSuite, test)));
+}