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

package/dist/esm/tests/unit/snapshot-cbor.test.js

@@ -0,0 +1,202 @@
+import { describe, expect, it } from "vitest";
+import { captureHybridSnapshot } from "../../lib/v3/understudy/a11y/snapshot/index.js";
+import { MockCDPSession } from "./helpers/mockCDPSession.js";
+import { StagehandDomProcessError } from "../../lib/v3/types/public/sdkErrors.js";
+function createFakePage(session) {
+    const frameTree = {
+        frame: {
+            id: "root",
+            loaderId: "root-loader",
+            url: "http://fake",
+            domainAndRegistry: "fake",
+            securityOrigin: "http://fake",
+            mimeType: "text/html",
+            secureContextType: "Secure",
+            crossOriginIsolatedContextType: "NotIsolated",
+            gatedAPIFeatures: [],
+        },
+        childFrames: [],
+    };
+    return {
+        mainFrameId: () => "root",
+        asProtocolFrameTree: () => frameTree,
+        listAllFrameIds: () => ["root"],
+        getSessionForFrame: () => session,
+        getOrdinal: () => 0,
+    };
+}
+function completeDomTree() {
+    return {
+        nodeId: 1,
+        backendNodeId: 1,
+        nodeType: 9,
+        nodeName: "#document",
+        childNodeCount: 1,
+        children: [
+            {
+                nodeId: 2,
+                backendNodeId: 2,
+                nodeType: 1,
+                nodeName: "HTML",
+                childNodeCount: 1,
+                children: [
+                    {
+                        nodeId: 3,
+                        backendNodeId: 3,
+                        nodeType: 1,
+                        nodeName: "BODY",
+                        childNodeCount: 1,
+                        children: [
+                            {
+                                nodeId: 4,
+                                backendNodeId: 4,
+                                nodeType: 1,
+                                nodeName: "DIV",
+                                childNodeCount: 0,
+                                children: [],
+                            },
+                        ],
+                    },
+                ],
+            },
+        ],
+    };
+}
+function truncatedDomTree() {
+    return {
+        nodeId: 1,
+        backendNodeId: 1,
+        nodeType: 9,
+        nodeName: "#document",
+        childNodeCount: 1,
+        children: [
+            {
+                nodeId: 2,
+                backendNodeId: 2,
+                nodeType: 1,
+                nodeName: "HTML",
+                childNodeCount: 1,
+                children: [],
+            },
+        ],
+    };
+}
+function htmlWithChildren() {
+    return {
+        nodeId: 2,
+        backendNodeId: 2,
+        nodeType: 1,
+        nodeName: "HTML",
+        childNodeCount: 1,
+        children: [
+            {
+                nodeId: 3,
+                backendNodeId: 3,
+                nodeType: 1,
+                nodeName: "BODY",
+                childNodeCount: 1,
+                children: [
+                    {
+                        nodeId: 4,
+                        backendNodeId: 4,
+                        nodeType: 1,
+                        nodeName: "DIV",
+                        childNodeCount: 0,
+                        children: [],
+                    },
+                ],
+            },
+        ],
+    };
+}
+function simpleAxNodes() {
+    const stringType = "string";
+    return [
+        {
+            nodeId: "1",
+            role: { type: stringType, value: "RootWebArea" },
+            backendDOMNodeId: 2,
+            childIds: ["2"],
+            ignored: false,
+        },
+        {
+            nodeId: "2",
+            role: { type: stringType, value: "generic" },
+            name: { type: stringType, value: "Content" },
+            backendDOMNodeId: 4,
+            parentId: "1",
+            childIds: [],
+            ignored: false,
+        },
+    ];
+}
+const baseHandlers = {
+    "DOM.enable": async () => ({}),
+    "Runtime.enable": async () => ({}),
+    "Accessibility.enable": async () => ({}),
+    "Accessibility.getFullAXTree": async () => ({ nodes: simpleAxNodes() }),
+};
+function makeCborError() {
+    return new Error("CBOR: stack limit exceeded");
+}
+describe("captureHybridSnapshot CBOR fallbacks", () => {
+    it("retries DOM.getDocument with reduced depths before succeeding", async () => {
+        let domCalls = 0;
+        const session = new MockCDPSession({
+            ...baseHandlers,
+            "DOM.getDocument": async (params) => {
+                domCalls += 1;
+                if (domCalls === 1)
+                    throw makeCborError();
+                expect(params?.depth).toBe(256);
+                return { root: completeDomTree() };
+            },
+        });
+        const page = createFakePage(session);
+        const snapshot = await captureHybridSnapshot(page);
+        expect(snapshot.combinedTree).toContain("html");
+        const depths = session
+            .callsFor("DOM.getDocument")
+            .map((c) => c.params?.depth);
+        expect(depths).toEqual([-1, 256]);
+    });
+    it("throws StagehandDomProcessError after all DOM.getDocument attempts fail", async () => {
+        const session = new MockCDPSession({
+            ...baseHandlers,
+            "DOM.getDocument": async () => {
+                throw makeCborError();
+            },
+        });
+        const page = createFakePage(session);
+        await expect(captureHybridSnapshot(page)).rejects.toThrow(StagehandDomProcessError);
+    });
+    it("hydrates truncated nodes by retrying DOM.describeNode depths", async () => {
+        let domAttempts = 0;
+        let describeAttempts = 0;
+        const session = new MockCDPSession({
+            ...baseHandlers,
+            "DOM.getDocument": async (params) => {
+                domAttempts += 1;
+                if (domAttempts === 1)
+                    throw makeCborError();
+                expect(params?.depth).toBe(256);
+                return { root: truncatedDomTree() };
+            },
+            "DOM.describeNode": async (params) => {
+                describeAttempts += 1;
+                if (describeAttempts === 1)
+                    throw makeCborError();
+                expect(params?.depth).toBe(64);
+                return { node: htmlWithChildren() };
+            },
+        });
+        const page = createFakePage(session);
+        const snapshot = await captureHybridSnapshot(page);
+        const describeDepths = session
+            .callsFor("DOM.describeNode")
+            .map((c) => c.params?.depth);
+        expect(describeDepths).toEqual([-1, 64]);
+        expect(snapshot.combinedXpathMap["0-4"]).toBe("/html[1]/body[1]/div[1]");
+    });
+});
+//# sourceMappingURL=snapshot-cbor.test.js.map

