fix webdriver tests

Author: mxschmitt

extension/background.js

@@ -53,7 +53,7 @@ class Extension {
     const debuggee = { tabId }
     await chrome.debugger.attach(debuggee, '1.3')
     if (chrome.runtime.lastError) {
-      console.error('Failed to attach debugger:', chrome.runtime.lastError.message);
+      debugLog('Failed to attach debugger:', chrome.runtime.lastError.message);
       return;
     }
     debugLog('Debugger attached to tab:', debuggee.tabId);
@@ -65,7 +65,7 @@ class Extension {
       if (socket.readyState === WebSocket.OPEN) {
         socket.send(JSON.stringify(data));
       } else {
-        console.error('WebSocket is not open. Cannot send data.');
+        debugLog('WebSocket is not open. Cannot send data.');
       }
     }
     adapter.onClose = async () => {
@@ -79,7 +79,7 @@ class Extension {
     });
     socket.addEventListener('message', (e) => adapter.onBrowserMessage(targetId, browserContextId, e));
     socket.addEventListener('error', (event) => {
-      console.error('WebSocket error:', event);
+      debugLog('WebSocket error:', event);
       adapter.detach();
     });
     socket.addEventListener('close', async () => {
@@ -108,7 +108,6 @@ class CDPAdapter {
    * @returns 
    */
   async onBrowserMessage(targetId, browserContextId, event) {
-    console.error('Received message from browser:', event);
     try {
       const message = JSON.parse(await event.data.text());
       if (message.method === 'Browser.getVersion') {
@@ -159,7 +158,7 @@ class CDPAdapter {
         });
       }
     } catch (e) {
-      console.error('Error processing WebSocket message:', e);
+      debugLog('Error processing WebSocket message:', e);
     }
   }
 

tests/fixtures.ts

@@ -169,15 +169,18 @@ export const test = baseTest.extend<TestFixtures & TestOptions, WorkerFixtures>(
   mcpExtensionPage: async ({ mcpExtension, mcpHeadless }, use) => {
     if (!mcpExtension)
       return await use(undefined);
+    const webSocketPort = 8900 + test.info().parallelIndex * 4;
     const pathToExtension = path.join(url.fileURLToPath(import.meta.url), '../../extension');
     const context = await chromium.launchPersistentContext('', {
       headless: mcpHeadless,
       args: [
         `--disable-extensions-except=${pathToExtension}`,
         `--load-extension=${pathToExtension}`,
-        '--enable-features=AllowContentInitiatedDataUrlNavigations'
+        '--enable-features=AllowContentInitiatedDataUrlNavigations',
+        `--remote-debugging-port=1234`,
       ],
       channel: 'chromium',
+      ...{ assistantMode: true, webSocketPort },
     });
     const page = context.pages()[0];
     // Do not auto dismiss dialogs.

tests/webdriver.spec.ts

@@ -16,8 +16,7 @@
 
 import { test, expect } from './fixtures.js';
 
-test('do not falsely advertise user agent as a test driver', async ({ client, server, mcpBrowser, mcpExtension }) => {
-  test.fixme(!!mcpExtension, 'To be investigated why this is failing');
+test('do not falsely advertise user agent as a test driver', async ({ client, server, mcpBrowser }) => {
   test.skip(mcpBrowser === 'firefox');
   test.skip(mcpBrowser === 'webkit');
   server.route('/', (req, res) => {