135 lines
4.0 KiB
TypeScript
135 lines
4.0 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 'playwright-core/src/server/utils/socksProxy';
|
|
|
|
import { SocksProxy } from '../../packages/playwright-core/lib/server/utils/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);
|
|
|
|
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();
|
|
}
|
|
}
|