@browserbasehq/orca 3.0.9-alpha-3 → 3.1.0-headers

package/dist/esm/tests/unit/api-multiregion.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api-multiregion.test.js","sourceRoot":"","sources":["../../../../tests/unit/api-multiregion.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEvE,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACvC,uCAAuC,CACxC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACvC,4CAA4C,CAC7C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAC1C,4CAA4C,CAC7C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAC5C,6CAA6C,CAC9C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAC1C,0CAA0C,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAC1C,+CAA+C,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAC7C,+CAA+C,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAC/C,gDAAgD,CACjD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CACxC,0CAA0C,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,4CAA4C;YAC5C,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAC/C,0CAA0C,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,4CAA4C;YAC5C,0EAA0E;YAC1E,MAAM,GAAG,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,0DAA0D;YAC1D,gDAAgD;YAChD,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;gBAChE,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC5C,MAAM,CAAC,kBAAkB,CAAC,MAAsC,CAAC,CAAC,CAAC,IAAI,CACrE,GAAG,OAAO,KAAK,CAChB,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { getApiUrlForRegion, REGION_API_URLS } from \"../../lib/v3/api\";\n\ndescribe(\"Multi-region API URL mapping\", () => {\n  describe(\"REGION_API_URLS constant\", () => {\n    it(\"should have the correct URL for us-west-2 (default)\", () => {\n      expect(REGION_API_URLS[\"us-west-2\"]).toBe(\n        \"https://api.stagehand.browserbase.com\",\n      );\n    });\n\n    it(\"should have the correct URL for us-east-1\", () => {\n      expect(REGION_API_URLS[\"us-east-1\"]).toBe(\n        \"https://api.use1.stagehand.browserbase.com\",\n      );\n    });\n\n    it(\"should have the correct URL for eu-central-1\", () => {\n      expect(REGION_API_URLS[\"eu-central-1\"]).toBe(\n        \"https://api.euc1.stagehand.browserbase.com\",\n      );\n    });\n\n    it(\"should have the correct URL for ap-southeast-1\", () => {\n      expect(REGION_API_URLS[\"ap-southeast-1\"]).toBe(\n        \"https://api.apse1.stagehand.browserbase.com\",\n      );\n    });\n  });\n\n  describe(\"getApiUrlForRegion\", () => {\n    it(\"should return the correct URL for us-west-2\", () => {\n      expect(getApiUrlForRegion(\"us-west-2\")).toBe(\n        \"https://api.stagehand.browserbase.com/v1\",\n      );\n    });\n\n    it(\"should return the correct URL for us-east-1\", () => {\n      expect(getApiUrlForRegion(\"us-east-1\")).toBe(\n        \"https://api.use1.stagehand.browserbase.com/v1\",\n      );\n    });\n\n    it(\"should return the correct URL for eu-central-1\", () => {\n      expect(getApiUrlForRegion(\"eu-central-1\")).toBe(\n        \"https://api.euc1.stagehand.browserbase.com/v1\",\n      );\n    });\n\n    it(\"should return the correct URL for ap-southeast-1\", () => {\n      expect(getApiUrlForRegion(\"ap-southeast-1\")).toBe(\n        \"https://api.apse1.stagehand.browserbase.com/v1\",\n      );\n    });\n\n    it(\"should return the default us-west-2 URL when no region is specified\", () => {\n      expect(getApiUrlForRegion(undefined)).toBe(\n        \"https://api.stagehand.browserbase.com/v1\",\n      );\n    });\n\n    it(\"should return the default us-west-2 URL for unknown regions\", () => {\n      // @ts-expect-error - testing invalid region\n      expect(getApiUrlForRegion(\"invalid-region\")).toBe(\n        \"https://api.stagehand.browserbase.com/v1\",\n      );\n    });\n  });\n\n  describe(\"URL /v1 suffix handling\", () => {\n    it(\"getApiUrlForRegion always includes /v1 suffix for consistency\", () => {\n      // getApiUrlForRegion returns a URL with /v1\n      // This documents the expected contract that all API base URLs include /v1\n      const url = getApiUrlForRegion(\"us-west-2\");\n      expect(url.endsWith(\"/v1\")).toBe(true);\n    });\n\n    it(\"all regional URLs should be base URLs without /v1 in REGION_API_URLS\", () => {\n      // Verify REGION_API_URLS contains base URLs (without /v1)\n      // The /v1 suffix is added by getApiUrlForRegion\n      for (const [region, baseUrl] of Object.entries(REGION_API_URLS)) {\n        expect(baseUrl.endsWith(\"/v1\")).toBe(false);\n        expect(getApiUrlForRegion(region as keyof typeof REGION_API_URLS)).toBe(\n          `${baseUrl}/v1`,\n        );\n      }\n    });\n  });\n});\n"]}

package/dist/esm/tests/unit/browserbase-session-accessors.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/browserbase-session-accessors.test.js