package/dist/esm/tests/unit/snapshot-cbor.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshot-cbor.test.js","sourceRoot":"","sources":["../../../../tests/unit/snapshot-cbor.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AACvF,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAE7D,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAKlF,SAAS,cAAc,CAAC,OAAuB;IAC7C,MAAM,SAAS,GAA4B;QACzC,KAAK,EAAE;YACL,EAAE,EAAE,MAA+B;YACnC,QAAQ,EAAE,aAA0C;YACpD,GAAG,EAAE,aAAa;YAClB,iBAAiB,EAAE,MAAM;YACzB,cAAc,EAAE,aAAa;YAC7B,QAAQ,EAAE,WAAW;YACrB,iBAAiB,EAAE,QAAQ;YAC3B,8BAA8B,EAAE,aAAa;YAC7C,gBAAgB,EAAE,EAAE;SACrB;QACD,WAAW,EAAE,EAAE;KAChB,CAAC;IAEF,OAAO;QACL,WAAW,EAAE,GAAG,EAAE,CAAC,MAAM;QACzB,mBAAmB,EAAE,GAAG,EAAE,CAAC,SAAS;QACpC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;QAC/B,kBAAkB,EAAE,GAAG,EAAE,CAAC,OAAO;QACjC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;KACD,CAAC;AACvB,CAAC;AAED,SAAS,eAAe;IACtB,OAAO;QACL,MAAM,EAAE,CAAC;QACT,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,WAAW;QACrB,cAAc,EAAE,CAAC;QACjB,QAAQ,EAAE;YACR;gBACE,MAAM,EAAE,CAAC;gBACT,aAAa,EAAE,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,QAAQ,EAAE,MAAM;gBAChB,cAAc,EAAE,CAAC;gBACjB,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,CAAC;wBACT,aAAa,EAAE,CAAC;wBAChB,QAAQ,EAAE,CAAC;wBACX,QAAQ,EAAE,MAAM;wBAChB,cAAc,EAAE,CAAC;wBACjB,QAAQ,EAAE;4BACR;gCACE,MAAM,EAAE,CAAC;gCACT,aAAa,EAAE,CAAC;gCAChB,QAAQ,EAAE,CAAC;gCACX,QAAQ,EAAE,KAAK;gCACf,cAAc,EAAE,CAAC;gCACjB,QAAQ,EAAE,EAAE;6BACb;yBACF;qBACF;iBACF;aACF;SACF;KACmB,CAAC;AACzB,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO;QACL,MAAM,EAAE,CAAC;QACT,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,WAAW;QACrB,cAAc,EAAE,CAAC;QACjB,QAAQ,EAAE;YACR;gBACE,MAAM,EAAE,CAAC;gBACT,aAAa,EAAE,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,QAAQ,EAAE,MAAM;gBAChB,cAAc,EAAE,CAAC;gBACjB,QAAQ,EAAE,EAAE;aACb;SACF;KACmB,CAAC;AACzB,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO;QACL,MAAM,EAAE,CAAC;QACT,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,MAAM;QAChB,cAAc,EAAE,CAAC;QACjB,QAAQ,EAAE;YACR;gBACE,MAAM,EAAE,CAAC;gBACT,aAAa,EAAE,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,QAAQ,EAAE,MAAM;gBAChB,cAAc,EAAE,CAAC;gBACjB,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,CAAC;wBACT,aAAa,EAAE,CAAC;wBAChB,QAAQ,EAAE,CAAC;wBACX,QAAQ,EAAE,KAAK;wBACf,cAAc,EAAE,CAAC;wBACjB,QAAQ,EAAE,EAAE;qBACb;iBACF;aACF;SACF;KACmB,CAAC;AACzB,CAAC;AAED,SAAS,aAAa;IACpB,MAAM,UAAU,GAAuC,QAAQ,CAAC;IAChE,OAAO;QACL;YACE,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,aAAa,EAAE;YAChD,gBAAgB,EAAE,CAAC;YACnB,QAAQ,EAAE,CAAC,GAAG,CAAC;YACf,OAAO,EAAE,KAAK;SACf;QACD;YACE,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE;YAC5C,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE;YAC5C,gBAAgB,EAAE,CAAC;YACnB,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,EAAc;YACxB,OAAO,EAAE,KAAK;SACf;KACF,CAAC;AACJ,CAAC;AAED,MAAM,YAAY,GAA4B;IAC5C,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAC9B,gBAAgB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAClC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IACxC,6BAA6B,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,CAAC;CACxE,CAAC;AAEF,SAAS,aAAa;IACpB,OAAO,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;AACjD,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,GAAG,YAAY;YACf,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;gBAClC,QAAQ,IAAI,CAAC,CAAC;gBACd,IAAI,QAAQ,KAAK,CAAC;oBAAE,MAAM,aAAa,EAAE,CAAC;gBAC1C,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChC,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,CAAC;YACrC,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,OAAO;aACnB,QAAQ,CAAC,iBAAiB,CAAC;aAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,GAAG,YAAY;YACf,iBAAiB,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,aAAa,EAAE,CAAC;YACxB,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACvD,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,gBAAgB,GAAG,CAAC,CAAC;QAEzB,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,GAAG,YAAY;YACf,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;gBAClC,WAAW,IAAI,CAAC,CAAC;gBACjB,IAAI,WAAW,KAAK,CAAC;oBAAE,MAAM,aAAa,EAAE,CAAC;gBAC7C,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAChC,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,CAAC;YACtC,CAAC;YACD,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;gBACnC,gBAAgB,IAAI,CAAC,CAAC;gBACtB,IAAI,gBAAgB,KAAK,CAAC;oBAAE,MAAM,aAAa,EAAE,CAAC;gBAClD,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC/B,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,CAAC;YACtC,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,cAAc,GAAG,OAAO;aAC3B,QAAQ,CAAC,kBAAkB,CAAC;aAC5B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport type { Protocol } from \"devtools-protocol\";\n\nimport { captureHybridSnapshot } from \"../../lib/v3/understudy/a11y/snapshot/index.js\";\nimport { MockCDPSession } from \"./helpers/mockCDPSession.js\";\nimport type { Page } from \"../../lib/v3/understudy/page.js\";\nimport { StagehandDomProcessError } from \"../../lib/v3/types/public/sdkErrors.js\";\nimport { CDPSessionLike } from \"../../lib/v3/understudy/cdp.js\";\n\ntype Handler = (params?: Record<string, unknown>) => Promise<unknown> | unknown;\n\nfunction createFakePage(session: CDPSessionLike): Page {\n  const frameTree: Protocol.Page.FrameTree = {\n    frame: {\n      id: \"root\" as Protocol.Page.FrameId,\n      loaderId: \"root-loader\" as Protocol.Network.LoaderId,\n      url: \"http://fake\",\n      domainAndRegistry: \"fake\",\n      securityOrigin: \"http://fake\",\n      mimeType: \"text/html\",\n      secureContextType: \"Secure\",\n      crossOriginIsolatedContextType: \"NotIsolated\",\n      gatedAPIFeatures: [],\n    },\n    childFrames: [],\n  };\n\n  return {\n    mainFrameId: () => \"root\",\n    asProtocolFrameTree: () => frameTree,\n    listAllFrameIds: () => [\"root\"],\n    getSessionForFrame: () => session,\n    getOrdinal: () => 0,\n  } as unknown as Page;\n}\n\nfunction completeDomTree(): Protocol.DOM.Node {\n  return {\n    nodeId: 1,\n    backendNodeId: 1,\n    nodeType: 9,\n    nodeName: \"#document\",\n    childNodeCount: 1,\n    children: [\n      {\n        nodeId: 2,\n        backendNodeId: 2,\n        nodeType: 1,\n        nodeName: \"HTML\",\n        childNodeCount: 1,\n        children: [\n          {\n            nodeId: 3,\n            backendNodeId: 3,\n            nodeType: 1,\n            nodeName: \"BODY\",\n            childNodeCount: 1,\n            children: [\n              {\n                nodeId: 4,\n                backendNodeId: 4,\n                nodeType: 1,\n                nodeName: \"DIV\",\n                childNodeCount: 0,\n                children: [],\n              },\n            ],\n          },\n        ],\n      },\n    ],\n  } as Protocol.DOM.Node;\n}\n\nfunction truncatedDomTree(): Protocol.DOM.Node {\n  return {\n    nodeId: 1,\n    backendNodeId: 1,\n    nodeType: 9,\n    nodeName: \"#document\",\n    childNodeCount: 1,\n    children: [\n      {\n        nodeId: 2,\n        backendNodeId: 2,\n        nodeType: 1,\n        nodeName: \"HTML\",\n        childNodeCount: 1,\n        children: [],\n      },\n    ],\n  } as Protocol.DOM.Node;\n}\n\nfunction htmlWithChildren(): Protocol.DOM.Node {\n  return {\n    nodeId: 2,\n    backendNodeId: 2,\n    nodeType: 1,\n    nodeName: \"HTML\",\n    childNodeCount: 1,\n    children: [\n      {\n        nodeId: 3,\n        backendNodeId: 3,\n        nodeType: 1,\n        nodeName: \"BODY\",\n        childNodeCount: 1,\n        children: [\n          {\n            nodeId: 4,\n            backendNodeId: 4,\n            nodeType: 1,\n            nodeName: \"DIV\",\n            childNodeCount: 0,\n            children: [],\n          },\n        ],\n      },\n    ],\n  } as Protocol.DOM.Node;\n}\n\nfunction simpleAxNodes(): Protocol.Accessibility.AXNode[] {\n  const stringType: Protocol.Accessibility.AXValueType = \"string\";\n  return [\n    {\n      nodeId: \"1\",\n      role: { type: stringType, value: \"RootWebArea\" },\n      backendDOMNodeId: 2,\n      childIds: [\"2\"],\n      ignored: false,\n    },\n    {\n      nodeId: \"2\",\n      role: { type: stringType, value: \"generic\" },\n      name: { type: stringType, value: \"Content\" },\n      backendDOMNodeId: 4,\n      parentId: \"1\",\n      childIds: [] as string[],\n      ignored: false,\n    },\n  ];\n}\n\nconst baseHandlers: Record<string, Handler> = {\n  \"DOM.enable\": async () => ({}),\n  \"Runtime.enable\": async () => ({}),\n  \"Accessibility.enable\": async () => ({}),\n  \"Accessibility.getFullAXTree\": async () => ({ nodes: simpleAxNodes() }),\n};\n\nfunction makeCborError(): Error {\n  return new Error(\"CBOR: stack limit exceeded\");\n}\n\ndescribe(\"captureHybridSnapshot CBOR fallbacks\", () => {\n  it(\"retries DOM.getDocument with reduced depths before succeeding\", async () => {\n    let domCalls = 0;\n    const session = new MockCDPSession({\n      ...baseHandlers,\n      \"DOM.getDocument\": async (params) => {\n        domCalls += 1;\n        if (domCalls === 1) throw makeCborError();\n        expect(params?.depth).toBe(256);\n        return { root: completeDomTree() };\n      },\n    });\n\n    const page = createFakePage(session);\n    const snapshot = await captureHybridSnapshot(page);\n\n    expect(snapshot.combinedTree).toContain(\"html\");\n    const depths = session\n      .callsFor(\"DOM.getDocument\")\n      .map((c) => c.params?.depth);\n    expect(depths).toEqual([-1, 256]);\n  });\n\n  it(\"throws StagehandDomProcessError after all DOM.getDocument attempts fail\", async () => {\n    const session = new MockCDPSession({\n      ...baseHandlers,\n      \"DOM.getDocument\": async () => {\n        throw makeCborError();\n      },\n    });\n\n    const page = createFakePage(session);\n    await expect(captureHybridSnapshot(page)).rejects.toThrow(\n      StagehandDomProcessError,\n    );\n  });\n\n  it(\"hydrates truncated nodes by retrying DOM.describeNode depths\", async () => {\n    let domAttempts = 0;\n    let describeAttempts = 0;\n\n    const session = new MockCDPSession({\n      ...baseHandlers,\n      \"DOM.getDocument\": async (params) => {\n        domAttempts += 1;\n        if (domAttempts === 1) throw makeCborError();\n        expect(params?.depth).toBe(256);\n        return { root: truncatedDomTree() };\n      },\n      \"DOM.describeNode\": async (params) => {\n        describeAttempts += 1;\n        if (describeAttempts === 1) throw makeCborError();\n        expect(params?.depth).toBe(64);\n        return { node: htmlWithChildren() };\n      },\n    });\n\n    const page = createFakePage(session);\n    const snapshot = await captureHybridSnapshot(page);\n\n    const describeDepths = session\n      .callsFor(\"DOM.describeNode\")\n      .map((c) => c.params?.depth);\n    expect(describeDepths).toEqual([-1, 64]);\n    expect(snapshot.combinedXpathMap[\"0-4\"]).toBe(\"/html[1]/body[1]/div[1]\");\n  });\n});\n"]}

