|
| 1 | +import { inflateSync } from "node:zlib"; |
| 2 | +import { describe, expect, it } from "vitest"; |
| 3 | +import { createMermaidInkUrl } from "../../src/utils/mermaidUrl"; |
| 4 | + |
| 5 | +function decodePayload(url: string): { |
| 6 | + code: string; |
| 7 | + mermaid: { theme: string; look: string }; |
| 8 | +} { |
| 9 | + const encoded = url.split("pako:")[1]; |
| 10 | + const base64 = encoded.replace(/-/g, "+").replace(/_/g, "/"); |
| 11 | + const padded = base64 + "=".repeat((4 - (base64.length % 4)) % 4); |
| 12 | + const buffer = Buffer.from(padded, "base64"); |
| 13 | + return JSON.parse(inflateSync(buffer).toString("utf-8")); |
| 14 | +} |
| 15 | + |
| 16 | +describe("createMermaidInkUrl", () => { |
| 17 | + const sample = "graph TD;A-->B;"; |
| 18 | + |
| 19 | + it("defaults to classic look when not specified", () => { |
| 20 | + const url = createMermaidInkUrl(sample, "img"); |
| 21 | + expect(url.startsWith("https://mermaid.ink/img/pako:")).toBe(true); |
| 22 | + const payload = decodePayload(url); |
| 23 | + expect(payload.mermaid.theme).toBe("default"); |
| 24 | + expect(payload.mermaid.look).toBe("classic"); |
| 25 | + expect(payload.code).toBe(sample); |
| 26 | + }); |
| 27 | + |
| 28 | + it("propagates handDrawn look into the mermaid.ink payload", () => { |
| 29 | + const url = createMermaidInkUrl(sample, "svg", "forest", "handDrawn"); |
| 30 | + expect(url.startsWith("https://mermaid.ink/svg/pako:")).toBe(true); |
| 31 | + const payload = decodePayload(url); |
| 32 | + expect(payload.mermaid.theme).toBe("forest"); |
| 33 | + expect(payload.mermaid.look).toBe("handDrawn"); |
| 34 | + }); |
| 35 | + |
| 36 | + it("uses the requested variant in the URL path", () => { |
| 37 | + const svgUrl = createMermaidInkUrl(sample, "svg"); |
| 38 | + const imgUrl = createMermaidInkUrl(sample, "img"); |
| 39 | + expect(svgUrl).toMatch(/^https:\/\/mermaid\.ink\/svg\//); |
| 40 | + expect(imgUrl).toMatch(/^https:\/\/mermaid\.ink\/img\//); |
| 41 | + }); |
| 42 | +}); |
0 commit comments