mirror of
https://github.com/microsoft/playwright.git
synced 2024-12-14 13:45:36 +03:00
121 lines
3.6 KiB
TypeScript
121 lines
3.6 KiB
TypeScript
/**
|
|
* 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.
|
|
*/
|
|
|
|
import path from 'path';
|
|
import type { BrowserType, Browser, LaunchOptions } from '../../index';
|
|
import { CommonFixtures, TestChildProcess } from './commonFixtures';
|
|
|
|
const playwrightPath = path.join(__dirname, '..', '..');
|
|
|
|
export type RemoteServerOptions = {
|
|
stallOnClose?: boolean;
|
|
disconnectOnSIGHUP?: boolean;
|
|
inCluster?: boolean;
|
|
url?: string;
|
|
};
|
|
|
|
export class RemoteServer {
|
|
private _process: TestChildProcess;
|
|
_output: Map<string, string>;
|
|
_outputCallback: Map<string, () => void>;
|
|
_browserType: BrowserType;
|
|
_exitAndDisconnectPromise: Promise<any>;
|
|
_browser: Browser;
|
|
_wsEndpoint: string;
|
|
|
|
async _start(childProcess: CommonFixtures['childProcess'], browserType: BrowserType, browserOptions: LaunchOptions, remoteServerOptions: RemoteServerOptions = {}) {
|
|
this._output = new Map();
|
|
this._outputCallback = new Map();
|
|
|
|
this._browserType = browserType;
|
|
// Copy options to prevent a large JSON string when launching subprocess.
|
|
// Otherwise, we get `Error: spawn ENAMETOOLONG` on Windows.
|
|
const launchOptions: LaunchOptions = {
|
|
args: browserOptions.args,
|
|
headless: browserOptions.headless,
|
|
channel: browserOptions.channel,
|
|
handleSIGINT: true,
|
|
handleSIGTERM: true,
|
|
handleSIGHUP: true,
|
|
executablePath: browserOptions.channel ? undefined : browserOptions.executablePath || browserType.executablePath(),
|
|
logger: undefined,
|
|
};
|
|
const options = {
|
|
playwrightPath,
|
|
browserTypeName: browserType.name(),
|
|
launchOptions,
|
|
...remoteServerOptions,
|
|
};
|
|
this._process = childProcess({
|
|
command: ['node', path.join(__dirname, 'remote-server-impl.js'), JSON.stringify(options)],
|
|
});
|
|
|
|
let index = 0;
|
|
this._process.onOutput = () => {
|
|
let match;
|
|
while ((match = this._process.output.substring(index).match(/\(([^()]+)=>([^()]+)\)/))) {
|
|
const key = match[1];
|
|
const value = match[2];
|
|
this._addOutput(key, value);
|
|
index += match.index + match[0].length;
|
|
}
|
|
};
|
|
|
|
this._wsEndpoint = await this.out('wsEndpoint');
|
|
|
|
if (remoteServerOptions.url) {
|
|
this._browser = await this._browserType.connect({ wsEndpoint: this._wsEndpoint });
|
|
const page = await this._browser.newPage();
|
|
await page.goto(remoteServerOptions.url);
|
|
}
|
|
}
|
|
|
|
_addOutput(key: string, value: string) {
|
|
this._output.set(key, value);
|
|
const cb = this._outputCallback.get(key);
|
|
this._outputCallback.delete(key);
|
|
if (cb)
|
|
cb();
|
|
}
|
|
|
|
async out(key: string) {
|
|
if (!this._output.has(key))
|
|
await new Promise<void>(f => this._outputCallback.set(key, f));
|
|
return this._output.get(key);
|
|
}
|
|
|
|
wsEndpoint() {
|
|
return this._wsEndpoint;
|
|
}
|
|
|
|
child() {
|
|
return this._process.process;
|
|
}
|
|
|
|
async childExitCode() {
|
|
return await this._process.exitCode;
|
|
}
|
|
|
|
async close() {
|
|
if (this._browser) {
|
|
await this._browser.close();
|
|
this._browser = undefined;
|
|
}
|
|
await this._process.close();
|
|
return await this.childExitCode();
|
|
}
|
|
}
|