2024-11-25 16:53:40 -06:00

569 lines
20 KiB
JavaScript

"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.TestServerDispatcher = void 0;
exports.resolveCtDirs = resolveCtDirs;
exports.runTestServer = runTestServer;
exports.runUIMode = runUIMode;
var _fs = _interopRequireDefault(require("fs"));
var _path = _interopRequireDefault(require("path"));
var _server = require("playwright-core/lib/server");
var _utils = require("playwright-core/lib/utils");
var _compilationCache = require("../transform/compilationCache");
var _reporters = require("./reporters");
var _tasks = require("./tasks");
var _utilsBundle = require("playwright-core/lib/utilsBundle");
var _list = _interopRequireDefault(require("../reporters/list"));
var _sigIntWatcher = require("./sigIntWatcher");
var _fsWatcher = require("../fsWatcher");
var _configLoader = require("../common/configLoader");
var _webServerPlugin = require("../plugins/webServerPlugin");
var _util = require("../util");
var _teleReceiver = require("../isomorphic/teleReceiver");
var _internalReporter = require("../reporters/internalReporter");
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
/**
* Copyright Microsoft Corporation. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
const originalStdoutWrite = process.stdout.write;
const originalStderrWrite = process.stderr.write;
class TestServer {
constructor(configLocation, configCLIOverrides) {
this._configLocation = void 0;
this._configCLIOverrides = void 0;
this._dispatcher = void 0;
this._configLocation = configLocation;
this._configCLIOverrides = configCLIOverrides;
}
async start(options) {
this._dispatcher = new TestServerDispatcher(this._configLocation, this._configCLIOverrides);
return await (0, _server.startTraceViewerServer)({
...options,
transport: this._dispatcher.transport
});
}
async stop() {
var _this$_dispatcher, _this$_dispatcher2;
await ((_this$_dispatcher = this._dispatcher) === null || _this$_dispatcher === void 0 ? void 0 : _this$_dispatcher._setInterceptStdio(false));
await ((_this$_dispatcher2 = this._dispatcher) === null || _this$_dispatcher2 === void 0 ? void 0 : _this$_dispatcher2.runGlobalTeardown());
}
}
class TestServerDispatcher {
constructor(configLocation, configCLIOverrides) {
this._configLocation = void 0;
this._configCLIOverrides = void 0;
this._watcher = void 0;
this._watchedProjectDirs = new Set();
this._ignoredProjectOutputs = new Set();
this._watchedTestDependencies = new Set();
this._testRun = void 0;
this.transport = void 0;
this._queue = Promise.resolve();
this._globalSetup = void 0;
this._devServer = void 0;
this._dispatchEvent = void 0;
this._plugins = void 0;
this._serializer = require.resolve('./uiModeReporter');
this._watchTestDirs = false;
this._closeOnDisconnect = false;
this._populateDependenciesOnList = false;
this._configLocation = configLocation;
this._configCLIOverrides = configCLIOverrides;
this.transport = {
onconnect: () => {},
dispatch: (method, params) => this[method](params),
onclose: () => {
if (this._closeOnDisconnect) (0, _utils.gracefullyProcessExitDoNotHang)(0);
}
};
this._watcher = new _fsWatcher.Watcher(events => {
const collector = new Set();
events.forEach(f => (0, _compilationCache.collectAffectedTestFiles)(f.file, collector));
this._dispatchEvent('testFilesChanged', {
testFiles: [...collector]
});
});
this._dispatchEvent = (method, params) => {
var _this$transport$sendE, _this$transport;
return (_this$transport$sendE = (_this$transport = this.transport).sendEvent) === null || _this$transport$sendE === void 0 ? void 0 : _this$transport$sendE.call(_this$transport, method, params);
};
}
async _wireReporter(messageSink) {
return await (0, _reporters.createReporterForTestServer)(this._serializer, messageSink);
}
async _collectingInternalReporter(...extraReporters) {
const report = [];
const collectingReporter = await (0, _reporters.createReporterForTestServer)(this._serializer, e => report.push(e));
return {
reporter: new _internalReporter.InternalReporter([collectingReporter, ...extraReporters]),
report
};
}
async initialize(params) {
// Note: this method can be called multiple times, for example from a new connection after UI mode reload.
this._serializer = params.serializer || require.resolve('./uiModeReporter');
this._closeOnDisconnect = !!params.closeOnDisconnect;
await this._setInterceptStdio(!!params.interceptStdio);
this._watchTestDirs = !!params.watchTestDirs;
this._populateDependenciesOnList = !!params.populateDependenciesOnList;
}
async ping() {}
async open(params) {
if ((0, _utils.isUnderTest)()) return;
// eslint-disable-next-line no-console
(0, _utilsBundle.open)('vscode://file/' + params.location.file + ':' + params.location.line).catch(e => console.error(e));
}
async resizeTerminal(params) {
process.stdout.columns = params.cols;
process.stdout.rows = params.rows;
process.stderr.columns = params.cols;
process.stderr.columns = params.rows;
}
async checkBrowsers() {
return {
hasBrowsers: hasSomeBrowsers()
};
}
async installBrowsers() {
await installBrowsers();
}
async runGlobalSetup(params) {
await this.runGlobalTeardown();
const {
reporter,
report
} = await this._collectingInternalReporter(new _list.default());
const config = await this._loadConfigOrReportError(reporter, this._configCLIOverrides);
if (!config) return {
status: 'failed',
report
};
const {
status,
cleanup
} = await (0, _tasks.runTasksDeferCleanup)(new _tasks.TestRun(config, reporter), [...(0, _tasks.createGlobalSetupTasks)(config)]);
if (status !== 'passed') await cleanup();else this._globalSetup = {
cleanup,
report
};
return {
report,
status
};
}
async runGlobalTeardown() {
const globalSetup = this._globalSetup;
const status = await (globalSetup === null || globalSetup === void 0 ? void 0 : globalSetup.cleanup());
this._globalSetup = undefined;
return {
status,
report: (globalSetup === null || globalSetup === void 0 ? void 0 : globalSetup.report) || []
};
}
async startDevServer(params) {
await this.stopDevServer({});
const {
reporter,
report
} = await this._collectingInternalReporter();
const config = await this._loadConfigOrReportError(reporter);
if (!config) return {
report,
status: 'failed'
};
const {
status,
cleanup
} = await (0, _tasks.runTasksDeferCleanup)(new _tasks.TestRun(config, reporter), [(0, _tasks.createLoadTask)('out-of-process', {
failOnLoadErrors: true,
filterOnly: false
}), (0, _tasks.createStartDevServerTask)()]);
if (status !== 'passed') await cleanup();else this._devServer = {
cleanup,
report
};
return {
report,
status
};
}
async stopDevServer(params) {
const devServer = this._devServer;
const status = await (devServer === null || devServer === void 0 ? void 0 : devServer.cleanup());
this._devServer = undefined;
return {
status,
report: (devServer === null || devServer === void 0 ? void 0 : devServer.report) || []
};
}
async clearCache(params) {
const reporter = new _internalReporter.InternalReporter([]);
const config = await this._loadConfigOrReportError(reporter);
if (!config) return;
await (0, _tasks.runTasks)(new _tasks.TestRun(config, reporter), [(0, _tasks.createClearCacheTask)(config)]);
}
async listFiles(params) {
var _params$projects;
const {
reporter,
report
} = await this._collectingInternalReporter();
const config = await this._loadConfigOrReportError(reporter);
if (!config) return {
status: 'failed',
report
};
config.cliProjectFilter = (_params$projects = params.projects) !== null && _params$projects !== void 0 && _params$projects.length ? params.projects : undefined;
const status = await (0, _tasks.runTasks)(new _tasks.TestRun(config, reporter), [(0, _tasks.createListFilesTask)(), (0, _tasks.createReportBeginTask)()]);
return {
report,
status
};
}
async listTests(params) {
let result;
this._queue = this._queue.then(async () => {
const {
config,
report,
status
} = await this._innerListTests(params);
if (config) await this._updateWatchedDirs(config);
result = {
report,
status
};
}).catch(printInternalError);
await this._queue;
return result;
}
async _innerListTests(params) {
var _params$projects2;
const overrides = {
...this._configCLIOverrides,
repeatEach: 1,
retries: 0
};
const {
reporter,
report
} = await this._collectingInternalReporter();
const config = await this._loadConfigOrReportError(reporter, overrides);
if (!config) return {
report,
reporter,
status: 'failed'
};
config.cliArgs = params.locations || [];
config.cliGrep = params.grep;
config.cliGrepInvert = params.grepInvert;
config.cliProjectFilter = (_params$projects2 = params.projects) !== null && _params$projects2 !== void 0 && _params$projects2.length ? params.projects : undefined;
config.cliListOnly = true;
const status = await (0, _tasks.runTasks)(new _tasks.TestRun(config, reporter), [(0, _tasks.createLoadTask)('out-of-process', {
failOnLoadErrors: false,
filterOnly: false,
populateDependencies: this._populateDependenciesOnList
}), (0, _tasks.createReportBeginTask)()]);
return {
config,
report,
reporter,
status
};
}
async _updateWatchedDirs(config) {
this._watchedProjectDirs = new Set();
this._ignoredProjectOutputs = new Set();
for (const p of config.projects) {
this._watchedProjectDirs.add(p.project.testDir);
this._ignoredProjectOutputs.add(p.project.outputDir);
}
const result = await resolveCtDirs(config);
if (result) {
this._watchedProjectDirs.add(result.templateDir);
this._ignoredProjectOutputs.add(result.outDir);
}
if (this._watchTestDirs) await this._updateWatcher(false);
}
async _updateWatcher(reportPending) {
await this._watcher.update([...this._watchedProjectDirs, ...this._watchedTestDependencies], [...this._ignoredProjectOutputs], reportPending);
}
async runTests(params) {
let result = {
status: 'passed'
};
this._queue = this._queue.then(async () => {
result = await this._innerRunTests(params).catch(e => {
printInternalError(e);
return {
status: 'failed'
};
});
});
await this._queue;
return result;
}
async _innerRunTests(params) {
var _params$projects3;
await this.stopTests();
const overrides = {
...this._configCLIOverrides,
repeatEach: 1,
retries: 0,
preserveOutputDir: true,
reporter: params.reporters ? params.reporters.map(r => [r]) : undefined,
use: {
...this._configCLIOverrides.use,
...(params.trace === 'on' ? {
trace: {
mode: 'on',
sources: false,
_live: true
}
} : {}),
...(params.trace === 'off' ? {
trace: 'off'
} : {}),
...(params.video === 'on' || params.video === 'off' ? {
video: params.video
} : {}),
...(params.headed !== undefined ? {
headless: !params.headed
} : {}),
_optionContextReuseMode: params.reuseContext ? 'when-possible' : undefined,
_optionConnectOptions: params.connectWsEndpoint ? {
wsEndpoint: params.connectWsEndpoint
} : undefined
},
...(params.updateSnapshots ? {
updateSnapshots: params.updateSnapshots
} : {}),
...(params.workers ? {
workers: params.workers
} : {})
};
if (params.trace === 'on') process.env.PW_LIVE_TRACE_STACKS = '1';else process.env.PW_LIVE_TRACE_STACKS = undefined;
const wireReporter = await this._wireReporter(e => this._dispatchEvent('report', e));
const config = await this._loadConfigOrReportError(new _internalReporter.InternalReporter([wireReporter]), overrides);
if (!config) return {
status: 'failed'
};
const testIdSet = params.testIds ? new Set(params.testIds) : null;
config.cliListOnly = false;
config.cliPassWithNoTests = true;
config.cliArgs = params.locations || [];
config.cliGrep = params.grep;
config.cliGrepInvert = params.grepInvert;
config.cliProjectFilter = (_params$projects3 = params.projects) !== null && _params$projects3 !== void 0 && _params$projects3.length ? params.projects : undefined;
config.testIdMatcher = testIdSet ? id => testIdSet.has(id) : undefined;
const configReporters = await (0, _reporters.createReporters)(config, 'test', true);
const reporter = new _internalReporter.InternalReporter([...configReporters, wireReporter]);
const stop = new _utils.ManualPromise();
const tasks = [(0, _tasks.createApplyRebaselinesTask)(), (0, _tasks.createLoadTask)('out-of-process', {
filterOnly: true,
failOnLoadErrors: false,
doNotRunDepsOutsideProjectFilter: true
}), ...(0, _tasks.createRunTestsTasks)(config)];
const run = (0, _tasks.runTasks)(new _tasks.TestRun(config, reporter), tasks, 0, stop).then(async status => {
this._testRun = undefined;
return status;
});
this._testRun = {
run,
stop
};
return {
status: await run
};
}
async watch(params) {
this._watchedTestDependencies = new Set();
for (const fileName of params.fileNames) {
this._watchedTestDependencies.add(fileName);
(0, _compilationCache.dependenciesForTestFile)(fileName).forEach(file => this._watchedTestDependencies.add(file));
}
await this._updateWatcher(true);
}
async findRelatedTestFiles(params) {
const errorReporter = (0, _reporters.createErrorCollectingReporter)();
const reporter = new _internalReporter.InternalReporter([errorReporter]);
const config = await this._loadConfigOrReportError(reporter);
if (!config) return {
errors: errorReporter.errors(),
testFiles: []
};
const status = await (0, _tasks.runTasks)(new _tasks.TestRun(config, reporter), [(0, _tasks.createLoadTask)('out-of-process', {
failOnLoadErrors: true,
filterOnly: false,
populateDependencies: true
})]);
if (status !== 'passed') return {
errors: errorReporter.errors(),
testFiles: []
};
return {
testFiles: (0, _compilationCache.affectedTestFiles)(params.files)
};
}
async stopTests() {
var _this$_testRun, _this$_testRun2;
(_this$_testRun = this._testRun) === null || _this$_testRun === void 0 || (_this$_testRun = _this$_testRun.stop) === null || _this$_testRun === void 0 || _this$_testRun.resolve();
await ((_this$_testRun2 = this._testRun) === null || _this$_testRun2 === void 0 ? void 0 : _this$_testRun2.run);
}
async _setInterceptStdio(intercept) {
if (process.env.PWTEST_DEBUG) return;
if (intercept) {
process.stdout.write = chunk => {
this._dispatchEvent('stdio', chunkToPayload('stdout', chunk));
return true;
};
process.stderr.write = chunk => {
this._dispatchEvent('stdio', chunkToPayload('stderr', chunk));
return true;
};
} else {
process.stdout.write = originalStdoutWrite;
process.stderr.write = originalStderrWrite;
}
}
async closeGracefully() {
(0, _utils.gracefullyProcessExitDoNotHang)(0);
}
async _loadConfig(overrides) {
try {
const config = await (0, _configLoader.loadConfig)(this._configLocation, overrides);
// Preserve plugin instances between setup and build.
if (!this._plugins) {
(0, _webServerPlugin.webServerPluginsForConfig)(config).forEach(p => config.plugins.push({
factory: p
}));
this._plugins = config.plugins || [];
} else {
config.plugins.splice(0, config.plugins.length, ...this._plugins);
}
return {
config
};
} catch (e) {
return {
config: null,
error: (0, _util.serializeError)(e)
};
}
}
async _loadConfigOrReportError(reporter, overrides) {
const {
config,
error
} = await this._loadConfig(overrides);
if (config) return config;
// Produce dummy config when it has an error.
reporter.onConfigure(_teleReceiver.baseFullConfig);
reporter.onError(error);
await reporter.onEnd({
status: 'failed'
});
await reporter.onExit();
return null;
}
}
exports.TestServerDispatcher = TestServerDispatcher;
async function runUIMode(configFile, configCLIOverrides, options) {
const configLocation = (0, _configLoader.resolveConfigLocation)(configFile);
return await innerRunTestServer(configLocation, configCLIOverrides, options, async (server, cancelPromise) => {
await (0, _server.installRootRedirect)(server, [], {
...options,
webApp: 'uiMode.html'
});
if (options.host !== undefined || options.port !== undefined) {
await (0, _server.openTraceInBrowser)(server.urlPrefix('human-readable'));
} else {
const page = await (0, _server.openTraceViewerApp)(server.urlPrefix('precise'), 'chromium', {
headless: (0, _utils.isUnderTest)() && process.env.PWTEST_HEADED_FOR_TEST !== '1',
persistentContextOptions: {
handleSIGINT: false
}
});
page.on('close', () => cancelPromise.resolve());
}
});
}
async function runTestServer(configFile, configCLIOverrides, options) {
const configLocation = (0, _configLoader.resolveConfigLocation)(configFile);
return await innerRunTestServer(configLocation, configCLIOverrides, options, async server => {
// eslint-disable-next-line no-console
console.log('Listening on ' + server.urlPrefix('precise').replace('http:', 'ws:') + '/' + server.wsGuid());
});
}
async function innerRunTestServer(configLocation, configCLIOverrides, options, openUI) {
if ((0, _configLoader.restartWithExperimentalTsEsm)(undefined, true)) return 'restarted';
const testServer = new TestServer(configLocation, configCLIOverrides);
const cancelPromise = new _utils.ManualPromise();
const sigintWatcher = new _sigIntWatcher.SigIntWatcher();
process.stdin.on('close', () => (0, _utils.gracefullyProcessExitDoNotHang)(0));
void sigintWatcher.promise().then(() => cancelPromise.resolve());
try {
const server = await testServer.start(options);
await openUI(server, cancelPromise, configLocation);
await cancelPromise;
} finally {
await testServer.stop();
sigintWatcher.disarm();
}
return sigintWatcher.hadSignal() ? 'interrupted' : 'passed';
}
function chunkToPayload(type, chunk) {
if (chunk instanceof Buffer) return {
type,
buffer: chunk.toString('base64')
};
return {
type,
text: chunk
};
}
function hasSomeBrowsers() {
for (const browserName of ['chromium', 'webkit', 'firefox']) {
try {
_server.registry.findExecutable(browserName).executablePathOrDie('javascript');
return true;
} catch {}
}
return false;
}
async function installBrowsers() {
const executables = _server.registry.defaultExecutables();
await _server.registry.install(executables, false);
}
function printInternalError(e) {
// eslint-disable-next-line no-console
console.error('Internal error:', e);
}
// TODO: remove CT dependency.
async function resolveCtDirs(config) {
const use = config.config.projects[0].use;
const relativeTemplateDir = use.ctTemplateDir || 'playwright';
const templateDir = await _fs.default.promises.realpath(_path.default.normalize(_path.default.join(config.configDir, relativeTemplateDir))).catch(() => undefined);
if (!templateDir) return null;
const outDir = use.ctCacheDir ? _path.default.resolve(config.configDir, use.ctCacheDir) : _path.default.resolve(templateDir, '.cache');
return {
outDir,
templateDir
};
}