include connect options

microsoft · Sep 1, 2022 · 0a3e7ca · 0a3e7ca
1 parent e5ce4c6
commit 0a3e7ca
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/packages/playwright-core/src/cli/docker.ts b/packages/playwright-core/src/cli/docker.ts
@@ -228,7 +228,8 @@ export async function startVRTService() {
 
   /* eslint-disable no-console */
   console.log(`✨ Launched Playwright ${getPlaywrightVersion()} Docker Container ✨`);
-  console.log(`- VNC session: http://localhost:7900?path=${utils.createGuid()}&resize=scale`);
+  console.log(``);
+  console.log(`- VNC session: http://127.0.0.1:7900?path=${utils.createGuid()}&resize=scale`);
   console.log(`- WS endpoint: ${endpoint}`);
   console.log(``);
   console.log(`You can now run tests with browsers from this container:`);

diff --git a/packages/playwright-test/src/cli.ts b/packages/playwright-test/src/cli.ts
@@ -132,6 +132,9 @@ async function runTests(args: string[], opts: { [key: string]: any }) {
     if (!wsEndpoint)
       throw new Error(`Playwright docker container is not running; run "npx playwright start-docker" first`);
     process.env.PW_TEST_CONNECT_WS_ENDPOINT = await docker.wsEndpoint();
+    process.env.PW_TEST_CONNECT_HEADERS = JSON.stringify({
+      'x-playwright-proxy': '*',
+    });
     process.env.PW_TEST_IS_DOCKER = '1';
   }
 

diff --git a/packages/playwright-test/src/index.ts b/packages/playwright-test/src/index.ts
@@ -71,7 +71,10 @@ export const test = _baseTest.extend<TestFixtures, WorkerFixtures>({
   headless: [({ launchOptions }, use) => use(launchOptions.headless ?? true), { scope: 'worker', option: true }],
   channel: [({ launchOptions }, use) => use(launchOptions.channel), { scope: 'worker', option: true }],
   launchOptions: [{}, { scope: 'worker', option: true }],
-  connectOptions: [process.env.PW_TEST_CONNECT_WS_ENDPOINT ? { wsEndpoint: process.env.PW_TEST_CONNECT_WS_ENDPOINT } : undefined, { scope: 'worker', option: true }],
+  connectOptions: [process.env.PW_TEST_CONNECT_WS_ENDPOINT ? {
+    wsEndpoint: process.env.PW_TEST_CONNECT_WS_ENDPOINT,
+    headers: process.env.PW_TEST_CONNECT_HEADERS ? JSON.parse(process.env.PW_TEST_CONNECT_HEADERS) : undefined,
+  } : undefined, { scope: 'worker', option: true }],
   screenshot: ['off', { scope: 'worker', option: true }],
   video: ['off', { scope: 'worker', option: true }],
   trace: ['off', { scope: 'worker', option: true }],