@@ -0,0 +1,100 @@
+import { describe, expect, it, vi, beforeEach, afterEach } from "vitest";
+import { V3 } from "../../lib/v3/v3.js";
+const MOCK_SESSION_ID = "session-123";
+const MOCK_SESSION_URL = `https://www.browserbase.com/sessions/${MOCK_SESSION_ID}`;
+const MOCK_DEBUG_URL = `https://debug.browserbase.com/${MOCK_SESSION_ID}`;
+vi.mock("../../lib/v3/understudy/context", () => {
+    class MockConnection {
+        onTransportClosed = vi.fn();
+        offTransportClosed = vi.fn();
+        send = vi.fn(async () => { });
+    }
+    class MockV3Context {
+        static async create() {
+            return new MockV3Context();
+        }
+        conn = new MockConnection();
+        pages() {
+            return [];
+        }
+        async close() {
+            // noop
+        }
+    }
+    return { V3Context: MockV3Context };
+});
+vi.mock("../../lib/v3/launch/browserbase", () => ({
+    createBrowserbaseSession: vi.fn(async () => ({
+        ws: "wss://mock-browserbase",
+        sessionId: MOCK_SESSION_ID,
+        bb: {
+            sessions: {
+                debug: vi.fn(async () => ({ debuggerUrl: MOCK_DEBUG_URL })),
+            },
+        },
+    })),
+}));
+vi.mock("../../lib/v3/launch/local", () => ({
+    launchLocalChrome: vi.fn(async () => ({
+        ws: "ws://local-cdp",
+        chrome: { kill: vi.fn(async () => { }) },
+    })),
+}));
+describe("browserbase accessors", () => {
+    beforeEach(() => {
+        process.env.BROWSERBASE_API_KEY = "fake-key";
+        process.env.BROWSERBASE_PROJECT_ID = "fake-project";
+    });
+    afterEach(() => {
+        delete process.env.BROWSERBASE_API_KEY;
+        delete process.env.BROWSERBASE_PROJECT_ID;
+        vi.clearAllMocks();
+    });
+    it("exposes Browserbase session and debug URLs after init", async () => {
+        const v3 = new V3({
+            env: "BROWSERBASE",
+            disableAPI: true,
+            verbose: 0,
+        });
+        try {
+            await v3.init();
+            expect(v3.browserbaseSessionURL).toBe(MOCK_SESSION_URL);
+            expect(v3.browserbaseDebugURL).toBe(MOCK_DEBUG_URL);
+        }
+        finally {
+            await v3.close().catch(() => { });
+        }
+    });
+    it("clears stored URLs after close", async () => {
+        const v3 = new V3({
+            env: "BROWSERBASE",
+            disableAPI: true,
+            verbose: 0,
+        });
+        await v3.init();
+        await v3.close();
+        expect(v3.browserbaseSessionURL).toBeUndefined();
+        expect(v3.browserbaseDebugURL).toBeUndefined();
+    });
+});
+describe("local accessors", () => {
+    it("stay empty for LOCAL environments", async () => {
+        const v3 = new V3({
+            env: "LOCAL",
+            disableAPI: true,
+            verbose: 0,
+            localBrowserLaunchOptions: {
+                cdpUrl: "ws://local-existing-session",
+            },
+        });
+        try {
+            await v3.init();
+            expect(v3.browserbaseSessionURL).toBeUndefined();
+            expect(v3.browserbaseDebugURL).toBeUndefined();
+        }
+        finally {
+            await v3.close().catch(() => { });
+        }
+    });
+});
+//# sourceMappingURL=browserbase-session-accessors.test.js.map

package/dist/esm/tests/unit/browserbase-session-accessors.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"browserbase-session-accessors.test.js","sourceRoot":"","sources":["../../../../tests/unit/browserbase-session-accessors.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAExC,MAAM,eAAe,GAAG,aAAa,CAAC;AACtC,MAAM,gBAAgB,GAAG,wCAAwC,eAAe,EAAE,CAAC;AACnF,MAAM,cAAc,GAAG,iCAAiC,eAAe,EAAE,CAAC;AAE1E,EAAE,CAAC,IAAI,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC9C,MAAM,cAAc;QAClB,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7B,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC;KAC9B;IAED,MAAM,aAAa;QACjB,MAAM,CAAC,KAAK,CAAC,MAAM;YACjB,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,GAAG,IAAI,cAAc,EAAE,CAAC;QAE5B,KAAK;YACH,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,KAAK,CAAC,KAAK;YACT,OAAO;QACT,CAAC;KACF;IAED,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;AACtC,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,iCAAiC,EAAE,GAAG,EAAE,CAAC,CAAC;IAChD,wBAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAC3C,EAAE,EAAE,wBAAwB;QAC5B,SAAS,EAAE,eAAe;QAC1B,EAAE,EAAE;YACF,QAAQ,EAAE;gBACR,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC,CAAC;aAC5D;SACF;KACF,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,iBAAiB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QACpC,EAAE,EAAE,gBAAgB;QACpB,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,EAAE;KACxC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,UAAU,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,sBAAsB,GAAG,cAAc,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QACvC,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;QAC1C,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;YAChB,GAAG,EAAE,aAAa;YAClB,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YAEhB,MAAM,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;YAChB,GAAG,EAAE,aAAa;YAClB,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QAEH,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAChB,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,aAAa,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC;YAChB,GAAG,EAAE,OAAO;YACZ,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,CAAC;YACV,yBAAyB,EAAE;gBACzB,MAAM,EAAE,6BAA6B;aACtC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YAChB,MAAM,CAAC,EAAE,CAAC,qBAAqB,CAAC,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,CAAC,EAAE,CAAC,mBAAmB,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,CAAC;gBAAS,CAAC;YACT,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi, beforeEach, afterEach } from \"vitest\";\nimport { V3 } from \"../../lib/v3/v3.js\";\n\nconst MOCK_SESSION_ID = \"session-123\";\nconst MOCK_SESSION_URL = `https://www.browserbase.com/sessions/${MOCK_SESSION_ID}`;\nconst MOCK_DEBUG_URL = `https://debug.browserbase.com/${MOCK_SESSION_ID}`;\n\nvi.mock(\"../../lib/v3/understudy/context\", () => {\n  class MockConnection {\n    onTransportClosed = vi.fn();\n    offTransportClosed = vi.fn();\n    send = vi.fn(async () => {});\n  }\n\n  class MockV3Context {\n    static async create(): Promise<MockV3Context> {\n      return new MockV3Context();\n    }\n\n    conn = new MockConnection();\n\n    pages(): never[] {\n      return [];\n    }\n\n    async close(): Promise<void> {\n      // noop\n    }\n  }\n\n  return { V3Context: MockV3Context };\n});\n\nvi.mock(\"../../lib/v3/launch/browserbase\", () => ({\n  createBrowserbaseSession: vi.fn(async () => ({\n    ws: \"wss://mock-browserbase\",\n    sessionId: MOCK_SESSION_ID,\n    bb: {\n      sessions: {\n        debug: vi.fn(async () => ({ debuggerUrl: MOCK_DEBUG_URL })),\n      },\n    },\n  })),\n}));\n\nvi.mock(\"../../lib/v3/launch/local\", () => ({\n  launchLocalChrome: vi.fn(async () => ({\n    ws: \"ws://local-cdp\",\n    chrome: { kill: vi.fn(async () => {}) },\n  })),\n}));\n\ndescribe(\"browserbase accessors\", () => {\n  beforeEach(() => {\n    process.env.BROWSERBASE_API_KEY = \"fake-key\";\n    process.env.BROWSERBASE_PROJECT_ID = \"fake-project\";\n  });\n\n  afterEach(() => {\n    delete process.env.BROWSERBASE_API_KEY;\n    delete process.env.BROWSERBASE_PROJECT_ID;\n    vi.clearAllMocks();\n  });\n\n  it(\"exposes Browserbase session and debug URLs after init\", async () => {\n    const v3 = new V3({\n      env: \"BROWSERBASE\",\n      disableAPI: true,\n      verbose: 0,\n    });\n\n    try {\n      await v3.init();\n\n      expect(v3.browserbaseSessionURL).toBe(MOCK_SESSION_URL);\n      expect(v3.browserbaseDebugURL).toBe(MOCK_DEBUG_URL);\n    } finally {\n      await v3.close().catch(() => {});\n    }\n  });\n\n  it(\"clears stored URLs after close\", async () => {\n    const v3 = new V3({\n      env: \"BROWSERBASE\",\n      disableAPI: true,\n      verbose: 0,\n    });\n\n    await v3.init();\n    await v3.close();\n\n    expect(v3.browserbaseSessionURL).toBeUndefined();\n    expect(v3.browserbaseDebugURL).toBeUndefined();\n  });\n});\n\ndescribe(\"local accessors\", () => {\n  it(\"stay empty for LOCAL environments\", async () => {\n    const v3 = new V3({\n      env: \"LOCAL\",\n      disableAPI: true,\n      verbose: 0,\n      localBrowserLaunchOptions: {\n        cdpUrl: \"ws://local-existing-session\",\n      },\n    });\n\n    try {\n      await v3.init();\n      expect(v3.browserbaseSessionURL).toBeUndefined();\n      expect(v3.browserbaseDebugURL).toBeUndefined();\n    } finally {\n      await v3.close().catch(() => {});\n    }\n  });\n});\n"]}