package/dist/esm/tests/unit/snapshot-dom-session-builders.test.d.ts

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

package/dist/esm/tests/unit/snapshot-dom-session-builders.test.js

@@ -0,0 +1,218 @@
+import { describe, expect, it } from "vitest";
+import { buildSessionDomIndex, domMapsForSession, getDomTreeWithFallback, hydrateDomTree, } from "../../lib/v3/understudy/a11y/snapshot/domTree.js";
+import { StagehandDomProcessError } from "../../lib/v3/types/public/sdkErrors.js";
+import { MockCDPSession } from "./helpers/mockCDPSession.js";
+let nextNodeId = 1;
+const makeDomNode = (overrides = {}) => {
+    const nodeId = overrides.nodeId ?? nextNodeId++;
+    const backendNodeId = overrides.backendNodeId ?? nextNodeId++;
+    const nodeName = overrides.nodeName ?? "DIV";
+    const nodeType = overrides.nodeType ?? 1;
+    const children = overrides.children ?? [];
+    return {
+        nodeId,
+        backendNodeId,
+        nodeName,
+        nodeType,
+        localName: overrides.localName ?? nodeName.toLowerCase(),
+        nodeValue: overrides.nodeValue ?? "",
+        childNodeCount: overrides.childNodeCount ?? children.length,
+        children,
+        shadowRoots: overrides.shadowRoots,
+        contentDocument: overrides.contentDocument,
+        isScrollable: overrides.isScrollable,
+    };
+};
+const buildSampleDomTree = () => {
+    const iframeChild = makeDomNode({ nodeName: "P" });
+    const iframeBody = makeDomNode({
+        nodeName: "BODY",
+        children: [iframeChild],
+        isScrollable: true,
+    });
+    const iframeHtml = makeDomNode({ nodeName: "HTML", children: [iframeBody] });
+    const iframeDoc = makeDomNode({
+        nodeName: "#document",
+        nodeType: 9,
+        children: [iframeHtml],
+    });
+    const iframeElement = makeDomNode({
+        nodeName: "IFRAME",
+        contentDocument: iframeDoc,
+    });
+    const scrollDiv = makeDomNode({
+        nodeName: "DIV",
+        isScrollable: true,
+    });
+    const body = makeDomNode({
+        nodeName: "BODY",
+        children: [scrollDiv, iframeElement],
+    });
+    const html = makeDomNode({ nodeName: "HTML", children: [body] });
+    const root = makeDomNode({
+        nodeName: "#document",
+        nodeType: 9,
+        children: [html],
+    });
+    return {
+        root,
+        html,
+        body,
+        scrollDiv,
+        iframeElement,
+        iframeDoc,
+        iframeHtml,
+        iframeBody,
+        iframeChild,
+    };
+};
+describe("hydrateDomTree", () => {
+    it("expands truncated nodes by calling DOM.describeNode", async () => {
+        const child = makeDomNode({ nodeName: "DIV" });
+        const root = makeDomNode({
+            nodeName: "HTML",
+            childNodeCount: 1,
+            children: [],
+        });
+        const session = new MockCDPSession({
+            "DOM.describeNode": async () => ({
+                node: {
+                    ...root,
+                    children: [child],
+                    childNodeCount: 1,
+                },
+            }),
+        });
+        await hydrateDomTree(session, root, true);
+        expect(root.children).toEqual([child]);
+    });
+    it("retries describeNode when CBOR errors occur before succeeding", async () => {
+        const child = makeDomNode({ nodeName: "DIV" });
+        const root = makeDomNode({
+            nodeName: "HTML",
+            childNodeCount: 1,
+            children: [],
+        });
+        let attempts = 0;
+        const session = new MockCDPSession({
+            "DOM.describeNode": async () => {
+                attempts++;
+                if (attempts === 1)
+                    throw new Error("CBOR: stack limit exceeded");
+                return { node: { ...root, children: [child], childNodeCount: 1 } };
+            },
+        });
+        await hydrateDomTree(session, root, true);
+        expect(attempts).toBe(2);
+        expect(root.children).toEqual([child]);
+    });
+    it("throws StagehandDomProcessError after exhausting describeNode retries", async () => {
+        const root = makeDomNode({
+            nodeName: "HTML",
+            childNodeCount: 1,
+            children: [],
+        });
+        const session = new MockCDPSession({
+            "DOM.describeNode": async () => {
+                throw new Error("CBOR: stack limit exceeded");
+            },
+        });
+        await expect(hydrateDomTree(session, root, true)).rejects.toBeInstanceOf(StagehandDomProcessError);
+    });
+});
+describe("getDomTreeWithFallback", () => {
+    it("retries DOM.getDocument after CBOR errors and returns the hydrated root", async () => {
+        const root = makeDomNode({
+            nodeName: "#document",
+            nodeType: 9,
+            children: [],
+        });
+        const depths = [];
+        const session = new MockCDPSession({
+            "DOM.getDocument": async (params) => {
+                const depth = (params?.depth ?? 0);
+                depths.push(depth);
+                if (depth === -1)
+                    throw new Error("CBOR: stack limit exceeded");
+                return { root };
+            },
+            "DOM.describeNode": async () => ({ node: root }),
+        });
+        const result = await getDomTreeWithFallback(session, true);
+        expect(result).toBe(root);
+        expect(depths).toEqual([-1, 256]);
+    });
+    it("propagates non-CBOR DOM.getDocument errors", async () => {
+        const session = new MockCDPSession({
+            "DOM.getDocument": async () => {
+                throw new Error("network fail");
+            },
+        });
+        await expect(getDomTreeWithFallback(session, false)).rejects.toThrow("network fail");
+    });
+    it("throws StagehandDomProcessError when all depth attempts hit CBOR limits", async () => {
+        const session = new MockCDPSession({
+            "DOM.getDocument": async () => {
+                throw new Error("CBOR: stack limit exceeded");
+            },
+        });
+        await expect(getDomTreeWithFallback(session, false)).rejects.toBeInstanceOf(StagehandDomProcessError);
+    });
+});
+describe("buildSessionDomIndex", () => {
+    it("collects absolute paths, scrollability, and content-document metadata", async () => {
+        const tree = buildSampleDomTree();
+        const session = new MockCDPSession({
+            "DOM.enable": async () => ({}),
+            "DOM.getDocument": async () => ({ root: tree.root }),
+            "DOM.describeNode": async () => ({ node: tree.root }),
+        });
+        const index = await buildSessionDomIndex(session, true);
+        expect(index.rootBackend).toBe(tree.root.backendNodeId);
+        expect(index.absByBe.get(tree.body.backendNodeId)).toBe("/html[1]/body[1]");
+        expect(index.absByBe.get(tree.scrollDiv.backendNodeId)).toBe("/html[1]/body[1]/div[1]");
+        expect(index.scrollByBe.get(tree.scrollDiv.backendNodeId)).toBe(true);
+        expect(index.docRootOf.get(tree.iframeHtml.backendNodeId)).toBe(tree.iframeDoc.backendNodeId);
+        expect(index.contentDocRootByIframe.get(tree.iframeElement.backendNodeId)).toBe(tree.iframeDoc.backendNodeId);
+    });
+});
+describe("domMapsForSession", () => {
+    it("derives frame-relative xpath/tag/scrollable maps for a frame's document root", async () => {
+        const tree = buildSampleDomTree();
+        const session = new MockCDPSession({
+            "DOM.enable": async () => ({}),
+            "DOM.getDocument": async () => ({ root: tree.root }),
+            "DOM.getFrameOwner": async () => ({
+                backendNodeId: tree.iframeElement.backendNodeId,
+            }),
+            "DOM.describeNode": async () => ({ node: tree.root }),
+        });
+        const encode = (frameId, backendNodeId) => `${frameId}-${backendNodeId}`;
+        const maps = await domMapsForSession(session, "frame-A", true, encode, true);
+        const iframeDocKey = `frame-A-${tree.iframeDoc.backendNodeId}`;
+        const iframeBodyKey = `frame-A-${tree.iframeBody.backendNodeId}`;
+        const iframeChildKey = `frame-A-${tree.iframeChild.backendNodeId}`;
+        expect(maps.tagNameMap[iframeDocKey]).toBe("#document");
+        expect(maps.xpathMap[iframeDocKey]).toBe("/");
+        expect(maps.xpathMap[iframeBodyKey]).toBe("/html[1]/body[1]");
+        expect(maps.xpathMap[iframeChildKey]).toBe("/html[1]/body[1]/p[1]");
+        expect(maps.scrollableMap[iframeBodyKey]).toBe(true);
+        expect(Object.keys(maps.tagNameMap)).not.toContain(`frame-A-${tree.html.backendNodeId}`);
+    });
+    it("falls back to the root document when frame owner lookup fails", async () => {
+        const tree = buildSampleDomTree();
+        const session = new MockCDPSession({
+            "DOM.enable": async () => ({}),
+            "DOM.getDocument": async () => ({ root: tree.root }),
+            "DOM.getFrameOwner": async () => {
+                throw new Error("owner lookup failed");
+            },
+            "DOM.describeNode": async () => ({ node: tree.root }),
+        });
+        const encode = (frameId, backendNodeId) => `${frameId}-${backendNodeId}`;
+        const maps = await domMapsForSession(session, "frame-B", false, encode, true);
+        expect(maps.xpathMap[`frame-B-${tree.html.backendNodeId}`]).toBe("/html[1]");
+        expect(maps.xpathMap[`frame-B-${tree.scrollDiv.backendNodeId}`]).toBe("/html[1]/body[1]/div[1]");
+    });
+});
+//# sourceMappingURL=snapshot-dom-session-builders.test.js.map

