@loki/runner/src/commands/approve/index.js

Loki test runner

/* eslint-disable global-require, import/no-dynamic-require */

const fs = require('fs-extra');
const path = require('path');
const { die } = require('../../console');
const parseOptions = require('./parse-options');
const getConfig = require('../../config');

const isPNG = (file) => file.substr(-4) === '.png';

function approve(args) {
  const config = getConfig();
  const { outputDir, differenceDir, referenceDir, diffOnly } = parseOptions(
    args,
    config
  );

  // If diff only is active, only copy over the files that were changed
  const inputDir = diffOnly ? differenceDir : outputDir;
  const files = fs.readdirSync(inputDir).filter(isPNG);

  if (!files.length) {
    die(
      'No images found to approve',
      'Run update command to generate reference files instead'
    );
  }

  if (diffOnly) {
    /**
     * If diff only is active, the reference directory should not be emptied.
     * Instead only the files that changed will be copied over, overwriting the existing ones.
     * The files are copied and not moved, so running loki approve without --diffOnly after running it with --diffOnly
     * would not delete them as they would no longer be present in the current images.
     */
    files.forEach((file) =>
      fs.copySync(path.join(outputDir, file), path.join(referenceDir, file), {
        overwrite: true,
      })
    );
    return;
  }

  fs.emptyDirSync(referenceDir);
  fs.ensureDirSync(referenceDir);

  files.forEach((file) =>
    fs.moveSync(path.join(outputDir, file), path.join(referenceDir, file))
  );
}

module.exports = approve;