package/dist/esm/tests/unit/cache-llm-resolution.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/cache-llm-resolution.test.js

@@ -0,0 +1,184 @@
+import { describe, expect, it, vi } from "vitest";
+import { ActCache } from "../../lib/v3/cache/ActCache.js";
+import { AgentCache } from "../../lib/v3/cache/AgentCache.js";
+function createFakeStorage(entry) {
+    return {
+        enabled: true,
+        readJson: vi.fn().mockResolvedValue({ value: entry }),
+        writeJson: vi.fn().mockResolvedValue({}),
+        directory: "/tmp/cache",
+    };
+}
+describe("Cache LLM client selection", () => {
+    it("ActCache uses provided override client during replay", async () => {
+        const action = {
+            selector: "xpath=/html/body/button",
+            description: "click button",
+            method: "click",
+            arguments: [],
+        };
+        const entry = {
+            version: 1,
+            instruction: "click button",
+            url: "https://example.com",
+            variableKeys: [],
+            actions: [action],
+            actionDescription: "click button",
+            message: "done",
+        };
+        const storage = createFakeStorage(entry);
+        const handler = {
+            takeDeterministicAction: vi.fn().mockResolvedValue({
+                success: true,
+                message: "ok",
+                actionDescription: "click button",
+                actions: [action],
+            }),
+        };
+        const defaultClient = { id: "default" };
+        const overrideClient = { id: "override" };
+        const cache = new ActCache({
+            storage,
+            logger: vi.fn(),
+            getActHandler: () => handler,
+            getDefaultLlmClient: () => defaultClient,
+            domSettleTimeoutMs: undefined,
+        });
+        const context = {
+            instruction: "click button",
+            cacheKey: "abc",
+            pageUrl: "https://example.com",
+            variableKeys: [],
+            variables: undefined,
+        };
+        const result = await cache.tryReplay(context, {}, undefined, overrideClient);
+        expect(result?.success).toBe(true);
+        expect(handler.takeDeterministicAction).toHaveBeenCalledTimes(1);
+        const call = vi.mocked(handler.takeDeterministicAction).mock.calls[0];
+        expect(call?.[3]).toBe(overrideClient);
+    });
+    it("AgentCache uses provided override client during replay", async () => {
+        const action = {
+            selector: "xpath=/html/body/input",
+            description: "type email",
+            method: "type",
+            arguments: ["test@example.com"],
+        };
+        const agentStep = {
+            type: "act",
+            instruction: "type email",
+            actions: [action],
+        };
+        const entry = {
+            version: 1,
+            instruction: "fill form",
+            startUrl: "https://example.com",
+            options: {},
+            configSignature: "sig",
+            steps: [agentStep],
+            result: { success: true, actions: [] },
+            timestamp: new Date().toISOString(),
+        };
+        const storage = {
+            enabled: true,
+            readJson: vi.fn().mockImplementation(async () => ({ value: entry })),
+            writeJson: vi.fn().mockResolvedValue({}),
+            directory: "/tmp/cache",
+        };
+        const handler = {
+            takeDeterministicAction: vi.fn().mockResolvedValue({
+                success: true,
+                message: "ok",
+                actionDescription: "type email",
+                actions: [action],
+            }),
+        };
+        const fakePage = {};
+        const ctx = {
+            awaitActivePage: vi.fn().mockResolvedValue(fakePage),
+        };
+        const defaultClient = { id: "default-agent" };
+        const overrideClient = { id: "override-agent" };
+        const cache = new AgentCache({
+            storage,
+            logger: vi.fn(),
+            getActHandler: () => handler,
+            getContext: () => ctx,
+            getDefaultLlmClient: () => defaultClient,
+            getBaseModelName: () => "openai/gpt-4.1-mini",
+            getSystemPrompt: () => undefined,
+            domSettleTimeoutMs: undefined,
+            act: vi.fn(),
+        });
+        const context = {
+            instruction: "fill form",
+            startUrl: "https://example.com",
+            options: {},
+            configSignature: "sig",
+            cacheKey: "agent-key",
+            variableKeys: [],
+        };
+        const result = await cache.tryReplay(context, overrideClient);
+        expect(result?.success).toBe(true);
+        expect(handler.takeDeterministicAction).toHaveBeenCalledTimes(1);
+        const call = vi.mocked(handler.takeDeterministicAction).mock.calls[0];
+        expect(call?.[3]).toBe(overrideClient);
+    });
+    it("AgentCache replays non-act steps without requiring an override client", async () => {
+        const gotoEntry = {
+            version: 1,
+            instruction: "navigate home",
+            startUrl: "https://example.com/source",
+            options: {},
+            configSignature: "sig",
+            steps: [
+                {
+                    type: "goto",
+                    url: "https://example.com/target",
+                    waitUntil: "load",
+                },
+            ],
+            result: { success: true, actions: [] },
+            timestamp: new Date().toISOString(),
+        };
+        const storage = {
+            enabled: true,
+            readJson: vi.fn().mockResolvedValue({ value: gotoEntry }),
+            writeJson: vi.fn().mockResolvedValue({}),
+            directory: "/tmp/cache",
+        };
+        const handler = {
+            takeDeterministicAction: vi.fn(),
+        };
+        const fakePage = { goto: vi.fn() };
+        const ctx = {
+            awaitActivePage: vi.fn().mockResolvedValue(fakePage),
+        };
+        const cache = new AgentCache({
+            storage,
+            logger: vi.fn(),
+            getActHandler: () => handler,
+            getContext: () => ctx,
+            getDefaultLlmClient: () => ({ id: "default" }),
+            getBaseModelName: () => "openai/gpt-4.1-mini",
+            getSystemPrompt: () => undefined,
+            domSettleTimeoutMs: undefined,
+            act: vi.fn(),
+        });
+        const context = {
+            instruction: "navigate home",
+            startUrl: "https://example.com/source",
+            options: {},
+            configSignature: "sig",
+            cacheKey: "agent-goto",
+            variableKeys: [],
+        };
+        const result = await cache.tryReplay(context);
+        expect(result?.success).toBe(true);
+        expect(handler.takeDeterministicAction).not.toHaveBeenCalled();
+        expect(fakePage.goto).toHaveBeenCalledWith("https://example.com/target", {
+            waitUntil: "load",
+        });
+    });
+});
+//# sourceMappingURL=cache-llm-resolution.test.js.map

