| // 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()); |