package/dist/esm/tests/unit/snapshot-dom-session-builders.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshot-dom-session-builders.test.js","sourceRoot":"","sources":["../../../../tests/unit/snapshot-dom-session-builders.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,sBAAsB,EACtB,cAAc,GACf,MAAM,kDAAkD,CAAC;AAC1D,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAE7D,IAAI,UAAU,GAAG,CAAC,CAAC;AACnB,MAAM,WAAW,GAAG,CAClB,YAAwC,EAAE,EACvB,EAAE;IACrB,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,IAAI,UAAU,EAAE,CAAC;IAChD,MAAM,aAAa,GAAG,SAAS,CAAC,aAAa,IAAI,UAAU,EAAE,CAAC;IAC9D,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,IAAI,KAAK,CAAC;IAC7C,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,IAAI,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,IAAI,EAAE,CAAC;IAC1C,OAAO;QACL,MAAM;QACN,aAAa;QACb,QAAQ;QACR,QAAQ;QACR,SAAS,EAAE,SAAS,CAAC,SAAS,IAAI,QAAQ,CAAC,WAAW,EAAE;QACxD,SAAS,EAAE,SAAS,CAAC,SAAS,IAAI,EAAE;QACpC,cAAc,EAAE,SAAS,CAAC,cAAc,IAAI,QAAQ,CAAC,MAAM;QAC3D,QAAQ;QACR,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,YAAY,EAAE,SAAS,CAAC,YAAY;KACrC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,kBAAkB,GAAG,GAAG,EAAE;IAC9B,MAAM,WAAW,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,WAAW,CAAC;QAC7B,QAAQ,EAAE,MAAM;QAChB,QAAQ,EAAE,CAAC,WAAW,CAAC;QACvB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,WAAW,CAAC;QAC5B,QAAQ,EAAE,WAAW;QACrB,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,CAAC,UAAU,CAAC;KACvB,CAAC,CAAC;IACH,MAAM,aAAa,GAAG,WAAW,CAAC;QAChC,QAAQ,EAAE,QAAQ;QAClB,eAAe,EAAE,SAAS;KAC3B,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,WAAW,CAAC;QAC5B,QAAQ,EAAE,KAAK;QACf,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,WAAW,CAAC;QACvB,QAAQ,EAAE,MAAM;QAChB,QAAQ,EAAE,CAAC,SAAS,EAAE,aAAa,CAAC;KACrC,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjE,MAAM,IAAI,GAAG,WAAW,CAAC;QACvB,QAAQ,EAAE,WAAW;QACrB,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,CAAC,IAAI,CAAC;KACjB,CAAC,CAAC;IACH,OAAO;QACL,IAAI;QACJ,IAAI;QACJ,IAAI;QACJ,SAAS;QACT,aAAa;QACb,SAAS;QACT,UAAU;QACV,UAAU;QACV,WAAW;KACZ,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,WAAW,CAAC;YACvB,QAAQ,EAAE,MAAM;YAChB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC/B,IAAI,EAAE;oBACJ,GAAG,IAAI;oBACP,QAAQ,EAAE,CAAC,KAAK,CAAC;oBACjB,cAAc,EAAE,CAAC;iBAClB;aACF,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,WAAW,CAAC;YACvB,QAAQ,EAAE,MAAM;YAChB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,kBAAkB,EAAE,KAAK,IAAI,EAAE;gBAC7B,QAAQ,EAAE,CAAC;gBACX,IAAI,QAAQ,KAAK,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;gBAClE,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;YACrE,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,IAAI,GAAG,WAAW,CAAC;YACvB,QAAQ,EAAE,MAAM;YAChB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,kBAAkB,EAAE,KAAK,IAAI,EAAE;gBAC7B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CACtE,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,IAAI,GAAG,WAAW,CAAC;YACvB,QAAQ,EAAE,WAAW;YACrB,QAAQ,EAAE,CAAC;YACX,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QACH,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;gBAClC,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,CAAW,CAAC;gBAC7C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,IAAI,KAAK,KAAK,CAAC,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;gBAChE,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,CAAC;YACD,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;SACjD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,iBAAiB,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;YAClC,CAAC;SACF,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,cAAc,CACf,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,iBAAiB,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;SACF,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CACzE,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9B,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;SACtD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAExD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC5E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAC1D,yBAAyB,CAC1B,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAC7D,IAAI,CAAC,SAAS,CAAC,aAAa,CAC7B,CAAC;QACF,MAAM,CACJ,KAAK,CAAC,sBAAsB,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CACnE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9B,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,mBAAmB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAChC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,aAAa;aAChD,CAAC;YACF,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;SACtD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,aAAqB,EAAE,EAAE,CACxD,GAAG,OAAO,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAClC,OAAO,EACP,SAAS,EACT,IAAI,EACJ,MAAM,EACN,IAAI,CACL,CAAC;QAEF,MAAM,YAAY,GAAG,WAAW,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC;QAC/D,MAAM,aAAa,GAAG,WAAW,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;QACjE,MAAM,cAAc,GAAG,WAAW,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;QAEnE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACpE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAChD,WAAW,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CACrC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9B,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,mBAAmB,EAAE,KAAK,IAAI,EAAE;gBAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YACD,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;SACtD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,aAAqB,EAAE,EAAE,CACxD,GAAG,OAAO,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAClC,OAAO,EACP,SAAS,EACT,KAAK,EACL,MAAM,EACN,IAAI,CACL,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI,CAC9D,UAAU,CACX,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI,CACnE,yBAAyB,CAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import type { Protocol } from \"devtools-protocol\";\nimport { describe, expect, it } from \"vitest\";\nimport {\n  buildSessionDomIndex,\n  domMapsForSession,\n  getDomTreeWithFallback,\n  hydrateDomTree,\n} from \"../../lib/v3/understudy/a11y/snapshot/domTree.js\";\nimport { StagehandDomProcessError } from \"../../lib/v3/types/public/sdkErrors.js\";\nimport { MockCDPSession } from \"./helpers/mockCDPSession.js\";\n\nlet nextNodeId = 1;\nconst makeDomNode = (\n  overrides: Partial<Protocol.DOM.Node> = {},\n): Protocol.DOM.Node => {\n  const nodeId = overrides.nodeId ?? nextNodeId++;\n  const backendNodeId = overrides.backendNodeId ?? nextNodeId++;\n  const nodeName = overrides.nodeName ?? \"DIV\";\n  const nodeType = overrides.nodeType ?? 1;\n  const children = overrides.children ?? [];\n  return {\n    nodeId,\n    backendNodeId,\n    nodeName,\n    nodeType,\n    localName: overrides.localName ?? nodeName.toLowerCase(),\n    nodeValue: overrides.nodeValue ?? \"\",\n    childNodeCount: overrides.childNodeCount ?? children.length,\n    children,\n    shadowRoots: overrides.shadowRoots,\n    contentDocument: overrides.contentDocument,\n    isScrollable: overrides.isScrollable,\n  };\n};\n\nconst buildSampleDomTree = () => {\n  const iframeChild = makeDomNode({ nodeName: \"P\" });\n  const iframeBody = makeDomNode({\n    nodeName: \"BODY\",\n    children: [iframeChild],\n    isScrollable: true,\n  });\n  const iframeHtml = makeDomNode({ nodeName: \"HTML\", children: [iframeBody] });\n  const iframeDoc = makeDomNode({\n    nodeName: \"#document\",\n    nodeType: 9,\n    children: [iframeHtml],\n  });\n  const iframeElement = makeDomNode({\n    nodeName: \"IFRAME\",\n    contentDocument: iframeDoc,\n  });\n  const scrollDiv = makeDomNode({\n    nodeName: \"DIV\",\n    isScrollable: true,\n  });\n  const body = makeDomNode({\n    nodeName: \"BODY\",\n    children: [scrollDiv, iframeElement],\n  });\n  const html = makeDomNode({ nodeName: \"HTML\", children: [body] });\n  const root = makeDomNode({\n    nodeName: \"#document\",\n    nodeType: 9,\n    children: [html],\n  });\n  return {\n    root,\n    html,\n    body,\n    scrollDiv,\n    iframeElement,\n    iframeDoc,\n    iframeHtml,\n    iframeBody,\n    iframeChild,\n  };\n};\n\ndescribe(\"hydrateDomTree\", () => {\n  it(\"expands truncated nodes by calling DOM.describeNode\", async () => {\n    const child = makeDomNode({ nodeName: \"DIV\" });\n    const root = makeDomNode({\n      nodeName: \"HTML\",\n      childNodeCount: 1,\n      children: [],\n    });\n\n    const session = new MockCDPSession({\n      \"DOM.describeNode\": async () => ({\n        node: {\n          ...root,\n          children: [child],\n          childNodeCount: 1,\n        },\n      }),\n    });\n\n    await hydrateDomTree(session, root, true);\n    expect(root.children).toEqual([child]);\n  });\n\n  it(\"retries describeNode when CBOR errors occur before succeeding\", async () => {\n    const child = makeDomNode({ nodeName: \"DIV\" });\n    const root = makeDomNode({\n      nodeName: \"HTML\",\n      childNodeCount: 1,\n      children: [],\n    });\n\n    let attempts = 0;\n    const session = new MockCDPSession({\n      \"DOM.describeNode\": async () => {\n        attempts++;\n        if (attempts === 1) throw new Error(\"CBOR: stack limit exceeded\");\n        return { node: { ...root, children: [child], childNodeCount: 1 } };\n      },\n    });\n\n    await hydrateDomTree(session, root, true);\n    expect(attempts).toBe(2);\n    expect(root.children).toEqual([child]);\n  });\n\n  it(\"throws StagehandDomProcessError after exhausting describeNode retries\", async () => {\n    const root = makeDomNode({\n      nodeName: \"HTML\",\n      childNodeCount: 1,\n      children: [],\n    });\n    const session = new MockCDPSession({\n      \"DOM.describeNode\": async () => {\n        throw new Error(\"CBOR: stack limit exceeded\");\n      },\n    });\n\n    await expect(hydrateDomTree(session, root, true)).rejects.toBeInstanceOf(\n      StagehandDomProcessError,\n    );\n  });\n});\n\ndescribe(\"getDomTreeWithFallback\", () => {\n  it(\"retries DOM.getDocument after CBOR errors and returns the hydrated root\", async () => {\n    const root = makeDomNode({\n      nodeName: \"#document\",\n      nodeType: 9,\n      children: [],\n    });\n    const depths: number[] = [];\n    const session = new MockCDPSession({\n      \"DOM.getDocument\": async (params) => {\n        const depth = (params?.depth ?? 0) as number;\n        depths.push(depth);\n        if (depth === -1) throw new Error(\"CBOR: stack limit exceeded\");\n        return { root };\n      },\n      \"DOM.describeNode\": async () => ({ node: root }),\n    });\n\n    const result = await getDomTreeWithFallback(session, true);\n    expect(result).toBe(root);\n    expect(depths).toEqual([-1, 256]);\n  });\n\n  it(\"propagates non-CBOR DOM.getDocument errors\", async () => {\n    const session = new MockCDPSession({\n      \"DOM.getDocument\": async () => {\n        throw new Error(\"network fail\");\n      },\n    });\n    await expect(getDomTreeWithFallback(session, false)).rejects.toThrow(\n      \"network fail\",\n    );\n  });\n\n  it(\"throws StagehandDomProcessError when all depth attempts hit CBOR limits\", async () => {\n    const session = new MockCDPSession({\n      \"DOM.getDocument\": async () => {\n        throw new Error(\"CBOR: stack limit exceeded\");\n      },\n    });\n    await expect(getDomTreeWithFallback(session, false)).rejects.toBeInstanceOf(\n      StagehandDomProcessError,\n    );\n  });\n});\n\ndescribe(\"buildSessionDomIndex\", () => {\n  it(\"collects absolute paths, scrollability, and content-document metadata\", async () => {\n    const tree = buildSampleDomTree();\n    const session = new MockCDPSession({\n      \"DOM.enable\": async () => ({}),\n      \"DOM.getDocument\": async () => ({ root: tree.root }),\n      \"DOM.describeNode\": async () => ({ node: tree.root }),\n    });\n\n    const index = await buildSessionDomIndex(session, true);\n\n    expect(index.rootBackend).toBe(tree.root.backendNodeId);\n    expect(index.absByBe.get(tree.body.backendNodeId)).toBe(\"/html[1]/body[1]\");\n    expect(index.absByBe.get(tree.scrollDiv.backendNodeId)).toBe(\n      \"/html[1]/body[1]/div[1]\",\n    );\n    expect(index.scrollByBe.get(tree.scrollDiv.backendNodeId)).toBe(true);\n    expect(index.docRootOf.get(tree.iframeHtml.backendNodeId)).toBe(\n      tree.iframeDoc.backendNodeId,\n    );\n    expect(\n      index.contentDocRootByIframe.get(tree.iframeElement.backendNodeId),\n    ).toBe(tree.iframeDoc.backendNodeId);\n  });\n});\n\ndescribe(\"domMapsForSession\", () => {\n  it(\"derives frame-relative xpath/tag/scrollable maps for a frame's document root\", async () => {\n    const tree = buildSampleDomTree();\n    const session = new MockCDPSession({\n      \"DOM.enable\": async () => ({}),\n      \"DOM.getDocument\": async () => ({ root: tree.root }),\n      \"DOM.getFrameOwner\": async () => ({\n        backendNodeId: tree.iframeElement.backendNodeId,\n      }),\n      \"DOM.describeNode\": async () => ({ node: tree.root }),\n    });\n\n    const encode = (frameId: string, backendNodeId: number) =>\n      `${frameId}-${backendNodeId}`;\n    const maps = await domMapsForSession(\n      session,\n      \"frame-A\",\n      true,\n      encode,\n      true,\n    );\n\n    const iframeDocKey = `frame-A-${tree.iframeDoc.backendNodeId}`;\n    const iframeBodyKey = `frame-A-${tree.iframeBody.backendNodeId}`;\n    const iframeChildKey = `frame-A-${tree.iframeChild.backendNodeId}`;\n\n    expect(maps.tagNameMap[iframeDocKey]).toBe(\"#document\");\n    expect(maps.xpathMap[iframeDocKey]).toBe(\"/\");\n    expect(maps.xpathMap[iframeBodyKey]).toBe(\"/html[1]/body[1]\");\n    expect(maps.xpathMap[iframeChildKey]).toBe(\"/html[1]/body[1]/p[1]\");\n    expect(maps.scrollableMap[iframeBodyKey]).toBe(true);\n    expect(Object.keys(maps.tagNameMap)).not.toContain(\n      `frame-A-${tree.html.backendNodeId}`,\n    );\n  });\n\n  it(\"falls back to the root document when frame owner lookup fails\", async () => {\n    const tree = buildSampleDomTree();\n    const session = new MockCDPSession({\n      \"DOM.enable\": async () => ({}),\n      \"DOM.getDocument\": async () => ({ root: tree.root }),\n      \"DOM.getFrameOwner\": async () => {\n        throw new Error(\"owner lookup failed\");\n      },\n      \"DOM.describeNode\": async () => ({ node: tree.root }),\n    });\n\n    const encode = (frameId: string, backendNodeId: number) =>\n      `${frameId}-${backendNodeId}`;\n    const maps = await domMapsForSession(\n      session,\n      \"frame-B\",\n      false,\n      encode,\n      true,\n    );\n\n    expect(maps.xpathMap[`frame-B-${tree.html.backendNodeId}`]).toBe(\n      \"/html[1]\",\n    );\n    expect(maps.xpathMap[`frame-B-${tree.scrollDiv.backendNodeId}`]).toBe(\n      \"/html[1]/body[1]/div[1]\",\n    );\n  });\n});\n"]}

