run.ts

// Copyright 2024 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import * as childProcess from 'child_process';
import * as fs from 'fs';
import * as glob from 'glob';
import * as os from 'os';
import * as path from 'path';
import yargs from 'yargs';
import unparse from 'yargs-unparser';

import {commandLineArgs} from './conductor/commandline.js';
import {
  BUILD_WITH_CHROMIUM,
  CHECKOUT_ROOT,
  GEN_DIR,
  isContainedInDirectory,
  PathPair,
  SOURCE_ROOT,
} from './conductor/paths.js';

const options = commandLineArgs(yargs(process.argv.slice(2)))
                    .options('skip-ninja', {type: 'boolean', desc: 'Skip rebuilding'})
                    .options('debug-driver', {type: 'boolean', hidden: true, desc: 'Debug the driver part of tests'})
                    .options('verbose', {alias: 'v', type: 'count', desc: 'Increases the log level'})
                    .options('bail', {type: 'boolean', alias: 'b', desc: ' bail after first test failure'})
                    .options('auto-watch', {
                      desc: 'watch changes to files and run tests automatically on file change (only for unit tests)'
                    })
                    .positional('tests', {
                      type: 'string',
                      desc: 'Path to the test suite, starting from out/Target/gen directory.',
                      normalize: true,
                      default: ['front_end', 'test/e2e', 'test/interactions', 'test/e2e_non_hosted'].map(
                          f => path.relative(process.cwd(), path.join(SOURCE_ROOT, f))),
                    })
                    .strict()
                    .parseSync();

const CONSUMED_OPTIONS = ['tests', 'skip-ninja', 'debug-driver', 'verbose', 'v', 'watch'];

let logLevel = 'error';
if (options['verbose'] === 1) {
  logLevel = 'info';
} else if (options['verbose'] === 2) {
  logLevel = 'debug';
}

function forwardOptions(): string[] {
  const forwardedOptions = {...options};
  for (const consume of CONSUMED_OPTIONS) {
    forwardedOptions[consume] = undefined;
  }

  // @ts-expect-error yargs and unparse have slightly different types
  const unparsed = unparse(forwardedOptions);
  const args: string[] = [];
  for (let i = 0; i < unparsed.length - 1; i++) {
    if (unparsed[i].startsWith('--') && !Number.isNaN(Number(unparsed[i + 1]))) {
      // Mocha errors on --repeat 1 as it expects --repeat=1. We assume
      // that this is the same for all args followed by a number.
      args.push(`${unparsed[i]}=${unparsed[i + 1]}`);
      i++;
    } else {
      args.push(unparsed[i]);
    }
  }
  return args;
}

function runProcess(exe: string, args: string[], options: childProcess.SpawnSyncOptionsWithStringEncoding) {
  if (logLevel !== 'error') {
    // eslint-disable-next-line no-console
    console.info(`Running '${exe}${args.length > 0 ? ` "${args.join('" "')}"` : ''}'`);
  }
  return childProcess.spawnSync(exe, args, options);
}

function ninja(stdio: 'inherit'|'pipe', ...args: string[]) {
  let buildRoot = path.dirname(GEN_DIR);
  while (!fs.existsSync(path.join(buildRoot, 'args.gn'))) {
    const parent = path.dirname(buildRoot);
    if (parent === buildRoot) {
      throw new Error('Failed to find a build directory containing args.gn');
    }
    buildRoot = parent;
  }
  const ninjaCommand = os.platform() === 'win32' ? 'autoninja.bat' : 'autoninja';
  // autoninja can't always find ninja if not run from the checkout root, so
  // run it from there and pass the build root as an argument.
  const result = runProcess(ninjaCommand, ['-C', buildRoot, ...args], {encoding: 'utf-8', cwd: CHECKOUT_ROOT, stdio});
  if (result.error) {
    throw result.error;
  }
  const {status, output: [, output]} = result;
  return {status, output};
}

const MOCHA_BIN_PATH = path.join(SOURCE_ROOT, 'node_modules', 'mocha', 'bin', 'mocha.js');

class Tests {
  readonly suite: PathPair;
  readonly extraPaths: PathPair[];
  protected readonly cwd = path.dirname(GEN_DIR);
  constructor(suite: string, ...extraSuites: string[]) {
    const suitePath = PathPair.get(suite);
    if (!suitePath) {
      throw new Error(`Could not locate the test suite '${suite}'`);
    }
    this.suite = suitePath;
    const extraPaths = extraSuites.map(p => [p, PathPair.get(p)]);
    const failures = extraPaths.filter(p => p[1] === null);
    if (failures.length > 0) {
      throw new Error(`Could not resolve extra paths for ${failures.map(p => p[0]).join()}`);
    }
    this.extraPaths = extraPaths.filter((p): p is[string, PathPair] => p[1] !== null).map(p => p[1]);
  }

  match(path: PathPair) {
    return [this.suite, ...this.extraPaths].some(
        pathToCheck => isContainedInDirectory(path.buildPath, pathToCheck.buildPath));
  }

