patchright-bun 1.58.2 → 1.59.0

package/lib/matchers/matchers.js

@@ -278,6 +278,8 @@ function toHaveTitle(page, expected, options = {}) {
   }, expected, options);
 }
 function toHaveURL(page, expected, options) {
+  if ((0, import_utils.isURLPattern)(expected))
+    return import_toHaveURL.toHaveURLWithPredicate.call(this, page, (url) => expected.test(url.href), options);
   if (typeof expected === "function")
     return import_toHaveURL.toHaveURLWithPredicate.call(this, page, expected, options);
   const baseURL = page.context()._options.baseURL;

package/lib/matchers/toMatchAriaSnapshot.js

@@ -41,7 +41,7 @@ async function toMatchAriaSnapshot(locator, expectedParam, options = {}) {
   const testInfo = (0, import_globals.currentTestInfo)();
   if (!testInfo)
     throw new Error(`toMatchAriaSnapshot() must be called during the test`);
-  if (testInfo._projectInternal.ignoreSnapshots)
+  if (testInfo._projectInternal.project.ignoreSnapshots)
     return { pass: !this.isNot, message: () => "", name: "toMatchAriaSnapshot", expected: "" };
   const updateSnapshots = testInfo.config.updateSnapshots;
   let expected;
@@ -68,6 +68,10 @@ async function toMatchAriaSnapshot(locator, expectedParam, options = {}) {
     }
   }
   expected = unshift(expected);
+  const globalChildren = testInfo._projectInternal.expect?.toMatchAriaSnapshot?.children;
+  if (globalChildren && !expected.match(/^- \/children:/m))
+    expected = `- /children: ${globalChildren}
+` + expected;
   const { matches: pass, received, log, timedOut, errorMessage } = await locator._expect("to.match.aria", { expectedValue: expected, isNot: this.isNot, timeout });
   const typedReceived = received;
   const message = () => {

package/lib/matchers/toMatchSnapshot.js

@@ -177,7 +177,7 @@ function toMatchSnapshot(received, nameOrOptions = {}, optOptions = {}) {
     throw new Error(`toMatchSnapshot() must be called during the test`);
   if (received instanceof Promise)
     throw new Error("An unresolved Promise was passed to toMatchSnapshot(), make sure to resolve it by adding await to it.");
-  if (testInfo._projectInternal.ignoreSnapshots)
+  if (testInfo._projectInternal.project.ignoreSnapshots)
     return { pass: !this.isNot, message: () => "", name: "toMatchSnapshot", expected: nameOrOptions };
   const configOptions = testInfo._projectInternal.expect?.toMatchSnapshot || {};
   const helper = new SnapshotHelper(
@@ -232,7 +232,7 @@ async function toHaveScreenshot(pageOrLocator, nameOrOptions = {}, optOptions =
   const testInfo = (0, import_globals.currentTestInfo)();
   if (!testInfo)
     throw new Error(`toHaveScreenshot() must be called during the test`);
-  if (testInfo._projectInternal.ignoreSnapshots)
+  if (testInfo._projectInternal.project.ignoreSnapshots)
     return { pass: !this.isNot, message: () => "", name: "toHaveScreenshot", expected: nameOrOptions };
   (0, import_util.expectTypes)(pageOrLocator, ["Page", "Locator"], "toHaveScreenshot");
   const [page, locator] = pageOrLocator.constructor.name === "Page" ? [pageOrLocator, void 0] : [pageOrLocator.page(), pageOrLocator];
@@ -265,45 +265,52 @@ async function toHaveScreenshot(pageOrLocator, nameOrOptions = {}, optOptions =
   if (this.isNot) {
     if (!hasSnapshot)
       return helper.handleMissingNegated();
-    expectScreenshotOptions.expected = await import_fs.default.promises.readFile(helper.expectedPath);
-    const isDifferent = !(await page._expectScreenshot(expectScreenshotOptions)).errorMessage;
-    return isDifferent ? helper.handleDifferentNegated() : helper.handleMatchingNegated();
   }
-  if (helper.updateSnapshots === "none" && !hasSnapshot)
+  if (!this.isNot && helper.updateSnapshots === "none" && !hasSnapshot)
     return helper.createMatcherResult(`A snapshot doesn't exist at ${helper.expectedPath}.`, false);
-  if (!hasSnapshot) {
-    const { actual: actual2, previous: previous2, diff: diff2, errorMessage: errorMessage2, log: log2, timedOut: timedOut2 } = await page._expectScreenshot(expectScreenshotOptions);
-    if (errorMessage2) {
-      const header2 = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut: timedOut2 });
-      return helper.handleDifferent(actual2, void 0, previous2, diff2, header2, errorMessage2, log2, this._stepInfo);
+  await page.screencast.hideOverlays();
+  try {
+    if (this.isNot) {
+      expectScreenshotOptions.expected = await import_fs.default.promises.readFile(helper.expectedPath);
+      const isDifferent = !(await page._expectScreenshot(expectScreenshotOptions)).errorMessage;
+      return isDifferent ? helper.handleDifferentNegated() : helper.handleMatchingNegated();
     }
-    return helper.handleMissing(actual2, this._stepInfo);
-  }
-  const expected = await import_fs.default.promises.readFile(helper.expectedPath);
-  expectScreenshotOptions.expected = helper.updateSnapshots === "all" ? void 0 : expected;
-  const { actual, previous, diff, errorMessage, log, timedOut } = await page._expectScreenshot(expectScreenshotOptions);
-  const writeFiles = (actualBuffer) => {
-    writeFileSync(helper.expectedPath, actualBuffer);
-    writeFileSync(helper.actualPath, actualBuffer);
-    console.log(helper.expectedPath + " is re-generated, writing actual.");
-    return helper.createMatcherResult(helper.expectedPath + " running with --update-snapshots, writing actual.", true);
-  };
-  if (!errorMessage) {
-    if (helper.updateSnapshots === "all" && actual && (0, import_utils.compareBuffersOrStrings)(actual, expected)) {
+    if (!hasSnapshot) {
+      const { actual: actual2, previous: previous2, diff: diff2, errorMessage: errorMessage2, log: log2, timedOut: timedOut2 } = await page._expectScreenshot(expectScreenshotOptions);
+      if (errorMessage2) {
+        const header2 = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut: timedOut2 });
+        return helper.handleDifferent(actual2, void 0, previous2, diff2, header2, errorMessage2, log2, this._stepInfo);
+      }
+      return helper.handleMissing(actual2, this._stepInfo);
+    }
+    const expected = await import_fs.default.promises.readFile(helper.expectedPath);
+    expectScreenshotOptions.expected = helper.updateSnapshots === "all" ? void 0 : expected;
+    const { actual, previous, diff, errorMessage, log, timedOut } = await page._expectScreenshot(expectScreenshotOptions);
+    const writeFiles = (actualBuffer) => {
+      writeFileSync(helper.expectedPath, actualBuffer);
+      writeFileSync(helper.actualPath, actualBuffer);
       console.log(helper.expectedPath + " is re-generated, writing actual.");
-      return writeFiles(actual);
+      return helper.createMatcherResult(helper.expectedPath + " running with --update-snapshots, writing actual.", true);
+    };
+    if (!errorMessage) {
+      if (helper.updateSnapshots === "all" && actual && (0, import_utils.compareBuffersOrStrings)(actual, expected)) {
+        console.log(helper.expectedPath + " is re-generated, writing actual.");
+        return writeFiles(actual);
+      }
+      return helper.handleMatching();
     }
-    return helper.handleMatching();
-  }
-  if (helper.updateSnapshots === "changed" || helper.updateSnapshots === "all") {
-    if (actual)
-      return writeFiles(actual);
-    let header2 = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut });
-    header2 += "  Failed to re-generate expected.\n";
-    return helper.handleDifferent(actual, expectScreenshotOptions.expected, previous, diff, header2, errorMessage, log, this._stepInfo);
+    if (helper.updateSnapshots === "changed" || helper.updateSnapshots === "all") {
+      if (actual)
+        return writeFiles(actual);
+      let header2 = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut });
+      header2 += "  Failed to re-generate expected.\n";
+      return helper.handleDifferent(actual, expectScreenshotOptions.expected, previous, diff, header2, errorMessage, log, this._stepInfo);
+    }
+    const header = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut });
+    return helper.handleDifferent(actual, expectScreenshotOptions.expected, previous, diff, header, errorMessage, log, this._stepInfo);
+  } finally {
+    await page.screencast.showOverlays();
   }
