75 lines
2.3 KiB
TypeScript
75 lines
2.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import {
|
|
captureDom,
|
|
captureFrames,
|
|
captureResources,
|
|
} from "../src/browser/CdpCaptureDocuments.js";
|
|
import { inspectWebPageInputSchema } from "../src/domain/browserObservation.js";
|
|
|
|
const origin = "https://app.example.test";
|
|
|
|
describe("CDP document capture", () => {
|
|
it("walks deeply nested frame trees iteratively and retains a bounded prefix", () => {
|
|
let frameTree: Record<string, unknown> = {
|
|
frame: { id: "frame-0", url: `${origin}/0` },
|
|
resources: [{ url: `${origin}/0.js`, type: "Script" }],
|
|
};
|
|
for (let index = 1; index < 2_000; index += 1) {
|
|
frameTree = {
|
|
frame: {
|
|
id: `frame-${String(index)}`,
|
|
url: `${origin}/${String(index)}`,
|
|
},
|
|
resources: [{ url: `${origin}/${String(index)}.js`, type: "Script" }],
|
|
childFrames: [frameTree],
|
|
};
|
|
}
|
|
const result = { frameTree };
|
|
const frames = captureFrames(result, new Set([origin]), 3);
|
|
const resources = captureResources(result, new Set([origin]), 2);
|
|
expect(frames).toMatchObject({ total: 2_000 });
|
|
expect(frames.items).toHaveLength(3);
|
|
expect(resources).toMatchObject({ total: 2_000 });
|
|
expect(resources.items).toHaveLength(2);
|
|
});
|
|
|
|
it("rebases parent indexes when allowed DOM documents are combined", () => {
|
|
const input = inspectWebPageInputSchema.parse({
|
|
cdp_endpoint: "http://127.0.0.1:9222",
|
|
allowed_origins: [origin],
|
|
target_id: "page-1",
|
|
approved: true,
|
|
});
|
|
const snapshot = {
|
|
strings: [`${origin}/main`, `${origin}/frame`, "#document", "DIV", ""],
|
|
documents: [
|
|
{
|
|
documentURL: 0,
|
|
nodes: {
|
|
nodeType: [9, 1],
|
|
nodeName: [2, 3],
|
|
nodeValue: [4, 4],
|
|
parentIndex: [-1, 0],
|
|
attributes: [[], []],
|
|
},
|
|
},
|
|
{
|
|
documentURL: 1,
|
|
nodes: {
|
|
nodeType: [9, 1],
|
|
nodeName: [2, 3],
|
|
nodeValue: [4, 4],
|
|
parentIndex: [-1, 0],
|
|
attributes: [[], []],
|
|
},
|
|
},
|
|
],
|
|
};
|
|
const capture = captureDom(snapshot, new Set([origin]), input);
|
|
expect(capture.nodes.map((node) => node.parent_index)).toEqual([
|
|
-1, 0, -1, 2,
|
|
]);
|
|
});
|
|
});
|