| // 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 os from 'os'; |
| import * as path from 'path'; |
| |
| import {commandLineArgs} from './conductor/commandline.js'; |
| import {defaultChromePath, GEN_DIR, isContainedInDirectory, PathPair, SOURCE_ROOT} from './conductor/paths.js'; |
| |
| const yargs = require('yargs'); |
| const unparse = require('yargs-unparser'); |
| 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('warn', {desc: 'Show deprecation warning'}) |
| .options('bail', {alias: 'b', desc: ' bail after first test failure'}) |
| .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'].map( |
| f => path.relative(process.cwd(), path.join(SOURCE_ROOT, f))), |
| }) |
| .strict() |
| .argv; |
| const CONSUMED_OPTIONS = ['tests', 'skip-ninja', 'debug-driver', 'bail', 'b', '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; |
| } |
| 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'; |
| const result = runProcess(ninjaCommand, args, {encoding: 'utf-8', cwd: buildRoot, stdio}); |
| if (result.error) { |
| throw result.error; |
| } |
| const {status, output: [, output]} = result; |
| return {status, output}; |
| } |
| |
| class Tests { |
| readonly suite: PathPair; |
| readonly extraPaths: PathPair[]; |
| 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, |
| '--', |
| ...tests.map(t => positionalTestArgs ? t.buildPath : `--tests=${t.buildPath}`), |
| ...forwardOptions(), |
| ]; |
| if (options['debug-driver']) { |
| argumentsForNode.unshift('--inspect-brk'); |
| } else if (options['debug']) { |
| argumentsForNode.unshift('--inspect'); |
| } |
| const result = runProcess( |
| process.argv[0], argumentsForNode, {encoding: 'utf-8', stdio: 'inherit', cwd: path.dirname(GEN_DIR)}); |
| 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'), |
| ], |
| /* positionalTestArgs= */ false, // Mocha interprets positional arguments as test files itself. Work around |
| // that by passing the tests as dashed args instead. |
| ); |
| } |
| } |
| |
| class KarmaTests extends Tests { |
| override run(tests: PathPair[]) { |
| if (os.type() === 'Windows_NT') { |
| const result = runProcess( |
| 'python3', |
| [ |
| path.join(SOURCE_ROOT, 'scripts', 'deps', 'set_lpac_acls.py'), |
| options['chrome-binary'] ?? defaultChromePath(), |
| ], |
| {encoding: 'utf-8', stdio: 'inherit'}); |
| if (result.error || (result.status ?? 1) !== 0) { |
| return false; |
| } |
| } |
| 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, |
| ]); |
| } |
| } |
| |
| function showDeprecationWarning(command: string) { |
| let alternative = undefined; |
| const debugFlag = command.includes('debug') || process.env['DEBUG_TEST'] ? ' --debug' : ''; |
| switch (command) { |
| case 'auto-e2etest-parallel-rdb': |
| case 'auto-e2etest-rdb': |
| case 'auto-interactionstest-rdb': |
| case 'auto-screenshotstest-rdb': |
| case 'auto-unittest-rdb': |
| alternative = `npm run rdb -- npm run ${command.substring(0, command.length - '-rdb'.length)}`; |
| break; |
| case 'auto-e2etest': |
| alternative = 'npm run test -- test/e2e'; |
| break; |
| case 'auto-screenshotstest': |
| alternative = 'npm run test -- test/interactions --fgrep "[screenshot]"'; |
| break; |
| case 'auto-unittest': |
| alternative = `npm run test -- front_end${debugFlag}`; |
| break; |
| case 'auto-unittest-coverage': |
| alternative = 'npm run test -- front_end --coverage'; |
| break; |
| case 'e2etest': |
| alternative = `npm run test -- test/e2e${debugFlag}`; |
| break; |
| case 'interactionstest': |
| alternative = `npm run test -- test/interactions${debugFlag}`; |
| break; |
| case 'perf': |
| alternative = 'npm run test -- test/perf'; |
| break; |
| case 'unittest': |
| alternative = `npm run test -- front_end${debugFlag}`; |
| break; |
| case 'e2etest-parallel': |
| case 'watch-unittest': |
| break; |
| default: |
| throw new Error(`Deprecation warning for '${options['warn']}' has no deprecation details`); |
| } |
| const format = process.stderr.hasColors() ? '\x1b[1;31m%s\x1b[0m' : '%s'; |
| console.error(format, `WARNING: The npm command '${command}' is deprecated and will be removed in the near future.`); |
| if (alternative) { |
| console.error(format, `Use \`${alternative}\` instead.`); |
| } |
| return 0; |
| } |
| |
| // TODO(333423685) |
| // - iterations |
| // - expanded-reporting |
| // - watch |
| // - layout? |
| function main() { |
| if (options['warn']) { |
| return showDeprecationWarning(options['warn']); |
| } |
| const tests: string[] = options['tests']; |
| |
| const testKinds = [ |
| new MochaTests(path.join(GEN_DIR, 'test/interactions')), |
| new MochaTests(path.join(GEN_DIR, 'test/e2e')), |
| new KarmaTests(path.join(GEN_DIR, 'front_end'), path.join(GEN_DIR, 'inspector_overlay')), |
| new MochaTests(path.join(GEN_DIR, 'test/perf')), |
| ]; |
| |
| if (!options['skip-ninja']) { |
| const {status} = ninja('inherit'); |
| if (status) { |
| return status; |
| } |
| } |
| |
| const suites = new Map<MochaTests, PathPair[]>(); |
| for (const t of tests) { |
| 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()); |