/** * Copyright (c) Microsoft Corporation. * * 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 childProcess from 'child_process'; import http from 'http'; import path from 'path'; import net from 'net'; import { contextTest, expect } from './config/browserTest'; import { PlaywrightClient } from '../lib/remote/playwrightClient'; import type { Page } from '..'; class OutOfProcessPlaywrightServer { private _driverProcess: childProcess.ChildProcess; private _receivedPortPromise: Promise; constructor(port: number, proxyPort: number) { this._driverProcess = childProcess.fork(path.join(__dirname, '..', 'lib', 'cli', 'cli.js'), ['run-server', port.toString()], { stdio: 'pipe', detached: true, env: { ...process.env, PW_SOCKS_PROXY_PORT: '1' } }); this._driverProcess.unref(); this._receivedPortPromise = new Promise((resolve, reject) => { this._driverProcess.stdout.on('data', (data: Buffer) => { const prefix = 'Listening on '; const line = data.toString(); if (line.startsWith(prefix)) resolve(line.substr(prefix.length)); }); this._driverProcess.stderr.on('data', (data: Buffer) => { console.log(data.toString()); }); this._driverProcess.on('exit', () => reject()); }); } async kill() { const waitForExit = new Promise(resolve => this._driverProcess.on('exit', () => resolve())); this._driverProcess.kill('SIGKILL'); await waitForExit; } public async wsEndpoint(): Promise { return await this._receivedPortPromise; } } const it = contextTest.extend<{ pageFactory: (redirectPortForTest?: number) => Promise }>({ pageFactory: async ({ browserName, browserOptions }, run, testInfo) => { const playwrightServers: OutOfProcessPlaywrightServer[] = []; await run(async (redirectPortForTest?: number): Promise => { const server = new OutOfProcessPlaywrightServer(0, 3200 + testInfo.workerIndex); playwrightServers.push(server); const service = await PlaywrightClient.connect({ wsEndpoint: await server.wsEndpoint(), }); const playwright = service.playwright(); playwright._enablePortForwarding(redirectPortForTest); const browser = await playwright[browserName].launch(browserOptions); return await browser.newPage(); }); for (const playwrightServer of playwrightServers) await playwrightServer.kill(); }, }); it.fixme(({ platform, browserName }) => browserName === 'webkit' && platform === 'win32'); it.skip(({ mode }) => mode !== 'default'); async function startTestServer() { const server = http.createServer((req: http.IncomingMessage, res: http.ServerResponse) => { res.end('from-retargeted-server'); }); await new Promise(resolve => server.listen(0, resolve)); return { testServerPort: (server.address() as net.AddressInfo).port, stopTestServer: () => server.close() }; } it('should forward non-forwarded requests', async ({ pageFactory, server }) => { let reachedOriginalTarget = false; server.setRoute('/foo.html', async (req, res) => { reachedOriginalTarget = true; res.end('original-target'); }); const page = await pageFactory(); await page.goto(server.PREFIX + '/foo.html'); expect(await page.content()).toContain('original-target'); expect(reachedOriginalTarget).toBe(true); }); it('should proxy localhost requests', async ({ pageFactory, server, browserName, platform }, workerInfo) => { it.skip(browserName === 'webkit' && platform === 'darwin'); const { testServerPort, stopTestServer } = await startTestServer(); let reachedOriginalTarget = false; server.setRoute('/foo.html', async (req, res) => { reachedOriginalTarget = true; res.end(''); }); const examplePort = 20_000 + workerInfo.workerIndex * 3; const page = await pageFactory(testServerPort); await page.goto(`http://localhost:${examplePort}/foo.html`); expect(await page.content()).toContain('from-retargeted-server'); expect(reachedOriginalTarget).toBe(false); stopTestServer(); }); it('should proxy local.playwright requests', async ({ pageFactory, server, browserName }, workerInfo) => { it.fixme(browserName === 'firefox', 'Firefox performs DNS on browser side'); const { testServerPort, stopTestServer } = await startTestServer(); let reachedOriginalTarget = false; server.setRoute('/foo.html', async (req, res) => { reachedOriginalTarget = true; res.end(''); }); const examplePort = 20_000 + workerInfo.workerIndex * 3; const page = await pageFactory(testServerPort); await page.goto(`http://local.playwright:${examplePort}/foo.html`); expect(await page.content()).toContain('from-retargeted-server'); expect(reachedOriginalTarget).toBe(false); stopTestServer(); }); it('should lead to the error page for forwarded requests when the connection is refused', async ({ pageFactory, browserName }, workerInfo) => { const examplePort = 20_000 + workerInfo.workerIndex * 3; const page = await pageFactory(); const error = await page.goto(`http://localhost:${examplePort}`).catch(e => e); if (browserName === 'chromium') expect(error.message).toContain('net::ERR_SOCKS_CONNECTION_FAILED at http://localhost:20'); else if (browserName === 'webkit') expect(error.message).toBeTruthy(); else if (browserName === 'firefox') expect(error.message).toContain('NS_ERROR_CONNECTION_REFUSED'); });