package/dist/esm/tests/unit/cache-llm-resolution.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache-llm-resolution.test.js","sourceRoot":"","sources":["../../../../tests/unit/cache-llm-resolution.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,kCAAkC,CAAC;AAmB9D,SAAS,iBAAiB,CAAI,KAAQ;IACpC,OAAO;QACL,OAAO,EAAE,IAAI;QACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACrD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACxC,SAAS,EAAE,YAAY;KACG,CAAC;AAC/B,CAAC;AAED,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,MAAM,GAAW;YACrB,QAAQ,EAAE,yBAAyB;YACnC,WAAW,EAAE,cAAc;YAC3B,MAAM,EAAE,OAAO;YACf,SAAS,EAAE,EAAE;SACd,CAAC;QAEF,MAAM,KAAK,GAAmB;YAC5B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,cAAc;YAC3B,GAAG,EAAE,qBAAqB;YAC1B,YAAY,EAAE,EAAE;YAChB,OAAO,EAAE,CAAC,MAAM,CAAC;YACjB,iBAAiB,EAAE,cAAc;YACjC,OAAO,EAAE,MAAM;SAChB,CAAC;QAEF,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG;YACd,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjD,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,IAAI;gBACb,iBAAiB,EAAE,cAAc;gBACjC,OAAO,EAAE,CAAC,MAAM,CAAC;aAClB,CAAC;SACsB,CAAC;QAC3B,MAAM,aAAa,GAAG,EAAE,EAAE,EAAE,SAAS,EAA0B,CAAC;QAChE,MAAM,cAAc,GAAG,EAAE,EAAE,EAAE,UAAU,EAA0B,CAAC;QAElE,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC;YACzB,OAAO;YACP,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,aAAa,EAAE,GAAG,EAAE,CAAC,OAAO;YAC5B,mBAAmB,EAAE,GAAG,EAAE,CAAC,aAAa;YACxC,kBAAkB,EAAE,SAAS;SAC9B,CAAC,CAAC;QAEH,MAAM,OAAO,GAAoB;YAC/B,WAAW,EAAE,cAAc;YAC3B,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE,qBAAqB;YAC9B,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,SAAS;SACrB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,CAClC,OAAO,EACP,EAAU,EACV,SAAS,EACT,cAAc,CACf,CAAC;QAEF,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,MAAM,GAAW;YACrB,QAAQ,EAAE,wBAAwB;YAClC,WAAW,EAAE,YAAY;YACzB,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,CAAC,kBAAkB,CAAC;SAChC,CAAC;QAEF,MAAM,SAAS,GAAuB;YACpC,IAAI,EAAE,KAAK;YACX,WAAW,EAAE,YAAY;YACzB,OAAO,EAAE,CAAC,MAAM,CAAC;SAClB,CAAC;QAEF,MAAM,KAAK,GAAqB;YAC9B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,WAAW;YACxB,QAAQ,EAAE,qBAAqB;YAC/B,OAAO,EAAE,EAAE;YACX,eAAe,EAAE,KAAK;YACtB,KAAK,EAAE,CAAC,SAAS,CAAC;YAClB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAiB;YACrD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACpE,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACxC,SAAS,EAAE,YAAY;SACG,CAAC;QAE7B,MAAM,OAAO,GAAG;YACd,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjD,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,IAAI;gBACb,iBAAiB,EAAE,YAAY;gBAC/B,OAAO,EAAE,CAAC,MAAM,CAAC;aAClB,CAAC;SACsB,CAAC;QAE3B,MAAM,QAAQ,GAAG,EAAU,CAAC;QAC5B,MAAM,GAAG,GAAG;YACV,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;SAC7B,CAAC;QAE1B,MAAM,aAAa,GAAG,EAAE,EAAE,EAAE,eAAe,EAA0B,CAAC;QACtE,MAAM,cAAc,GAAG,EAAE,EAAE,EAAE,gBAAgB,EAA0B,CAAC;QAExE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC;YAC3B,OAAO;YACP,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,aAAa,EAAE,GAAG,EAAE,CAAC,OAAO;YAC5B,UAAU,EAAE,GAAG,EAAE,CAAC,GAAG;YACrB,mBAAmB,EAAE,GAAG,EAAE,CAAC,aAAa;YACxC,gBAAgB,EAAE,GAAG,EAAE,CAAC,qBAAuC;YAC/D,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;YAChC,kBAAkB,EAAE,SAAS;YAC7B,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAsB;YACjC,WAAW,EAAE,WAAW;YACxB,QAAQ,EAAE,qBAAqB;YAC/B,OAAO,EAAE,EAAE;YACX,eAAe,EAAE,KAAK;YACtB,QAAQ,EAAE,WAAW;YACrB,YAAY,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QAE9D,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,SAAS,GAAqB;YAClC,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,4BAA4B;YACtC,OAAO,EAAE,EAAE;YACX,eAAe,EAAE,KAAK;YACtB,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,GAAG,EAAE,4BAA4B;oBACjC,SAAS,EAAE,MAAM;iBAClB;aACF;YACD,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAiB;YACrD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YACzD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACxC,SAAS,EAAE,YAAY;SACG,CAAC;QAE7B,MAAM,OAAO,GAAG;YACd,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;SACR,CAAC;QAE3B,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAqB,CAAC;QACtD,MAAM,GAAG,GAAG;YACV,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;SAC7B,CAAC;QAE1B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC;YAC3B,OAAO;YACP,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,aAAa,EAAE,GAAG,EAAE,CAAC,OAAO;YAC5B,UAAU,EAAE,GAAG,EAAE,CAAC,GAAG;YACrB,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAyB;YACtE,gBAAgB,EAAE,GAAG,EAAE,CAAC,qBAAuC;YAC/D,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;YAChC,kBAAkB,EAAE,SAAS;YAC7B,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAsB;YACjC,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,4BAA4B;YACtC,OAAO,EAAE,EAAE;YACX,eAAe,EAAE,KAAK;YACtB,QAAQ,EAAE,YAAY;YACtB,YAAY,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAE9C,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,4BAA4B,EAAE;YACvE,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport { ActCache } from \"../../lib/v3/cache/ActCache.js\";\nimport { AgentCache } from \"../../lib/v3/cache/AgentCache.js\";\nimport type { CacheStorage } from \"../../lib/v3/cache/CacheStorage.js\";\nimport type { ActHandler } from \"../../lib/v3/handlers/actHandler.js\";\nimport type { LLMClient } from \"../../lib/v3/llm/LLMClient.js\";\nimport type { Page } from \"../../lib/v3/understudy/page.js\";\nimport type { V3Context } from \"../../lib/v3/understudy/context.js\";\nimport type {\n  ActCacheContext,\n  CachedActEntry,\n  CachedAgentEntry,\n  AgentCacheContext,\n  AgentReplayActStep,\n} from \"../../lib/v3/types/private/index.js\";\nimport type {\n  Action,\n  AgentResult,\n  AvailableModel,\n} from \"../../lib/v3/types/public/index.js\";\n\nfunction createFakeStorage<T>(entry: T): CacheStorage {\n  return {\n    enabled: true,\n    readJson: vi.fn().mockResolvedValue({ value: entry }),\n    writeJson: vi.fn().mockResolvedValue({}),\n    directory: \"/tmp/cache\",\n  } as unknown as CacheStorage;\n}\n\ndescribe(\"Cache LLM client selection\", () => {\n  it(\"ActCache uses provided override client during replay\", async () => {\n    const action: Action = {\n      selector: \"xpath=/html/body/button\",\n      description: \"click button\",\n      method: \"click\",\n      arguments: [],\n    };\n\n    const entry: CachedActEntry = {\n      version: 1,\n      instruction: \"click button\",\n      url: \"https://example.com\",\n      variableKeys: [],\n      actions: [action],\n      actionDescription: \"click button\",\n      message: \"done\",\n    };\n\n    const storage = createFakeStorage(entry);\n    const handler = {\n      takeDeterministicAction: vi.fn().mockResolvedValue({\n        success: true,\n        message: \"ok\",\n        actionDescription: \"click button\",\n        actions: [action],\n      }),\n    } as unknown as ActHandler;\n    const defaultClient = { id: \"default\" } as unknown as LLMClient;\n    const overrideClient = { id: \"override\" } as unknown as LLMClient;\n\n    const cache = new ActCache({\n      storage,\n      logger: vi.fn(),\n      getActHandler: () => handler,\n      getDefaultLlmClient: () => defaultClient,\n      domSettleTimeoutMs: undefined,\n    });\n\n    const context: ActCacheContext = {\n      instruction: \"click button\",\n      cacheKey: \"abc\",\n      pageUrl: \"https://example.com\",\n      variableKeys: [],\n      variables: undefined,\n    };\n\n    const result = await cache.tryReplay(\n      context,\n      {} as Page,\n      undefined,\n      overrideClient,\n    );\n\n    expect(result?.success).toBe(true);\n    expect(handler.takeDeterministicAction).toHaveBeenCalledTimes(1);\n    const call = vi.mocked(handler.takeDeterministicAction).mock.calls[0];\n    expect(call?.[3]).toBe(overrideClient);\n  });\n\n  it(\"AgentCache uses provided override client during replay\", async () => {\n    const action: Action = {\n      selector: \"xpath=/html/body/input\",\n      description: \"type email\",\n      method: \"type\",\n      arguments: [\"test@example.com\"],\n    };\n\n    const agentStep: AgentReplayActStep = {\n      type: \"act\",\n      instruction: \"type email\",\n      actions: [action],\n    };\n\n    const entry: CachedAgentEntry = {\n      version: 1,\n      instruction: \"fill form\",\n      startUrl: \"https://example.com\",\n      options: {},\n      configSignature: \"sig\",\n      steps: [agentStep],\n      result: { success: true, actions: [] } as AgentResult,\n      timestamp: new Date().toISOString(),\n    };\n\n    const storage = {\n      enabled: true,\n      readJson: vi.fn().mockImplementation(async () => ({ value: entry })),\n      writeJson: vi.fn().mockResolvedValue({}),\n      directory: \"/tmp/cache\",\n    } as unknown as CacheStorage;\n\n    const handler = {\n      takeDeterministicAction: vi.fn().mockResolvedValue({\n        success: true,\n        message: \"ok\",\n        actionDescription: \"type email\",\n        actions: [action],\n      }),\n    } as unknown as ActHandler;\n\n    const fakePage = {} as Page;\n    const ctx = {\n      awaitActivePage: vi.fn().mockResolvedValue(fakePage),\n    } as unknown as V3Context;\n\n    const defaultClient = { id: \"default-agent\" } as unknown as LLMClient;\n    const overrideClient = { id: \"override-agent\" } as unknown as LLMClient;\n\n    const cache = new AgentCache({\n      storage,\n      logger: vi.fn(),\n      getActHandler: () => handler,\n      getContext: () => ctx,\n      getDefaultLlmClient: () => defaultClient,\n      getBaseModelName: () => \"openai/gpt-4.1-mini\" as AvailableModel,\n      getSystemPrompt: () => undefined,\n      domSettleTimeoutMs: undefined,\n      act: vi.fn(),\n    });\n\n    const context: AgentCacheContext = {\n      instruction: \"fill form\",\n      startUrl: \"https://example.com\",\n      options: {},\n      configSignature: \"sig\",\n      cacheKey: \"agent-key\",\n      variableKeys: [],\n    };\n\n    const result = await cache.tryReplay(context, overrideClient);\n\n    expect(result?.success).toBe(true);\n    expect(handler.takeDeterministicAction).toHaveBeenCalledTimes(1);\n    const call = vi.mocked(handler.takeDeterministicAction).mock.calls[0];\n    expect(call?.[3]).toBe(overrideClient);\n  });\n\n  it(\"AgentCache replays non-act steps without requiring an override client\", async () => {\n    const gotoEntry: CachedAgentEntry = {\n      version: 1,\n      instruction: \"navigate home\",\n      startUrl: \"https://example.com/source\",\n      options: {},\n      configSignature: \"sig\",\n      steps: [\n        {\n          type: \"goto\",\n          url: \"https://example.com/target\",\n          waitUntil: \"load\",\n        },\n      ],\n      result: { success: true, actions: [] } as AgentResult,\n      timestamp: new Date().toISOString(),\n    };\n\n    const storage = {\n      enabled: true,\n      readJson: vi.fn().mockResolvedValue({ value: gotoEntry }),\n      writeJson: vi.fn().mockResolvedValue({}),\n      directory: \"/tmp/cache\",\n    } as unknown as CacheStorage;\n\n    const handler = {\n      takeDeterministicAction: vi.fn(),\n    } as unknown as ActHandler;\n\n    const fakePage = { goto: vi.fn() } as unknown as Page;\n    const ctx = {\n      awaitActivePage: vi.fn().mockResolvedValue(fakePage),\n    } as unknown as V3Context;\n\n    const cache = new AgentCache({\n      storage,\n      logger: vi.fn(),\n      getActHandler: () => handler,\n      getContext: () => ctx,\n      getDefaultLlmClient: () => ({ id: \"default\" }) as unknown as LLMClient,\n      getBaseModelName: () => \"openai/gpt-4.1-mini\" as AvailableModel,\n      getSystemPrompt: () => undefined,\n      domSettleTimeoutMs: undefined,\n      act: vi.fn(),\n    });\n\n    const context: AgentCacheContext = {\n      instruction: \"navigate home\",\n      startUrl: \"https://example.com/source\",\n      options: {},\n      configSignature: \"sig\",\n      cacheKey: \"agent-goto\",\n      variableKeys: [],\n    };\n\n    const result = await cache.tryReplay(context);\n\n    expect(result?.success).toBe(true);\n    expect(handler.takeDeterministicAction).not.toHaveBeenCalled();\n    expect(fakePage.goto).toHaveBeenCalledWith(\"https://example.com/target\", {\n      waitUntil: \"load\",\n    });\n  });\n});\n"]}

