import { Element, isTag } from "domhandler"; import { parseDocument } from "htmlparser2"; import { existsOne, filter, find, findAll, findOne, findOneChild, } from "./querying"; import { ElementType } from "domelementtype"; describe("querying", () => { const manyNodesWide = parseDocument( `${"

".repeat(200_000)}Text`, ).children; describe("find", () => { it("should accept many children without RangeError", () => expect( find( (elem) => elem.type === ElementType.Tag, manyNodesWide, true, Infinity, ), ).toHaveLength(200_001)); it("should respect limit", () => expect( find( (elem) => elem.type === ElementType.Tag, manyNodesWide, true, 20, ), ).toHaveLength(20)); it("should find text nodes", () => expect( find( (elem) => elem.type === ElementType.Text, manyNodesWide, true, Infinity, ), ).toHaveLength(1)); }); describe("findAll", () => { it("should accept many children without RangeError", () => { expect( findAll((elem) => elem.name === "div", manyNodesWide), ).toHaveLength(200_000); }); }); describe("filter", () => { it("should accept many children without RangeError", () => expect( filter((elem) => elem.type === ElementType.Tag, manyNodesWide), ).toHaveLength(200_001)); it("should turn a single node into an array", () => expect( filter( (elem) => elem.type === ElementType.Tag, manyNodesWide[0], ), ).toHaveLength(200_001)); }); describe("findOneChild", () => { it("should find elements", () => expect( findOneChild( (elem) => isTag(elem) && elem.name === "body", manyNodesWide, ), ).toBe(manyNodesWide[0])); it("should only query direct children", () => expect( findOneChild( (elem) => isTag(elem) && elem.name === "div", manyNodesWide, ), ).toBeUndefined()); }); describe("findOne", () => { it("should find elements", () => expect( findOne((elem) => elem.name === "body", manyNodesWide, true), ).toBe(manyNodesWide[0])); it("should find elements in children", () => expect( findOne((elem) => elem.name === "div", manyNodesWide, true), ).toBe((manyNodesWide[0] as Element).children[0])); it("should not find elements in children if recurse is false", () => expect( findOne((elem) => elem.name === "div", manyNodesWide, false), ).toBeNull()); it("should return `null` if nothing is found", () => expect(findOne(() => false, manyNodesWide)).toBeNull()); }); describe("existsOne", () => { it("should find elements", () => expect( existsOne((elem) => elem.name === "body", manyNodesWide), ).toBeTruthy()); it("should find elements in children", () => expect( existsOne((elem) => elem.name === "div", manyNodesWide), ).toBeTruthy()); it("should return `false` if nothing is found", () => expect(existsOne(() => false, manyNodesWide)).toBeFalsy()); }); });