package/dist/esm/tests/unit/snapshot-dom-tree-utils.test.d.ts

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

package/dist/esm/tests/unit/snapshot-dom-tree-utils.test.js

@@ -0,0 +1,105 @@
+import { describe, expect, it } from "vitest";
+import { collectDomTraversalTargets, findNodeByBackendId, mergeDomNodes, shouldExpandNode, } from "../../lib/v3/understudy/a11y/snapshot/domTree.js";
+let nextNodeId = 1;
+const makeNode = (overrides = {}) => {
+    const base = {
+        nodeId: nextNodeId++,
+        backendNodeId: nextNodeId++,
+        nodeType: 1,
+        nodeName: "DIV",
+        localName: "div",
+        nodeValue: "",
+        childNodeCount: overrides.childNodeCount ??
+            (overrides.children ? overrides.children.length : 0),
+    };
+    return { ...base, ...overrides };
+};
+describe("shouldExpandNode", () => {
+    it("returns true when declared children exceed realized children", () => {
+        const node = makeNode({
+            childNodeCount: 2,
+            children: [makeNode()],
+        });
+        expect(shouldExpandNode(node)).toBe(true);
+    });
+    it("returns false when all declared children are realized", () => {
+        const child = makeNode();
+        const node = makeNode({
+            childNodeCount: 1,
+            children: [child],
+        });
+        expect(shouldExpandNode(node)).toBe(false);
+    });
+});
+describe("mergeDomNodes", () => {
+    it("overrides structural fields with expanded node data", () => {
+        const originalChildren = [makeNode({ nodeName: "SPAN" })];
+        const target = makeNode({
+            childNodeCount: 1,
+            children: originalChildren,
+            shadowRoots: [makeNode({ nodeName: "shadow-old" })],
+            contentDocument: makeNode({ nodeName: "doc-old" }),
+        });
+        const source = makeNode({
+            childNodeCount: 3,
+            children: [makeNode({ nodeName: "DIV" })],
+            shadowRoots: [],
+            contentDocument: makeNode({ nodeName: "doc-new" }),
+        });
+        mergeDomNodes(target, source);
+        expect(target.childNodeCount).toBe(3);
+        expect(target.children).toEqual(source.children);
+        expect(target.shadowRoots).toEqual([]);
+        expect(target.contentDocument?.nodeName).toBe("doc-new");
+    });
+    it("preserves original structures when source omits them", () => {
+        const child = makeNode();
+        const target = makeNode({
+            childNodeCount: 1,
+            children: [child],
+        });
+        const source = makeNode({
+            childNodeCount: 5,
+        });
+        mergeDomNodes(target, source);
+        expect(target.childNodeCount).toBe(5);
+        expect(target.children).toEqual([child]);
+    });
+});
+describe("collectDomTraversalTargets", () => {
+    it("returns children, shadow roots, and content document in order", () => {
+        const childA = makeNode({ nodeName: "CHILD-A" });
+        const childB = makeNode({ nodeName: "CHILD-B" });
+        const shadow = makeNode({ nodeName: "SHADOW" });
+        const content = makeNode({ nodeName: "CONTENT" });
+        const node = makeNode({
+            children: [childA, childB],
+            shadowRoots: [shadow],
+            contentDocument: content,
+        });
+        const targets = collectDomTraversalTargets(node);
+        expect(targets).toEqual([childA, childB, shadow, content]);
+    });
+});
+describe("findNodeByBackendId", () => {
+    it("finds nodes nested within children and shadow roots", () => {
+        const target = makeNode({ backendNodeId: 999, nodeName: "TARGET" });
+        const root = makeNode({
+            children: [
+                makeNode({
+                    children: [makeNode(), target],
+                }),
+            ],
+            shadowRoots: [makeNode()],
+        });
+        expect(findNodeByBackendId(root, 999)).toBe(target);
+    });
+    it("returns undefined when no node matches the backend id", () => {
+        const root = makeNode({
+            children: [makeNode()],
+            shadowRoots: [makeNode()],
+        });
+        expect(findNodeByBackendId(root, 123456)).toBeUndefined();
+    });
+});
+//# sourceMappingURL=snapshot-dom-tree-utils.test.js.map