package/dist/esm/tests/unit/cdp-connection-close.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/cdp-connection-close.test.js

@@ -0,0 +1,72 @@
+import { describe, it, expect, afterEach } from "vitest";
+import { WebSocketServer } from "ws";
+import { CdpConnection } from "../../lib/v3/understudy/cdp.js";
+/**
+ * Races a promise against a timeout. Returns "resolved" if the promise
+ * settles before the deadline, or "timeout" if it doesn't.
+ */
+// TODO: dedupe this with the implementation in testUtils.ts after we unify the test directories
+function raceTimeout(promise, ms) {
+    let timer;
+    const timeout = new Promise((resolve) => {
+        timer = setTimeout(() => resolve("timeout"), ms);
+    });
+    return Promise.race([promise, timeout]).finally(() => clearTimeout(timer));
+}
+/**
+ * Creates a local WebSocket server and connects a CdpConnection to it.
+ * Returns the connection plus a handle to the server-side socket.
+ */
+async function createPair() {
+    const wss = new WebSocketServer({ port: 0 });
+    const port = wss.address().port;
+    const serverSocketPromise = new Promise((resolve) => {
+        wss.once("connection", resolve);
+    });
+    const conn = await CdpConnection.connect(`ws://localhost:${port}`);
+    const serverSocket = await serverSocketPromise;
+    return { conn, serverSocket, wss };
+}
+describe("CdpConnection", () => {
+    let wss = null;
+    afterEach(async () => {
+        if (wss) {
+            await new Promise((resolve) => wss.close(() => resolve()));
+            wss = null;
+        }
+    });
+    describe("close() when WebSocket is already closed", () => {
+        it("resolves instead of hanging forever", async () => {
+            const pair = await createPair();
+            wss = pair.wss;
+            // Wait for the client-side close event to be fully processed.
+            const transportClosed = new Promise((resolve) => {
+                pair.conn.onTransportClosed(() => resolve());
+            });
+            // Simulate the hosted API terminating the Browserbase session:
+            // the server closes the WebSocket from its side.
+            pair.serverSocket.close();
+            await transportClosed;
+            // conn.close() on an already-CLOSED WebSocket must resolve.
+            // Without the fix it awaits a "close" event that already fired → hangs.
+            const result = await raceTimeout(pair.conn.close().then(() => "resolved"), 3_000);
+            expect(result).toBe("resolved");
+        });
+    });
+    describe("inflight CDP calls on unexpected close", () => {
+        it("rejects pending calls instead of hanging forever", async () => {
+            const pair = await createPair();
+            wss = pair.wss;
+            // Send a CDP command; the mock server will never reply.
+            const pending = pair.conn.send("Runtime.evaluate", {
+                expression: "1+1",
+            });
+            // Server terminates the connection while the call is inflight.
+            pair.serverSocket.close();
+            // The pending promise must reject, not hang.
+            const result = await raceTimeout(pending.then(() => "resolved").catch(() => "rejected"), 3_000);
+            expect(result).toBe("rejected");
+        });
+    });
+});
+//# sourceMappingURL=cdp-connection-close.test.js.map

