", - "<span>before</span>", - "before html", - "", - "after html", - "<span>after</span>", - "
", - ].join(""), - replace: "<span>replace</span>
", - replaceHtml: "replace
", - remove: "", -}; - -const commentPropertiesMacro: Macro< - [(rw: HTMLRewriter, comments: (comment: Comment) => void) => HTMLRewriter] -> = async (t, func) => { - t.plan(3); - const res = await func(new HTMLRewriter(), (comment) => { - t.false(comment.removed); - t.is(comment.text, "test"); - comment.text = "new"; - }).transform(""); - t.is(res, ""); -}; -test("handles comment properties", commentPropertiesMacro, (rw, comments) => - rw.on("p", { comments }) -); -test( - "handles comment mutations", - mutationsMacro, - (rw, comments) => rw.on("p", { comments }), - commentsMutationsInput, - commentsMutationsExpected -); -test("comment allows chaining", async (t) => { - t.plan(4); - await new HTMLRewriter() - .on("p", { - comments(comment) { - t.is(comment.before(""), comment); - t.is(comment.after(""), comment); - t.is(comment.replace(""), comment); - t.is(comment.remove(), comment); - }, - }) - .transform(""); -}); -const commentAsyncHandlerMacro: Macro< - [(rw: HTMLRewriter, comments: (c: Comment) => Promisetest
"); - t.is( - res, - [ - "", - "test", - "
", - "<span>append</span>", - "append html", - ].join("") - ); -}); -test("document end allows chaining", async (t) => { - t.plan(1); - await new HTMLRewriter() - .onDocument({ - end(end) { - t.is(end.append(""), end); - }, - }) - .transform("test
"); -}); -test("handles document end async handler", async (t) => { - const res = await new HTMLRewriter() - .onDocument({ - async end(end) { - await wait(50); - end.append("append html", { html: true }); - }, - }) - .transform("test
"); - t.is(res, "test
append html"); -}); -test("handles document end class handler", async (t) => { - class Handler { - constructor(private content: string) {} - // noinspection JSUnusedGlobalSymbols - end(end: DocumentEnd) { - end.append(this.content, { html: true }); - } - } - const res = await new HTMLRewriter() - .onDocument(new Handler("append html")) - .transform("test
"); - t.is(res, "test
append html"); -}); diff --git a/test/element.spec.ts b/test/element.spec.ts deleted file mode 100644 index ba01a7c..0000000 --- a/test/element.spec.ts +++ /dev/null @@ -1,272 +0,0 @@ -import test from "ava"; -import { Element } from ".."; -import { HTMLRewriter, mutationsMacro, wait } from "."; - -const elementMutationsInput = "test
"; -const elementMutationsExpected = { - beforeAfter: [ - "<span>before</span>", - "before html", - "", - "test", - "
", - "after html", - "<span>after</span>", - ].join(""), - replace: "<span>replace</span>", - replaceHtml: "replace", - remove: "", -}; - -test("handles element properties", async (t) => { - t.plan(6); - const res = await new HTMLRewriter() - .on("p", { - element(element) { - t.is(element.tagName, "p"); - element.tagName = "h1"; - t.false(element.removed); - t.is(element.namespaceURI, "http://www.w3.org/1999/xhtml"); - - // Check element.attributes is an IterableIterator - t.deepEqual(element.attributes.next(), { - value: ["class", "red"], - done: false, - }); - t.deepEqual([...element.attributes], [["class", "red"]]); - }, - }) - .transform('test
'); - t.is(res, 'test
'); -}); -test("handles element attribute methods", async (t) => { - t.plan(5); - const res = await new HTMLRewriter() - .on("p", { - element(element) { - t.is(element.getAttribute("class"), "red"); - t.is(element.getAttribute("id"), null); - t.true(element.hasAttribute("class")); - t.false(element.hasAttribute("id")); - element.setAttribute("id", "header"); - element.removeAttribute("class"); - }, - }) - .transform('test
'); - t.is(res, 'test
'); -}); -test( - "handles element mutations", - mutationsMacro, - (rw, element) => rw.on("p", { element }), - elementMutationsInput, - elementMutationsExpected -); -test("handles element specific mutations", async (t) => { - // prepend/append - let res = await new HTMLRewriter() - .on("p", { - element(element) { - element.prepend("prepend"); - element.prepend("prepend html", { html: true }); - element.append("append"); - element.append("append html", { html: true }); - }, - }) - .transform("test
"); - t.is( - res, - [ - "", - "prepend html", - "<span>prepend</span>", - "test", - "<span>append</span>", - "append html", - "
", - ].join("") - ); - - // setInnerContent - res = await new HTMLRewriter() - .on("p", { - element(element) { - element.setInnerContent("replace"); - }, - }) - .transform("test
"); - t.is(res, "<span>replace</span>
"); - res = await new HTMLRewriter() - .on("p", { - element(element) { - element.setInnerContent("replace", { html: true }); - }, - }) - .transform("test
"); - t.is(res, "replace
"); - - // removeAndKeepContent - res = await new HTMLRewriter() - .on("p", { - element(element) { - element.removeAndKeepContent(); - }, - }) - .transform("test
"); - t.is(res, "test"); -}); -test("element allows chaining", async (t) => { - t.plan(10); - await new HTMLRewriter() - .on("p", { - element(element) { - t.is(element.before(""), element); - t.is(element.after(""), element); - t.is(element.replace(""), element); - t.is(element.remove(), element); - t.is(element.setAttribute("test", ""), element); - t.is(element.removeAttribute("test"), element); - t.is(element.prepend(""), element); - t.is(element.append(""), element); - t.is(element.setInnerContent(""), element); - t.is(element.removeAndKeepContent(), element); - }, - }) - .transform("test
"); -}); -test.serial("handles element async handler", async (t) => { - const res = await new HTMLRewriter() - .on("p", { - async element(element) { - await wait(50); - element.setInnerContent("new"); - }, - }) - .transform("test
"); - t.is(res, "new
"); -}); -test("handles element class handler", async (t) => { - class Handler { - constructor(private content: string) {} - // noinspection JSUnusedGlobalSymbols - element(element: Element) { - element.setInnerContent(this.content); - } - } - const res = await new HTMLRewriter() - .on("p", new Handler("new")) - .transform("test
"); - t.is(res, "new
"); -}); - -test("handles end tag properties", async (t) => { - const res = await new HTMLRewriter() - .on("p", { - element(element) { - element.onEndTag(function (end) { - t.is(this, element); - t.is(end.name, "p"); - end.name = "h1"; - }); - }, - }) - .transform("test
"); - t.is(res, "test"); -}); -test("handles end tag mutations", async (t) => { - const input = "
test
"; - const beforeAfterExpected = [ - "", - "test", - "<span>before</span>", - "before html", - "
", - "after html", - "<span>after</span>", - ].join(""); - const removeExpected = "test"; - - // before/after - let res = await new HTMLRewriter() - .on("p", { - element(element) { - const that = this; - element.onEndTag((end) => { - t.is(this, that); - end.before("before"); - end.before("before html", { html: true }); - end.after("after"); - end.after("after html", { html: true }); - }); - }, - }) - .transform(input); - t.is(res, beforeAfterExpected); - - // remove - res = await new HTMLRewriter() - .on("p", { - element(element) { - element.onEndTag((end) => { - end.remove(); - }); - }, - }) - .transform(input); - t.is(res, removeExpected); -}); -test("end tag allows chaining", async (t) => { - t.plan(3); - await new HTMLRewriter() - .on("p", { - element(element) { - element.onEndTag((end) => { - t.is(end.before(""), end); - t.is(end.after(""), end); - t.is(end.remove(), end); - }); - }, - }) - .transform("
test
"); -}); -test.serial("handles end tag async handler", async (t) => { - const res = await new HTMLRewriter() - .on("p", { - element(element) { - element.onEndTag(async (end) => { - await wait(50); - end.before("!"); - }); - }, - }) - .transform("test
"); - t.is(res, "test!
"); -}); -test("uses last end tag handler", async (t) => { - const res = await new HTMLRewriter() - .on("p", { - element(element) { - element.onEndTag((end) => { - end.before("1"); - }); - element.onEndTag((end) => { - end.before("2"); - }); - }, - }) - .transform("test
"); - t.is(res, "test2
"); -}); -test("throws error on no end tag", async (t) => { - const res = new HTMLRewriter() - .on("img", { - element(element) { - element.onEndTag(() => t.fail()); - }, - }) - .transform('old h1
old h2
old p
"); - t.is(res, "new h1
new h2
new p
"); -}); - -test("handles streaming", async (t) => { - t.plan(8); // 6 for text handler + 2 at the end - const expectedTextChunks = ["te", "st", ""]; - - const outputChunks: string[] = []; - const decoder = new TextDecoder(); - const rewriter = new RawHTMLRewriter((chunk) => - outputChunks.push(decoder.decode(chunk)) - ).on("p", { - text(text) { - t.is(text.text, expectedTextChunks.shift()); - t.is(text.lastInTextNode, text.text === ""); - }, - }); - - const inputChunks = [ - '', - "", - "te", - "st", - "
", - "