Spaces:
Runtime error
Runtime error
| /** | |
| * Unit tests for src/lib/services/htmlRewriter.ts | |
| * | |
| * Tests HTML rewriting for the embedded-service reverse proxy: | |
| * - <base href> injection | |
| * - Path-absolute URL rewriting in <a>, <link>, <script>, <img>, <form>, <iframe>, <source> | |
| * - Non-rewrite cases: external URLs, protocol-relative, mailto:, javascript: | |
| * - Edge cases: HTML without <head>, empty href, multi-URL srcset | |
| */ | |
| import { describe, it } from "node:test"; | |
| import assert from "node:assert/strict"; | |
| import { rewriteHtml } from "../../../src/lib/services/htmlRewriter.ts"; | |
| const PREFIX = "/dashboard/providers/services/9router/embed"; | |
| // Helper: assert string contains substring | |
| function assertContains(haystack: string, needle: string, msg?: string): void { | |
| assert.ok( | |
| haystack.includes(needle), | |
| msg ?? `Expected output to contain: ${JSON.stringify(needle)}\nActual: ${haystack}` | |
| ); | |
| } | |
| function assertNotContains(haystack: string, needle: string, msg?: string): void { | |
| assert.ok( | |
| !haystack.includes(needle), | |
| msg ?? `Expected output NOT to contain: ${JSON.stringify(needle)}\nActual: ${haystack}` | |
| ); | |
| } | |
| // ─── base tag injection ─────────────────────────────────────────────────────── | |
| describe("rewriteHtml — <base> injection", () => { | |
| it("inserts <base href> as first child of <head>", () => { | |
| const input = "<html><head></head><body></body></html>"; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| }); | |
| it("inserts <base href> before other head children", () => { | |
| const input = "<html><head><title>T</title></head><body></body></html>"; | |
| const out = rewriteHtml(input, PREFIX); | |
| const baseIdx = out.indexOf("<base "); | |
| const titleIdx = out.indexOf("<title>"); | |
| assert.ok( | |
| baseIdx < titleIdx, | |
| `<base> should appear before <title>; base=${baseIdx}, title=${titleIdx}` | |
| ); | |
| }); | |
| it("strips trailing slash from prefix before forming base href", () => { | |
| const out = rewriteHtml("<html><head></head><body></body></html>", `${PREFIX}/`); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| // Should not have double slash | |
| assertNotContains(out, `<base href="${PREFIX}//">`); | |
| }); | |
| it("handles HTML without <head> by creating one", () => { | |
| const out = rewriteHtml("<html><body><p>hello</p></body></html>", PREFIX); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| }); | |
| it("updates existing <base> href instead of inserting a second one", () => { | |
| const input = '<html><head><base href="/old/"></head><body></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| assertNotContains(out, 'href="/old/"'); | |
| // Exactly one <base tag | |
| const matches = out.match(/<base /g); | |
| assert.equal(matches?.length ?? 0, 1, "should have exactly one <base> element"); | |
| }); | |
| }); | |
| // ─── link rewriting ─────────────────────────────────────────────────────────── | |
| describe("rewriteHtml — <a href> rewriting", () => { | |
| it("rewrites path-absolute <a href> to prefixed URL", () => { | |
| const input = '<html><head></head><body><a href="/foo">link</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `href="${PREFIX}/foo"`); | |
| }); | |
| it("does NOT rewrite https:// external URL", () => { | |
| const input = '<html><head></head><body><a href="https://example.com/foo">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="https://example.com/foo"'); | |
| }); | |
| it("does NOT rewrite http:// external URL", () => { | |
| const input = '<html><head></head><body><a href="http://external.com">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="http://external.com"'); | |
| }); | |
| it("does NOT rewrite protocol-relative URL (//cdn)", () => { | |
| const input = '<html><head></head><body><a href="//cdn.example.com/x">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="//cdn.example.com/x"'); | |
| }); | |
| it("does NOT rewrite mailto: URL", () => { | |
| const input = | |
| '<html><head></head><body><a href="mailto:admin@example.com">email</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="mailto:admin@example.com"'); | |
| }); | |
| it("does NOT rewrite javascript: URL", () => { | |
| const input = '<html><head></head><body><a href="javascript:void(0)">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="javascript:void(0)"'); | |
| }); | |
| it("does NOT rewrite relative URL (no leading slash)", () => { | |
| const input = '<html><head></head><body><a href="relative/path">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="relative/path"'); | |
| }); | |
| it("does NOT rewrite anchor-only href", () => { | |
| const input = '<html><head></head><body><a href="#section">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href="#section"'); | |
| }); | |
| }); | |
| // ─── other element rewriting ────────────────────────────────────────────────── | |
| describe("rewriteHtml — other element attributes", () => { | |
| it("rewrites <link href='/stylesheet.css'>", () => { | |
| const input = | |
| '<html><head><link rel="stylesheet" href="/styles.css"></head><body></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `href="${PREFIX}/styles.css"`); | |
| }); | |
| it("rewrites <script src='/app.js'>", () => { | |
| const input = '<html><head></head><body><script src="/app.js"></script></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `src="${PREFIX}/app.js"`); | |
| }); | |
| it("rewrites <img src='/logo.png'>", () => { | |
| const input = '<html><head></head><body><img src="/logo.png"></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `src="${PREFIX}/logo.png"`); | |
| }); | |
| it("rewrites <form action='/submit'>", () => { | |
| const input = | |
| '<html><head></head><body><form action="/submit"><button>Go</button></form></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `action="${PREFIX}/submit"`); | |
| }); | |
| it("rewrites <iframe src='/frame'>", () => { | |
| const input = '<html><head></head><body><iframe src="/frame"></iframe></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `src="${PREFIX}/frame"`); | |
| }); | |
| it("rewrites <source src='/video.mp4'>", () => { | |
| const input = '<html><head></head><body><source src="/video.mp4"></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `src="${PREFIX}/video.mp4"`); | |
| }); | |
| it("does NOT rewrite img srcset with commas (multi-URL — v1 limitation)", () => { | |
| const input = | |
| '<html><head></head><body><img srcset="/img.png 1x, /img@2x.png 2x"></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| // Original srcset must be preserved unchanged | |
| assertContains(out, 'srcset="/img.png 1x, /img@2x.png 2x"'); | |
| }); | |
| }); | |
| // ─── non-anchor markup preservation ────────────────────────────────────────── | |
| describe("rewriteHtml — preserves non-affected markup", () => { | |
| it("preserves text content inside elements", () => { | |
| const input = "<html><head></head><body><p>Hello World</p></body></html>"; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, "Hello World"); | |
| }); | |
| it("preserves existing non-rewritten attributes", () => { | |
| const input = | |
| '<html><head></head><body><a href="/foo" class="nav" id="link1">x</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'class="nav"'); | |
| assertContains(out, 'id="link1"'); | |
| }); | |
| it("preserves data attributes unchanged", () => { | |
| const input = | |
| '<html><head></head><body><div data-url="/foo" data-val="bar"></div></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| // data-url is not in REWRITABLE_ATTRS — stays as-is | |
| assertContains(out, 'data-url="/foo"'); | |
| }); | |
| it("preserves empty href unchanged", () => { | |
| const input = '<html><head></head><body><a href="">empty</a></body></html>'; | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, 'href=""'); | |
| }); | |
| }); | |
| // ─── edge cases ─────────────────────────────────────────────────────────────── | |
| describe("rewriteHtml — edge cases", () => { | |
| it("handles empty HTML string gracefully", () => { | |
| const out = rewriteHtml("", PREFIX); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| }); | |
| it("handles HTML fragment (no <html> tag) — parse5 auto-completes tree", () => { | |
| const out = rewriteHtml('<a href="/foo">link</a>', PREFIX); | |
| assertContains(out, `<base href="${PREFIX}/">`); | |
| assertContains(out, `href="${PREFIX}/foo"`); | |
| }); | |
| it("multiple path-absolute URLs in same document are all rewritten", () => { | |
| const input = [ | |
| "<html><head>", | |
| '<link href="/a.css">', | |
| "</head><body>", | |
| '<a href="/page1">1</a>', | |
| '<a href="/page2">2</a>', | |
| '<script src="/app.js"></script>', | |
| "</body></html>", | |
| ].join(""); | |
| const out = rewriteHtml(input, PREFIX); | |
| assertContains(out, `href="${PREFIX}/a.css"`); | |
| assertContains(out, `href="${PREFIX}/page1"`); | |
| assertContains(out, `href="${PREFIX}/page2"`); | |
| assertContains(out, `src="${PREFIX}/app.js"`); | |
| }); | |
| }); | |