  protected run(tests: PathPair[], args: string[], positionalTestArgs = true) {
    const argumentsForNode = [
      ...args,
      ...(options['auto-watch'] ? ['--auto-watch', '--no-single-run'] : []),
      '--',
      ...tests.map(t => positionalTestArgs ? t.buildPath : `--tests=${t.buildPath}`),
      ...forwardOptions(),
    ];
    if (options['debug-driver']) {
      argumentsForNode.unshift('--inspect-brk');
    } else if (options['debug'] && !argumentsForNode.includes('--inspect-brk')) {
      argumentsForNode.unshift('--inspect');
    }

    const result = runProcess(process.argv[0], argumentsForNode, {
      encoding: 'utf-8',
      stdio: 'inherit',
      cwd: this.cwd,
    });
    return !result.error && (result.status ?? 1) === 0;
  }
}

class MochaTests extends Tests {
  override run(tests: PathPair[]) {
    return super.run(
        tests,
        [
          MOCHA_BIN_PATH,
          '--config',
          path.join(this.suite.buildPath, 'mocharc.js'),
          '-u',
          path.join(this.suite.buildPath, '..', 'conductor', 'mocha-interface.js'),
        ],
        /* positionalTestArgs= */ false,  // Mocha interprets positional arguments as test files itself. Work around
                                          // that by passing the tests as dashed args instead.
    );
  }
}

class NonHostedMochaTests extends Tests {
  override run(tests: PathPair[]) {
    const args = [
      MOCHA_BIN_PATH,
      '--config',
      path.join(this.suite.buildPath, 'mocharc.js'),
      '-u',
      path.join(this.suite.buildPath, 'conductor', 'mocha-interface.js'),
    ];
    if (options['debug']) {
      args.unshift('--inspect-brk');
    }
    return super.run(
        tests,
        args,
        /* positionalTestArgs= */ false,  // Mocha interprets positional arguments as test files itself. Work around
                                          // that by passing the tests as dashed args instead.
    );
  }
}

/**
 * Workaround the fact that these test don't have
 * build output in out/Default like dir.
 */
class ScriptPathPair extends PathPair {
  static getFromPair(pair: PathPair) {
    return new ScriptPathPair(pair.sourcePath, pair.sourcePath);
  }
}

class ScriptsMochaTests extends Tests {
  override readonly cwd = SOURCE_ROOT;

  override run(tests: PathPair[]) {
    return super.run(
        tests.map(test => ScriptPathPair.getFromPair(test)),
        ['--experimental-strip-types', '--no-warnings=ExperimentalWarning', MOCHA_BIN_PATH, '--extension=ts,js'],
    );
  }

  override match(path: PathPair): boolean {
    return [this.suite, ...this.extraPaths].some(
        pathToCheck => isContainedInDirectory(path.sourcePath, pathToCheck.sourcePath));
  }
}

class KarmaTests extends Tests {
  override run(tests: PathPair[]) {
    return super.run(tests, [
      path.join(SOURCE_ROOT, 'node_modules', 'karma', 'bin', 'karma'),
      'start',
      path.join(GEN_DIR, 'test', 'unit', 'karma.conf.js'),
      '--log-level',
      logLevel,
    ]);
  }
}

// TODO(333423685)
// - watch
function main() {
  const tests: string[] = typeof options['tests'] === 'string' ? [options['tests']] : options['tests'];
  const testKinds = [
    new KarmaTests(path.join(GEN_DIR, 'front_end'), path.join(GEN_DIR, 'inspector_overlay')),
    new MochaTests(path.join(GEN_DIR, 'test/interactions')),
    new MochaTests(path.join(GEN_DIR, 'test/e2e')),
    new NonHostedMochaTests(path.join(GEN_DIR, 'test/e2e_non_hosted')),
    new MochaTests(path.join(GEN_DIR, 'test/perf')),
    new ScriptsMochaTests(path.join(SOURCE_ROOT, 'scripts/eslint_rules/tests')),
    new ScriptsMochaTests(path.join(SOURCE_ROOT, 'scripts/stylelint_rules/tests')),
    new ScriptsMochaTests(path.join(SOURCE_ROOT, 'scripts/build/tests')),
  ];

  if (!options['skip-ninja']) {
    // For a devtools only checkout, it is fast enough to build everything. For
    // a chromium checkout we want to build only the targets that are needed.
    const targets = BUILD_WITH_CHROMIUM ?
        [
          'chrome',
          'third_party/devtools-frontend/src/test:test',
          'third_party/devtools-frontend/src/scripts/hosted_mode:hosted_mode',
          'third_party/devtools-frontend/src/scripts/component_server:component_server',
        ] :
        [];
    const {status} = ninja('inherit', ...targets);
    if (status) {
      return status;
    }
  }

  const suites = new Map<MochaTests, PathPair[]>();
  const testFiles = tests.flatMap(t => {
    const globbed = glob.glob.sync(t);
    return globbed.length > 0 ? globbed : t;
  });
  for (const t of testFiles) {
    const repoPath = PathPair.get(t);
    if (!repoPath) {
      console.error(`Could not locate the test input for '${t}'`);
      continue;
    }

    const suite = testKinds.find(kind => kind.match(repoPath));
    if (suite === undefined) {
      console.error(`Unknown test suite for '${repoPath.sourcePath}'`);
      continue;
    }

    suites.get(suite)?.push(repoPath) ?? suites.set(suite, [repoPath]);
  }

  if (suites.size > 0) {
    const success = Array.from(suites).every(([suite, files]) => suite.run(files));
    return success ? 0 : 1;
  }
  if (tests.length > 0) {
    return 1;
  }
  const success = testKinds.every(kind => kind.run([kind.suite]));
  return success ? 0 : 1;
}

process.exit(main());