package/dist/esm/tests/unit/cdp-connection-close.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cdp-connection-close.test.js","sourceRoot":"","sources":["../../../../tests/unit/cdp-connection-close.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,eAAe,EAAqC,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAE/D;;;GAGG;AACH,gGAAgG;AAChG,SAAS,WAAW,CAClB,OAAmB,EACnB,EAAU;IAEV,IAAI,KAAoC,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAY,CAAC,OAAO,EAAE,EAAE;QACjD,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7E,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,UAAU;IAKvB,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,IAAI,GAAI,GAAG,CAAC,OAAO,EAAuB,CAAC,IAAI,CAAC;IAEtD,MAAM,mBAAmB,GAAG,IAAI,OAAO,CAAkB,CAAC,OAAO,EAAE,EAAE;QACnE,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,MAAM,IAAI,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC;IACnE,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC;IAE/C,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;AACrC,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,GAAG,GAA2B,IAAI,CAAC;IAEvC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,GAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAClE,GAAG,GAAG,IAAI,CAAC;QACb,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACxD,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,IAAI,GAAG,MAAM,UAAU,EAAE,CAAC;YAChC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;YAEf,8DAA8D;YAC9D,MAAM,eAAe,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;gBACpD,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,+DAA+D;YAC/D,iDAAiD;YACjD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC1B,MAAM,eAAe,CAAC;YAEtB,4DAA4D;YAC5D,wEAAwE;YACxE,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,EACxC,KAAK,CACN,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACtD,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,IAAI,GAAG,MAAM,UAAU,EAAE,CAAC;YAChC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;YAEf,wDAAwD;YACxD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE;gBACjD,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YAEH,+DAA+D;YAC/D,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAE1B,6CAA6C;YAC7C,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,EACtD,KAAK,CACN,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, it, expect, afterEach } from \"vitest\";\nimport { WebSocketServer, type WebSocket as ServerWebSocket } from \"ws\";\nimport { CdpConnection } from \"../../lib/v3/understudy/cdp.js\";\n\n/**\n * Races a promise against a timeout. Returns \"resolved\" if the promise\n * settles before the deadline, or \"timeout\" if it doesn't.\n */\n// TODO: dedupe this with the implementation in testUtils.ts after we unify the test directories\nfunction raceTimeout<T>(\n  promise: Promise<T>,\n  ms: number,\n): Promise<T | \"timeout\"> {\n  let timer: ReturnType<typeof setTimeout>;\n  const timeout = new Promise<\"timeout\">((resolve) => {\n    timer = setTimeout(() => resolve(\"timeout\"), ms);\n  });\n  return Promise.race([promise, timeout]).finally(() => clearTimeout(timer));\n}\n\n/**\n * Creates a local WebSocket server and connects a CdpConnection to it.\n * Returns the connection plus a handle to the server-side socket.\n */\nasync function createPair(): Promise<{\n  conn: CdpConnection;\n  serverSocket: ServerWebSocket;\n  wss: WebSocketServer;\n}> {\n  const wss = new WebSocketServer({ port: 0 });\n  const port = (wss.address() as { port: number }).port;\n\n  const serverSocketPromise = new Promise<ServerWebSocket>((resolve) => {\n    wss.once(\"connection\", resolve);\n  });\n\n  const conn = await CdpConnection.connect(`ws://localhost:${port}`);\n  const serverSocket = await serverSocketPromise;\n\n  return { conn, serverSocket, wss };\n}\n\ndescribe(\"CdpConnection\", () => {\n  let wss: WebSocketServer | null = null;\n\n  afterEach(async () => {\n    if (wss) {\n      await new Promise<void>((resolve) => wss!.close(() => resolve()));\n      wss = null;\n    }\n  });\n\n  describe(\"close() when WebSocket is already closed\", () => {\n    it(\"resolves instead of hanging forever\", async () => {\n      const pair = await createPair();\n      wss = pair.wss;\n\n      // Wait for the client-side close event to be fully processed.\n      const transportClosed = new Promise<void>((resolve) => {\n        pair.conn.onTransportClosed(() => resolve());\n      });\n\n      // Simulate the hosted API terminating the Browserbase session:\n      // the server closes the WebSocket from its side.\n      pair.serverSocket.close();\n      await transportClosed;\n\n      // conn.close() on an already-CLOSED WebSocket must resolve.\n      // Without the fix it awaits a \"close\" event that already fired → hangs.\n      const result = await raceTimeout(\n        pair.conn.close().then(() => \"resolved\"),\n        3_000,\n      );\n\n      expect(result).toBe(\"resolved\");\n    });\n  });\n\n  describe(\"inflight CDP calls on unexpected close\", () => {\n    it(\"rejects pending calls instead of hanging forever\", async () => {\n      const pair = await createPair();\n      wss = pair.wss;\n\n      // Send a CDP command; the mock server will never reply.\n      const pending = pair.conn.send(\"Runtime.evaluate\", {\n        expression: \"1+1\",\n      });\n\n      // Server terminates the connection while the call is inflight.\n      pair.serverSocket.close();\n\n      // The pending promise must reject, not hang.\n      const result = await raceTimeout(\n        pending.then(() => \"resolved\").catch(() => \"rejected\"),\n        3_000,\n      );\n\n      expect(result).toBe(\"rejected\");\n    });\n  });\n});\n"]}