-  const header = (0, import_utils.formatMatcherMessage)(this.utils, { promise: this.promise, isNot: this.isNot, matcherName: "toHaveScreenshot", locator: locator?.toString(), expectation: "expected", timeout, timedOut });
-  return helper.handleDifferent(actual, expectScreenshotOptions.expected, previous, diff, header, errorMessage, log, this._stepInfo);
 }
 function writeFileSync(aPath, content) {
   import_fs.default.mkdirSync(import_path.default.dirname(aPath), { recursive: true });

package/lib/mcp/test/browserBackend.js

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,52 +17,58 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var browserBackend_exports = {};
 __export(browserBackend_exports, {
-  createCustomMessageHandler: () => createCustomMessageHandler
+  createCustomMessageHandler: () => createCustomMessageHandler,
+  runDaemonForContext: () => runDaemonForContext
 });
 module.exports = __toCommonJS(browserBackend_exports);
-var import_config = require("../browser/config");
-var import_browserServerBackend = require("../browser/browserServerBackend");
-var import_tab = require("../browser/tab");
-var import_util = require("../../util");
-var import_browserContextFactory = require("../browser/browserContextFactory");
+var import_crypto = __toESM(require("crypto"));
+var import_utils = require("patchright-bun-core/lib/utils");
 function createCustomMessageHandler(testInfo, context) {
   let backend;
+  const config = { capabilities: ["testing"] };
+  let tools;
   return async (data) => {
+    if (!tools)
+      tools = await import("playwright-core/lib/tools/exports");
+    const toolList = tools.filteredTools(config);
     if (data.initialize) {
       if (backend)
         throw new Error("MCP backend is already initialized");
-      backend = new import_browserServerBackend.BrowserServerBackend({ ...import_config.defaultConfig, capabilities: ["testing"] }, (0, import_browserContextFactory.identityBrowserContextFactory)(context));
+      backend = new tools.BrowserBackend(config, context, toolList);
       await backend.initialize(data.initialize.clientInfo);
-      const pausedMessage = await generatePausedMessage(testInfo, context);
+      const pausedMessage = await generatePausedMessage(tools, testInfo, context);
       return { initialize: { pausedMessage } };
     }
-    if (data.listTools) {
-      if (!backend)
-        throw new Error("MCP backend is not initialized");
-      return { listTools: await backend.listTools() };
-    }
     if (data.callTool) {
       if (!backend)
         throw new Error("MCP backend is not initialized");
       return { callTool: await backend.callTool(data.callTool.name, data.callTool.arguments) };
     }
     if (data.close) {
-      backend?.serverClosed();
+      await backend?.dispose();
       backend = void 0;
       return { close: {} };
     }
     throw new Error("Unknown MCP request");
   };
 }
-async function generatePausedMessage(testInfo, context) {
+async function generatePausedMessage(tools, testInfo, context) {
   const lines = [];
   if (testInfo.errors.length) {
     lines.push(`### Paused on error:`);
     for (const error of testInfo.errors)
-      lines.push((0, import_util.stripAnsiEscapes)(error.message || ""));
+      lines.push((0, import_utils.stripAnsiEscapes)(error.message || ""));
   } else {
     lines.push(`### Paused at end of test. ready for interaction`);
   }
@@ -73,17 +81,17 @@ async function generatePausedMessage(testInfo, context) {
       `- Page URL: ${page.url()}`,
       `- Page Title: ${await page.title()}`.trim()
     );
-    let console = testInfo.errors.length ? await import_tab.Tab.collectConsoleMessages(page) : [];
-    console = console.filter((msg) => msg.type === "error");
-    if (console.length) {
+    let console2 = testInfo.errors.length ? await tools.Tab.collectConsoleMessages(page) : [];
+    console2 = console2.filter((msg) => msg.type === "error");
+    if (console2.length) {
       lines.push("- Console Messages:");
-      for (const message of console)
+      for (const message of console2)
         lines.push(`  - ${message.toString()}`);
     }
     lines.push(
       `- Page Snapshot:`,
       "```yaml",
-      (await page._snapshotForAI()).full,
+      await page.ariaSnapshot({ mode: "ai" }),
       "```"
     );
   }
@@ -92,7 +100,22 @@ async function generatePausedMessage(testInfo, context) {
     lines.push(`### Task`, `Try recovering from the error prior to continuing`);
   return lines.join("\n");
 }
+async function runDaemonForContext(testInfo, context) {
+  if (testInfo._configInternal.configCLIOverrides.debug !== "cli")
+    return false;
+  const sessionName = `tw-${import_crypto.default.randomBytes(3).toString("hex")}`;
+  await context.browser().bind(sessionName, { workspaceDir: testInfo.project.testDir });
+  console.log([
+    `### The test is currently paused at the start`,
+    ``,
+    `### Debugging Instructions`,
+    `- Run "playwright-cli attach ${sessionName}" to attach to this test`
+  ].join("\n"));
+  await context.debugger.requestPause();
+  return true;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  createCustomMessageHandler
+  createCustomMessageHandler,
+  runDaemonForContext
 });

package/lib/mcp/test/generatorTools.js

@@ -35,7 +35,7 @@ __export(generatorTools_exports, {
 module.exports = __toCommonJS(generatorTools_exports);
 var import_fs = __toESM(require("fs"));
 var import_path = __toESM(require("path"));
-var import_mcpBundle = require("patchright-bun-core/lib/mcpBundle");
+var import_zodBundle = require("patchright-bun-core/lib/zodBundle");
 var import_testTool = require("./testTool");
 var import_testContext = require("./testContext");
 const setupPage = (0, import_testTool.defineTestTool)({
@@ -43,10 +43,10 @@ const setupPage = (0, import_testTool.defineTestTool)({
     name: "generator_setup_page",
     title: "Setup generator page",
     description: "Setup the page for test.",
-    inputSchema: import_mcpBundle.z.object({
-      plan: import_mcpBundle.z.string().describe("The plan for the test. This should be the actual test plan with all the steps."),
-      project: import_mcpBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
-      seedFile: import_mcpBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
+    inputSchema: import_zodBundle.z.object({
+      plan: import_zodBundle.z.string().describe("The plan for the test. This should be the actual test plan with all the steps."),
+      project: import_zodBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
+      seedFile: import_zodBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
     }),
     type: "readOnly"
   },
@@ -62,7 +62,7 @@ const generatorReadLog = (0, import_testTool.defineTestTool)({
     name: "generator_read_log",
     title: "Retrieve test log",
     description: "Retrieve the performed test log",
-    inputSchema: import_mcpBundle.z.object({}),
+    inputSchema: import_zodBundle.z.object({}),
     type: "readOnly"
   },
   handle: async (context) => {
@@ -80,9 +80,9 @@ const generatorWriteTest = (0, import_testTool.defineTestTool)({
     name: "generator_write_test",
     title: "Write test",
     description: "Write the generated test to the test file",
-    inputSchema: import_mcpBundle.z.object({
-      fileName: import_mcpBundle.z.string().describe("The file to write the test to"),
-      code: import_mcpBundle.z.string().describe("The generated test code")
+    inputSchema: import_zodBundle.z.object({
+      fileName: import_zodBundle.z.string().describe("The file to write the test to"),
+      code: import_zodBundle.z.string().describe("The generated test code")
     }),
     type: "readOnly"
   },

package/lib/mcp/test/plannerTools.js

@@ -35,16 +35,16 @@ __export(plannerTools_exports, {
 module.exports = __toCommonJS(plannerTools_exports);
 var import_fs = __toESM(require("fs"));
 var import_path = __toESM(require("path"));
-var import_mcpBundle = require("patchright-bun-core/lib/mcpBundle");
+var import_zodBundle = require("patchright-bun-core/lib/zodBundle");
 var import_testTool = require("./testTool");
 const setupPage = (0, import_testTool.defineTestTool)({
   schema: {
     name: "planner_setup_page",
     title: "Setup planner page",
     description: "Setup the page for test planning",
-    inputSchema: import_mcpBundle.z.object({
-      project: import_mcpBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
-      seedFile: import_mcpBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
+    inputSchema: import_zodBundle.z.object({
+      project: import_zodBundle.z.string().optional().describe('Project to use for setup. For example: "chromium", if no project is provided uses the first project in the config.'),
+      seedFile: import_zodBundle.z.string().optional().describe('A seed file contains a single test that is used to setup the page for testing, for example: "tests/seed.spec.ts". If no seed file is provided, a default seed file is created.')
     }),
     type: "readOnly"
   },
@@ -54,17 +54,17 @@ const setupPage = (0, import_testTool.defineTestTool)({
     return { content: [{ type: "text", text: output }], isError: status !== "paused" };
   }
 });
-const planSchema = import_mcpBundle.z.object({
-  overview: import_mcpBundle.z.string().describe("A brief overview of the application to be tested"),
-  suites: import_mcpBundle.z.array(import_mcpBundle.z.object({
-    name: import_mcpBundle.z.string().describe("The name of the suite"),
-    seedFile: import_mcpBundle.z.string().describe("A seed file that was used to setup the page for testing."),
-    tests: import_mcpBundle.z.array(import_mcpBundle.z.object({
-      name: import_mcpBundle.z.string().describe("The name of the test"),
-      file: import_mcpBundle.z.string().describe('The file the test should be saved to, for example: "tests/<suite-name>/<test-name>.spec.ts".'),
-      steps: import_mcpBundle.z.array(import_mcpBundle.z.object({
-        perform: import_mcpBundle.z.string().optional().describe(`Action to perform. For example: 'Click on the "Submit" button'.`),
-        expect: import_mcpBundle.z.string().array().describe(`Expected result of the action where appropriate. For example: 'The page should show the "Thank you for your submission" message'`)
+const planSchema = import_zodBundle.z.object({
+  overview: import_zodBundle.z.string().describe("A brief overview of the application to be tested"),
+  suites: import_zodBundle.z.array(import_zodBundle.z.object({
+    name: import_zodBundle.z.string().describe("The name of the suite"),
+    seedFile: import_zodBundle.z.string().describe("A seed file that was used to setup the page for testing."),
+    tests: import_zodBundle.z.array(import_zodBundle.z.object({
+      name: import_zodBundle.z.string().describe("The name of the test"),
+      file: import_zodBundle.z.string().describe('The file the test should be saved to, for example: "tests/<suite-name>/<test-name>.spec.ts".'),
+      steps: import_zodBundle.z.array(import_zodBundle.z.object({
+        perform: import_zodBundle.z.string().optional().describe(`Action to perform. For example: 'Click on the "Submit" button'.`),
+        expect: import_zodBundle.z.string().array().describe(`Expected result of the action where appropriate. For example: 'The page should show the "Thank you for your submission" message'`)
       }))
     }))
   }))
@@ -92,8 +92,8 @@ const saveTestPlan = (0, import_testTool.defineTestTool)({
     title: "Save test plan as markdown file",
     description: "Save the test plan as a markdown file",
     inputSchema: planSchema.extend({
-      name: import_mcpBundle.z.string().describe('The name of the test plan, for example: "Test Plan".'),
-      fileName: import_mcpBundle.z.string().describe('The file to save the test plan to, for example: "spec/test.plan.md". Relative to the workspace root.')
+      name: import_zodBundle.z.string().describe('The name of the test plan, for example: "Test Plan".'),
+      fileName: import_zodBundle.z.string().describe('The file to save the test plan to, for example: "spec/test.plan.md". Relative to the workspace root.')
     }),
     type: "readOnly"
   },

package/lib/mcp/test/testBackend.js

@@ -28,59 +28,58 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var testBackend_exports = {};
 __export(testBackend_exports, {
-  TestServerBackend: () => TestServerBackend
+  TestServerBackend: () => TestServerBackend,
+  testServerBackendTools: () => testServerBackendTools
 });
 module.exports = __toCommonJS(testBackend_exports);
-var import_mcpBundle = require("patchright-bun-core/lib/mcpBundle");
-var mcp = __toESM(require("../sdk/exports"));
+var import_events = __toESM(require("events"));
+var import_zodBundle = require("patchright-bun-core/lib/zodBundle");
+var import_exports = require("patchright-bun-core/lib/tools/exports");
 var import_testContext = require("./testContext");
 var testTools = __toESM(require("./testTools.js"));
 var generatorTools = __toESM(require("./generatorTools.js"));
 var plannerTools = __toESM(require("./plannerTools.js"));
-var import_tools = require("../browser/tools");
-class TestServerBackend {
+const typesWithIntent = ["action", "assertion", "input"];
+const testServerBackendTools = [
+  plannerTools.saveTestPlan,
+  plannerTools.setupPage,
+  plannerTools.submitTestPlan,
+  generatorTools.setupPage,
+  generatorTools.generatorReadLog,
+  generatorTools.generatorWriteTest,
+  testTools.listTests,
+  testTools.runTests,
+  testTools.debugTest,
+  ...import_exports.browserTools.map((tool) => wrapBrowserTool(tool))
+];
+class TestServerBackend extends import_events.default {
   constructor(configPath, options) {
+    super();
     this.name = "Playwright";
     this.version = "0.0.1";
-    this._tools = [
-      plannerTools.saveTestPlan,
-      plannerTools.setupPage,
-      plannerTools.submitTestPlan,
-      generatorTools.setupPage,
-      generatorTools.generatorReadLog,
-      generatorTools.generatorWriteTest,
-      testTools.listTests,
-      testTools.runTests,
-      testTools.debugTest,
-      ...import_tools.browserTools.map((tool) => wrapBrowserTool(tool))
-    ];
     this._options = options || {};
     this._configPath = configPath;
   }
   async initialize(clientInfo) {
     this._context = new import_testContext.TestContext(clientInfo, this._configPath, this._options);
   }
-  async listTools() {
-    return this._tools.map((tool) => mcp.toMcpTool(tool.schema));
-  }
   async callTool(name, args) {
-    const tool = this._tools.find((tool2) => tool2.schema.name === name);
+    const tool = testServerBackendTools.find((tool2) => tool2.schema.name === name);
     if (!tool)
-      throw new Error(`Tool not found: ${name}. Available tools: ${this._tools.map((tool2) => tool2.schema.name).join(", ")}`);
+      throw new Error(`Tool not found: ${name}. Available tools: ${testServerBackendTools.map((tool2) => tool2.schema.name).join(", ")}`);
     try {
       return await tool.handle(this._context, tool.schema.inputSchema.parse(args || {}));
     } catch (e) {
       return { content: [{ type: "text", text: String(e) }], isError: true };
     }
   }
-  serverClosed() {
-    void this._context?.close();
+  async dispose() {
+    await this._context?.close();
   }
 }
-const typesWithIntent = ["action", "assertion", "input"];
 function wrapBrowserTool(tool) {
   const inputSchema = typesWithIntent.includes(tool.schema.type) ? tool.schema.inputSchema.extend({
-    intent: import_mcpBundle.z.string().describe("The intent of the call, for example the test step description plan idea")
+    intent: import_zodBundle.z.string().describe("The intent of the call, for example the test step description plan idea")
   }) : tool.schema.inputSchema;
   return {
     schema: {
@@ -95,5 +94,6 @@ function wrapBrowserTool(tool) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  TestServerBackend
+  TestServerBackend,
+  testServerBackendTools
 });

package/lib/mcp/test/testContext.js

@@ -37,16 +37,15 @@ var import_fs = __toESM(require("fs"));
 var import_os = __toESM(require("os"));
 var import_path = __toESM(require("path"));
 var import_utils = require("patchright-bun-core/lib/utils");
+var import_exports = require("patchright-bun-core/lib/tools/exports");
+var import_utilsBundle = require("patchright-bun-core/lib/utilsBundle");
 var import_base = require("../../reporters/base");
 var import_list = __toESM(require("../../reporters/list"));
 var import_streams = require("./streams");
 var import_util = require("../../util");
 var import_testRunner = require("../../runner/testRunner");
 var import_seed = require("./seed");
-var import_exports = require("../sdk/exports");
 var import_configLoader = require("../../common/configLoader");
-var import_response = require("../browser/response");
-var import_log = require("../log");
 class GeneratorJournal {
   constructor(rootPath, plan, seed) {
     this._rootPath = rootPath;
@@ -78,9 +77,8 @@ ${step.code}
 class TestContext {
   constructor(clientInfo, configPath, options) {
     this._clientInfo = clientInfo;
-    const rootPath = (0, import_exports.firstRootPath)(clientInfo);
-    this._configLocation = (0, import_configLoader.resolveConfigLocation)(configPath || rootPath);
-    this.rootPath = rootPath || this._configLocation.configDir;
+    this._configLocation = (0, import_configLoader.resolveConfigLocation)(configPath || clientInfo.cwd);
+    this.rootPath = clientInfo.cwd || this._configLocation.configDir;
     if (options?.headless !== void 0)
       this.computedHeaded = !options.headless;
     else
@@ -213,7 +211,7 @@ class TestContext {
     return { output: testRunnerAndScreen.output.join("\n"), status };
   }
   async close() {
-    await this._cleanupTestRunner().catch(import_log.logUnhandledError);
+    await this._cleanupTestRunner().catch((e) => (0, import_utilsBundle.debug)("pw:mcp:error")(e));
   }
   async sendMessageToPausedTest(request) {
     const sendMessage = this._testRunnerAndScreen?.sendMessageToPausedTest;
@@ -223,7 +221,7 @@ class TestContext {
     if (result.error)
       throw new Error(result.error.message);
     if (typeof request?.callTool?.arguments?.["intent"] === "string") {
-      const response = (0, import_response.parseResponse)(result.response.callTool);
+      const response = (0, import_exports.parseResponse)(result.response.callTool);
       if (response && !response.isError && response.code)
         this.generatorJournal?.logStep(request.callTool.arguments["intent"], response.code);
     }

package/lib/mcp/test/testTools.js

@@ -33,7 +33,7 @@ __export(testTools_exports, {
   runTests: () => runTests
 });
 module.exports = __toCommonJS(testTools_exports);
-var import_mcpBundle = require("patchright-bun-core/lib/mcpBundle");
+var import_zodBundle = require("patchright-bun-core/lib/zodBundle");
 var import_listModeReporter = __toESM(require("../../reporters/listModeReporter"));
 var import_testTool = require("./testTool");
 const listTests = (0, import_testTool.defineTestTool)({
@@ -41,7 +41,7 @@ const listTests = (0, import_testTool.defineTestTool)({
     name: "test_list",
     title: "List tests",
     description: "List tests",
-    inputSchema: import_mcpBundle.z.object({}),
+    inputSchema: import_zodBundle.z.object({}),
     type: "readOnly"
   },
   handle: async (context) => {
@@ -56,9 +56,9 @@ const runTests = (0, import_testTool.defineTestTool)({
     name: "test_run",
     title: "Run tests",
     description: "Run tests",
-    inputSchema: import_mcpBundle.z.object({
-      locations: import_mcpBundle.z.array(import_mcpBundle.z.string()).optional().describe('Folder, file or location to run: "test/e2e" or "test/e2e/file.spec.ts" or "test/e2e/file.spec.ts:20"'),
-      projects: import_mcpBundle.z.array(import_mcpBundle.z.string()).optional().describe('Projects to run, projects from playwright.config.ts, by default runs all projects. Running with "chromium" is a good start')
+    inputSchema: import_zodBundle.z.object({
+      locations: import_zodBundle.z.array(import_zodBundle.z.string()).optional().describe('Folder, file or location to run: "test/e2e" or "test/e2e/file.spec.ts" or "test/e2e/file.spec.ts:20"'),
+      projects: import_zodBundle.z.array(import_zodBundle.z.string()).optional().describe('Projects to run, projects from playwright.config.ts, by default runs all projects. Running with "chromium" is a good start')
     }),
     type: "readOnly"
   },
@@ -76,10 +76,10 @@ const debugTest = (0, import_testTool.defineTestTool)({
     name: "test_debug",
     title: "Debug single test",
     description: "Debug single test",
-    inputSchema: import_mcpBundle.z.object({
-      test: import_mcpBundle.z.object({
-        id: import_mcpBundle.z.string().describe("Test ID to debug."),
-        title: import_mcpBundle.z.string().describe("Human readable test title for granting permission to debug the test.")
+    inputSchema: import_zodBundle.z.object({
+      test: import_zodBundle.z.object({
+        id: import_zodBundle.z.string().describe("Test ID to debug."),
+        title: import_zodBundle.z.string().describe("Human readable test title for granting permission to debug the test.")
       })
     }),
     type: "readOnly"

package/lib/plugins/webServerPlugin.js

@@ -53,7 +53,8 @@ class WebServerPlugin {
   }
   async setup(config, configDir, reporter) {
     this._reporter = reporter;
-    this._isAvailableCallback = this._options.url ? getIsAvailableFunction(this._options.url, this._checkPortOnly, !!this._options.ignoreHTTPSErrors, this._reporter.onStdErr?.bind(this._reporter)) : void 0;
+    if (this._options.url)
+      this._isAvailableCallback = getIsAvailableFunction(this._options.url, this._checkPortOnly, !!this._options.ignoreHTTPSErrors, this._reporter.onStdErr?.bind(this._reporter));
     this._options.cwd = this._options.cwd ? import_path.default.resolve(configDir, this._options.cwd) : configDir;
     try {
       await this._startProcess();