package/dist/esm/tests/unit/snapshot-dom-tree-utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshot-dom-tree-utils.test.js","sourceRoot":"","sources":["../../../../tests/unit/snapshot-dom-tree-utils.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,0BAA0B,EAC1B,mBAAmB,EACnB,aAAa,EACb,gBAAgB,GACjB,MAAM,kDAAkD,CAAC;AAE1D,IAAI,UAAU,GAAG,CAAC,CAAC;AACnB,MAAM,QAAQ,GAAG,CACf,YAAwC,EAAE,EACvB,EAAE;IACrB,MAAM,IAAI,GAAsB;QAC9B,MAAM,EAAE,UAAU,EAAE;QACpB,aAAa,EAAE,UAAU,EAAE;QAC3B,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,KAAK;QAChB,SAAS,EAAE,EAAE;QACb,cAAc,EACZ,SAAS,CAAC,cAAc;YACxB,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;KACvD,CAAC;IACF,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,SAAS,EAAE,CAAC;AACnC,CAAC,CAAC;AAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;SACvB,CAAC,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,CAAC,KAAK,CAAC;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,gBAAgB,GAAG,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,QAAQ,CAAC;YACtB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,gBAAgB;YAC1B,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;YACnD,eAAe,EAAE,QAAQ,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;SACnD,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC;YACtB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACzC,WAAW,EAAE,EAAE;YACf,eAAe,EAAE,QAAQ,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;SACnD,CAAC,CAAC;QAEH,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAE9B,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,QAAQ,CAAC;YACtB,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,CAAC,KAAK,CAAC;SAClB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC;YACtB,cAAc,EAAE,CAAC;SAClB,CAAC,CAAC;QAEH,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAE9B,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QAChD,MAAM,OAAO,GAAG,QAAQ,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QAElD,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;YAC1B,WAAW,EAAE,CAAC,MAAM,CAAC;YACrB,eAAe,EAAE,OAAO;SACzB,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpE,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,QAAQ,EAAE;gBACR,QAAQ,CAAC;oBACP,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC;iBAC/B,CAAC;aACH;YACD,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC;SAC1B,CAAC,CAAC;QAEH,MAAM,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;YACtB,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,mBAAmB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import type { Protocol } from \"devtools-protocol\";\nimport { describe, expect, it } from \"vitest\";\nimport {\n  collectDomTraversalTargets,\n  findNodeByBackendId,\n  mergeDomNodes,\n  shouldExpandNode,\n} from \"../../lib/v3/understudy/a11y/snapshot/domTree.js\";\n\nlet nextNodeId = 1;\nconst makeNode = (\n  overrides: Partial<Protocol.DOM.Node> = {},\n): Protocol.DOM.Node => {\n  const base: Protocol.DOM.Node = {\n    nodeId: nextNodeId++,\n    backendNodeId: nextNodeId++,\n    nodeType: 1,\n    nodeName: \"DIV\",\n    localName: \"div\",\n    nodeValue: \"\",\n    childNodeCount:\n      overrides.childNodeCount ??\n      (overrides.children ? overrides.children.length : 0),\n  };\n  return { ...base, ...overrides };\n};\n\ndescribe(\"shouldExpandNode\", () => {\n  it(\"returns true when declared children exceed realized children\", () => {\n    const node = makeNode({\n      childNodeCount: 2,\n      children: [makeNode()],\n    });\n    expect(shouldExpandNode(node)).toBe(true);\n  });\n\n  it(\"returns false when all declared children are realized\", () => {\n    const child = makeNode();\n    const node = makeNode({\n      childNodeCount: 1,\n      children: [child],\n    });\n    expect(shouldExpandNode(node)).toBe(false);\n  });\n});\n\ndescribe(\"mergeDomNodes\", () => {\n  it(\"overrides structural fields with expanded node data\", () => {\n    const originalChildren = [makeNode({ nodeName: \"SPAN\" })];\n    const target = makeNode({\n      childNodeCount: 1,\n      children: originalChildren,\n      shadowRoots: [makeNode({ nodeName: \"shadow-old\" })],\n      contentDocument: makeNode({ nodeName: \"doc-old\" }),\n    });\n    const source = makeNode({\n      childNodeCount: 3,\n      children: [makeNode({ nodeName: \"DIV\" })],\n      shadowRoots: [],\n      contentDocument: makeNode({ nodeName: \"doc-new\" }),\n    });\n\n    mergeDomNodes(target, source);\n\n    expect(target.childNodeCount).toBe(3);\n    expect(target.children).toEqual(source.children);\n    expect(target.shadowRoots).toEqual([]);\n    expect(target.contentDocument?.nodeName).toBe(\"doc-new\");\n  });\n\n  it(\"preserves original structures when source omits them\", () => {\n    const child = makeNode();\n    const target = makeNode({\n      childNodeCount: 1,\n      children: [child],\n    });\n    const source = makeNode({\n      childNodeCount: 5,\n    });\n\n    mergeDomNodes(target, source);\n\n    expect(target.childNodeCount).toBe(5);\n    expect(target.children).toEqual([child]);\n  });\n});\n\ndescribe(\"collectDomTraversalTargets\", () => {\n  it(\"returns children, shadow roots, and content document in order\", () => {\n    const childA = makeNode({ nodeName: \"CHILD-A\" });\n    const childB = makeNode({ nodeName: \"CHILD-B\" });\n    const shadow = makeNode({ nodeName: \"SHADOW\" });\n    const content = makeNode({ nodeName: \"CONTENT\" });\n\n    const node = makeNode({\n      children: [childA, childB],\n      shadowRoots: [shadow],\n      contentDocument: content,\n    });\n\n    const targets = collectDomTraversalTargets(node);\n    expect(targets).toEqual([childA, childB, shadow, content]);\n  });\n});\n\ndescribe(\"findNodeByBackendId\", () => {\n  it(\"finds nodes nested within children and shadow roots\", () => {\n    const target = makeNode({ backendNodeId: 999, nodeName: \"TARGET\" });\n    const root = makeNode({\n      children: [\n        makeNode({\n          children: [makeNode(), target],\n        }),\n      ],\n      shadowRoots: [makeNode()],\n    });\n\n    expect(findNodeByBackendId(root, 999)).toBe(target);\n  });\n\n  it(\"returns undefined when no node matches the backend id\", () => {\n    const root = makeNode({\n      children: [makeNode()],\n      shadowRoots: [makeNode()],\n    });\n    expect(findNodeByBackendId(root, 123456)).toBeUndefined();\n  });\n});\n"]}

package/dist/esm/tests/unit/snapshot-focus-selectors-utils.test.d.ts

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