package/dist/esm/tests/unit/context-extra-http-headers.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/context-extra-http-headers.test.js

@@ -0,0 +1,56 @@
+import { describe, expect, it } from "vitest";
+import { V3Context } from "../../lib/v3/understudy/context.js";
+import { MockCDPSession } from "./helpers/mockCDPSession.js";
+import { StagehandSetExtraHTTPHeadersError } from "../../lib/v3/types/public/sdkErrors.js";
+const makeContext = (sessions) => {
+    const sessionsById = new Map(sessions.map((session) => [session.id, session]));
+    return {
+        _sessionInit: new Set(sessions.map((session) => session.id)),
+        conn: {
+            getSession: (id) => sessionsById.get(id),
+        },
+        extraHttpHeaders: null,
+    };
+};
+describe("V3Context.setExtraHTTPHeaders", () => {
+    const setExtraHTTPHeaders = V3Context.prototype.setExtraHTTPHeaders;
+    it("sends headers to all sessions", async () => {
+        const sessionA = new MockCDPSession({}, "session-a");
+        const sessionB = new MockCDPSession({}, "session-b");
+        const ctx = makeContext([sessionA, sessionB]);
+        await setExtraHTTPHeaders.call(ctx, {
+            "x-stagehand-test": "yes",
+        });
+        for (const session of [sessionA, sessionB]) {
+            expect(session.callsFor("Network.enable").length).toBe(1);
+            expect(session.callsFor("Network.setExtraHTTPHeaders")[0]?.params).toEqual({
+                headers: { "x-stagehand-test": "yes" },
+            });
+        }
+    });
+    it("throws a custom error with session failure details", async () => {
+        const sessionA = new MockCDPSession({
+            "Network.setExtraHTTPHeaders": () => {
+                throw new Error("boom");
+            },
+        }, "session-a");
+        const sessionB = new MockCDPSession({}, "session-b");
+        const ctx = makeContext([sessionA, sessionB]);
+        const promise = setExtraHTTPHeaders.call(ctx, {
+            "x-stagehand-test": "yes",
+        });
+        await expect(promise).rejects.toBeInstanceOf(StagehandSetExtraHTTPHeadersError);
+        try {
+            await promise;
+        }
+        catch (error) {
+            const err = error;
+            expect(err.failures).toHaveLength(1);
+            expect(err.failures[0]).toContain("session=session-a");
+            expect(err.failures[0]).toContain("boom");
+        }
+        expect(sessionA.callsFor("Network.setExtraHTTPHeaders").length).toBe(1);
+        expect(sessionB.callsFor("Network.setExtraHTTPHeaders").length).toBe(1);
+    });
+});
+//# sourceMappingURL=context-extra-http-headers.test.js.map

