042161e1ce
This reapplies what we reverted in https://github.com/microsoft/playwright/pull/32989. Max and me debugged this, and found that the test failures come from SOCKS proxy now preferring IPv6 over IPv4. We've updated the tests and made sure that this doesn't mask any breaking change. I'm enabling CQ1 to make sure we don't oversee any other CI failures.
135 lines
4 KiB
TypeScript
135 lines
4 KiB
TypeScript
/**
|
|
* 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 type { Fixtures } from '@playwright/test';
|
|
import path from 'path';
|
|
import { TestServer } from './testserver';
|
|
import { TestProxy } from './proxy';
|
|
import type { SocksSocketRequestedPayload } from '../../packages/playwright-core/src/common/socksProxy';
|
|
|
|
import { SocksProxy } from '../../packages/playwright-core/lib/common/socksProxy';
|
|
|
|
export type ServerWorkerOptions = {
|
|
loopback?: string;
|
|
__servers: ServerFixtures;
|
|
};
|
|
|
|
export type ServerFixtures = {
|
|
server: TestServer;
|
|
httpsServer: TestServer;
|
|
socksPort: number;
|
|
proxyServer: TestProxy;
|
|
asset: (p: string) => string;
|
|
};
|
|
|
|
export const serverFixtures: Fixtures<ServerFixtures, ServerWorkerOptions> = {
|
|
loopback: [undefined, { scope: 'worker', option: true }],
|
|
__servers: [async ({ loopback }, run, workerInfo) => {
|
|
const assetsPath = path.join(__dirname, '..', 'assets');
|
|
const cachedPath = path.join(__dirname, '..', 'assets', 'cached');
|
|
|
|
const port = 8907 + workerInfo.workerIndex * 4;
|
|
const server = await TestServer.create(assetsPath, port, loopback);
|
|
server.enableHTTPCache(cachedPath);
|
|
|
|
const httpsPort = port + 1;
|
|
const httpsServer = await TestServer.createHTTPS(assetsPath, httpsPort, loopback);
|
|
httpsServer.enableHTTPCache(cachedPath);
|
|
|
|
const socksServer = new MockSocksServer();
|
|
const socksPort = port + 2;
|
|
await socksServer.listen(socksPort, loopback);
|
|
|
|
const proxyPort = port + 3;
|
|
const proxyServer = await TestProxy.create(proxyPort);
|
|
|
|
await run({
|
|
asset: (p: string) => path.join(__dirname, '..', 'assets', ...p.split('/')),
|
|
server,
|
|
httpsServer,
|
|
socksPort,
|
|
proxyServer,
|
|
});
|
|
|
|
await Promise.all([
|
|
server.stop(),
|
|
httpsServer.stop(),
|
|
socksServer.close(),
|
|
proxyServer.stop(),
|
|
]);
|
|
}, { scope: 'worker' }],
|
|
|
|
server: async ({ __servers }, run) => {
|
|
__servers.server.reset();
|
|
await run(__servers.server);
|
|
},
|
|
|
|
httpsServer: async ({ __servers }, run) => {
|
|
__servers.httpsServer.reset();
|
|
await run(__servers.httpsServer);
|
|
},
|
|
|
|
socksPort: async ({ __servers }, run) => {
|
|
await run(__servers.socksPort);
|
|
},
|
|
|
|
proxyServer: async ({ __servers }, run) => {
|
|
__servers.proxyServer.reset();
|
|
await run(__servers.proxyServer);
|
|
},
|
|
|
|
asset: async ({ __servers }, run) => {
|
|
await run(__servers.asset);
|
|
},
|
|
};
|
|
|
|
export class MockSocksServer {
|
|
private _socksProxy: SocksProxy;
|
|
|
|
constructor() {
|
|
this._socksProxy = new SocksProxy();
|
|
this._socksProxy.setPattern('*');
|
|
this._socksProxy.addListener(SocksProxy.Events.SocksRequested, async (payload: SocksSocketRequestedPayload) => {
|
|
this._socksProxy.socketConnected({
|
|
uid: payload.uid,
|
|
host: '127.0.0.1',
|
|
port: 0,
|
|
});
|
|
});
|
|
this._socksProxy.addListener(SocksProxy.Events.SocksData, async (payload: SocksSocketRequestedPayload) => {
|
|
const body = '<html><title>Served by the SOCKS proxy</title></html>';
|
|
const data = Buffer.from([
|
|
'HTTP/1.1 200 OK',
|
|
'Connection: close',
|
|
'Content-Type: text/html',
|
|
'Content-Length: ' + Buffer.byteLength(body),
|
|
'',
|
|
body
|
|
].join('\r\n'));
|
|
this._socksProxy.sendSocketData({ uid: payload.uid, data });
|
|
this._socksProxy.sendSocketEnd({ uid: payload.uid });
|
|
});
|
|
}
|
|
|
|
async listen(port: number, hostname: string) {
|
|
await this._socksProxy.listen(port, hostname);
|
|
}
|
|
|
|
async close() {
|
|
await this._socksProxy.close();
|
|
}
|
|
}
|