blob: 73e65b6f1bcbf813770cb6f221cf99f7f7a9d8d5 [file] [log] [blame]
// 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() {
const forwardedOptions = {...options};
for (const consume of CONSUMED_OPTIONS) {
forwardedOptions[consume] = undefined;
}
// @ts-expect-error yargs and unparse have slightly different types
return unparse(forwardedOptions);
}
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};
}
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,
[
path.join(SOURCE_ROOT, 'node_modules', 'mocha', 'bin', 'mocha'),
'--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 = [
path.join(SOURCE_ROOT, 'node_modules', 'mocha', 'bin', 'mocha'),
'--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',
path.join(SOURCE_ROOT, 'node_modules', 'mocha', 'bin', 'mocha'), '--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());