package/dist/esm/tests/unit/context-extra-http-headers.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-extra-http-headers.test.js","sourceRoot":"","sources":["../../../../tests/unit/context-extra-http-headers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,iCAAiC,EAAE,MAAM,wCAAwC,CAAC;AAU3F,MAAM,WAAW,GAAG,CAAC,QAA0B,EAAe,EAAE;IAC9D,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CACjD,CAAC;IACF,OAAO;QACL,YAAY,EAAE,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5D,IAAI,EAAE;YACJ,UAAU,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;SACjD;QACD,gBAAgB,EAAE,IAAI;KACvB,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,MAAM,mBAAmB,GAAG,SAAS,CAAC,SAAS,CAAC,mBAG9B,CAAC;IAEnB,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QAE9C,MAAM,mBAAmB,CAAC,IAAI,CAAC,GAAG,EAAE;YAClC,kBAAkB,EAAE,KAAK;SAC1B,CAAC,CAAC;QAEH,KAAK,MAAM,OAAO,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1D,MAAM,CACJ,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAC3D,CAAC,OAAO,CAAC;gBACR,OAAO,EAAE,EAAE,kBAAkB,EAAE,KAAK,EAAE;aACvC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,IAAI,cAAc,CACjC;YACE,6BAA6B,EAAE,GAAG,EAAE;gBAClC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;YAC1B,CAAC;SACF,EACD,WAAW,CACZ,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,WAAW,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,CAAC,GAAG,EAAE;YAC5C,kBAAkB,EAAE,KAAK;SAC1B,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,cAAc,CAC1C,iCAAiC,CAClC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAA0C,CAAC;YACvD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YACvD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { V3Context } from \"../../lib/v3/understudy/context.js\";\nimport { MockCDPSession } from \"./helpers/mockCDPSession.js\";\nimport { StagehandSetExtraHTTPHeadersError } from \"../../lib/v3/types/public/sdkErrors.js\";\n\ntype ContextStub = {\n  _sessionInit: Set<string>;\n  conn: {\n    getSession: (id: string) => MockCDPSession | undefined;\n  };\n  extraHttpHeaders: Record<string, string> | null;\n};\n\nconst makeContext = (sessions: MockCDPSession[]): ContextStub => {\n  const sessionsById = new Map(\n    sessions.map((session) => [session.id, session]),\n  );\n  return {\n    _sessionInit: new Set(sessions.map((session) => session.id)),\n    conn: {\n      getSession: (id: string) => sessionsById.get(id),\n    },\n    extraHttpHeaders: null,\n  };\n};\n\ndescribe(\"V3Context.setExtraHTTPHeaders\", () => {\n  const setExtraHTTPHeaders = V3Context.prototype.setExtraHTTPHeaders as (\n    this: ContextStub,\n    headers: Record<string, string>,\n  ) => Promise<void>;\n\n  it(\"sends headers to all sessions\", async () => {\n    const sessionA = new MockCDPSession({}, \"session-a\");\n    const sessionB = new MockCDPSession({}, \"session-b\");\n    const ctx = makeContext([sessionA, sessionB]);\n\n    await setExtraHTTPHeaders.call(ctx, {\n      \"x-stagehand-test\": \"yes\",\n    });\n\n    for (const session of [sessionA, sessionB]) {\n      expect(session.callsFor(\"Network.enable\").length).toBe(1);\n      expect(\n        session.callsFor(\"Network.setExtraHTTPHeaders\")[0]?.params,\n      ).toEqual({\n        headers: { \"x-stagehand-test\": \"yes\" },\n      });\n    }\n  });\n\n  it(\"throws a custom error with session failure details\", async () => {\n    const sessionA = new MockCDPSession(\n      {\n        \"Network.setExtraHTTPHeaders\": () => {\n          throw new Error(\"boom\");\n        },\n      },\n      \"session-a\",\n    );\n    const sessionB = new MockCDPSession({}, \"session-b\");\n    const ctx = makeContext([sessionA, sessionB]);\n\n    const promise = setExtraHTTPHeaders.call(ctx, {\n      \"x-stagehand-test\": \"yes\",\n    });\n\n    await expect(promise).rejects.toBeInstanceOf(\n      StagehandSetExtraHTTPHeadersError,\n    );\n\n    try {\n      await promise;\n    } catch (error) {\n      const err = error as StagehandSetExtraHTTPHeadersError;\n      expect(err.failures).toHaveLength(1);\n      expect(err.failures[0]).toContain(\"session=session-a\");\n      expect(err.failures[0]).toContain(\"boom\");\n    }\n\n    expect(sessionA.callsFor(\"Network.setExtraHTTPHeaders\").length).toBe(1);\n    expect(sessionB.callsFor(\"Network.setExtraHTTPHeaders\").length).toBe(1);\n  });\n});\n"]}

package/dist/esm/tests/unit/cookies.test.d.ts

@@ -0,0 +1 @@
+export {};