123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292
// deno-lint-ignore-file// import 'https://gist.githubusercontent.com/qwtel/b14f0f81e3a96189f7771f83ee113f64/raw/TestRequest.ts'import {  assert,  assertExists,  assertEquals,  assertStrictEquals,  assertStringIncludes,  assertThrows,  assertRejects,  assertArrayIncludes,} from 'https://deno.land/std@0.133.0/testing/asserts.ts'const { test } = Deno;
import { HTMLRewriter, Element } from "../index.ts";
export const timeout = (t: number) => new Promise(r => setTimeout(r, t));
// const elementMutationsInput = "<p>test</p>";// const elementMutationsExpected = {//   beforeAfter: [//     "&lt;span&gt;before&lt;/span&gt;",//     "<span>before html</span>",//     "<p>",//     "test",//     "</p>",//     "<span>after html</span>",//     "&lt;span&gt;after&lt;/span&gt;",//   ].join(""),//   replace: "&lt;span&gt;replace&lt;/span&gt;",//   replaceHtml: "<span>replace</span>",//   remove: "",// };
test("handles element properties", async () => {  // t.plan(6);  const res = new HTMLRewriter()    .on("p", {      element(element) {        assertEquals(element.tagName, "p");        element.tagName = "h1";        // assert(element.removed); // ??? ?? why        assertEquals(element.namespaceURI, "http://www.w3.org/1999/xhtml");
        // Check element.attributes is an IterableIterator        assertEquals(element.attributes.next(), {          value: ["class", "red"],          done: false,        });        assertEquals([...element.attributes], [["class", "red"]]);      },    })    .transform(new Response('<p class="red">test</p>'));  assertEquals(await res.text(), '<h1 class="red">test</h1>');});
test("handles element attribute methods", async () => {  const res = new HTMLRewriter()    .on("p", {      element(element) {        assertEquals(element.getAttribute("class"), "red");        assertEquals(element.getAttribute("id"), null);        assert(element.hasAttribute("class"));        assert(!element.hasAttribute("id"));        element.setAttribute("id", "header");        element.removeAttribute("class");      },    })    .transform(new Response('<p class="red">test</p>'));  assertEquals(await res.text(), '<p id="header">test</p>');});
// test(//   "handles element mutations",//   mutationsMacro,//   (rw, element) => rw.on("p", { element }),//   elementMutationsInput,//   elementMutationsExpected// );
test("handles element specific mutations", async () => {  // prepend/append  let res = new HTMLRewriter()    .on("p", {      element(element) {        element.prepend("<span>prepend</span>");        element.prepend("<span>prepend html</span>", { html: true });        element.append("<span>append</span>");        element.append("<span>append html</span>", { html: true });      },    })    .transform(new Response("<p>test</p>"));  assertEquals(    await res.text(),    [      "<p>",      "<span>prepend html</span>",      "&lt;span&gt;prepend&lt;/span&gt;",      "test",      "&lt;span&gt;append&lt;/span&gt;",      "<span>append html</span>",      "</p>",    ].join("")  );
  // setInnerContent  res = new HTMLRewriter()    .on("p", {      element(element) {        element.setInnerContent("<span>replace</span>");      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>&lt;span&gt;replace&lt;/span&gt;</p>");  res = new HTMLRewriter()    .on("p", {      element(element) {        element.setInnerContent("<span>replace</span>", { html: true });      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p><span>replace</span></p>");
  // removeAndKeepContent  res = new HTMLRewriter()    .on("p", {      element(element) {        element.removeAndKeepContent();      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "test");});
test("element allows chaining", async () => {  new HTMLRewriter()    .on("p", {      element(element) {        assertEquals(element.before(""), element);        assertEquals(element.after(""), element);        assertEquals(element.replace(""), element);        assertEquals(element.remove(), element);        assertEquals(element.setAttribute("test", ""), element);        assertEquals(element.removeAttribute("test"), element);        assertEquals(element.prepend(""), element);        assertEquals(element.append(""), element);        assertEquals(element.setInnerContent(""), element);        assertEquals(element.removeAndKeepContent(), element);      },    })    .transform(new Response("<p>test</p>"));});
test("handles element async handler", async (t) => {  const res = new HTMLRewriter()    .on("p", {      async element(element) {        await timeout(50);        element.setInnerContent("new");      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>new</p>");});
test("handles element class handler", async (t) => {  class Handler {    constructor(private content: string) {}    // noinspection JSUnusedGlobalSymbols    element(element: Element) {      element.setInnerContent(this.content);    }  }  const res = new HTMLRewriter()    .on("p", new Handler("new"))    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>new</p>");});
test("handles end tag properties", async (t) => {  const res = new HTMLRewriter()    .on("p", {      element(element) {        element.onEndTag(function (end) {          assertEquals(this, element);          assertEquals(end.name, "p");          end.name = "h1";        });      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>test</h1>");});
test("handles end tag mutations", async (t) => {  const input = new Response("<p>test</p>");  const beforeAfterExpected = [    "<p>",    "test",    "&lt;span&gt;before&lt;/span&gt;",    "<span>before html</span>",    "</p>",    "<span>after html</span>",    "&lt;span&gt;after&lt;/span&gt;",  ].join("");  const removeExpected = "<p>test";
  // before/after  let res = new HTMLRewriter()    .on("p", {      element(element) {        const that = this;        element.onEndTag((end) => {          assertStrictEquals(this, that);          end.before("<span>before</span>");          end.before("<span>before html</span>", { html: true });          end.after("<span>after</span>");          end.after("<span>after html</span>", { html: true });        });      },    })    .transform(input.clone());  assertEquals(await res.text(), beforeAfterExpected);
  // remove  res = new HTMLRewriter()    .on("p", {      element(element) {        element.onEndTag((end) => {          end.remove();        });      },    })    .transform(input.clone());  assertEquals(await res.text(), removeExpected);});
test("end tag allows chaining", async () => {  new HTMLRewriter()    .on("p", {      element(element) {        element.onEndTag((end) => {          assertEquals(end.before(""), end);          assertEquals(end.after(""), end);          assertEquals(end.remove(), end);        });      },    })    .transform(new Response("<p>test</p>"));});
test("handles end tag async handler", async () => {  const res = new HTMLRewriter()    .on("p", {      element(element) {        element.onEndTag(async (end) => {          await timeout(50);          end.before("!");        });      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>test!</p>");});
test("uses last end tag handler", async (t) => {  const res = new HTMLRewriter()    .on("p", {      element(element) {        element.onEndTag((end) => {          end.before("1");        });        element.onEndTag((end) => {          end.before("2");        });      },    })    .transform(new Response("<p>test</p>"));  assertEquals(await res.text(), "<p>test2</p>");});
test("throws error on no end tag", async () => {  const res = new HTMLRewriter()    .on("img", {      element(element) {        element.onEndTag(() => { throw Error('error') });      },    })    .transform(new Response('<img src="" alt="">'));  await assertRejects(() => res.arrayBuffer(), TypeError, "Parser error: No end tag.")});