import { describe, expect, it, vi } from "vitest"; import { buildNotifyRequest, isNotifyUrlValid, normalizeDiscordMention, sendNotification, truncateContent } from "../src/main/notify"; const noSleep = async (): Promise => {}; const WEBHOOK = "https://discord.com/api/webhooks/123/abc"; describe("normalizeDiscordMention", () => { it("wraps a bare user ID as a pinging mention", () => { expect(normalizeDiscordMention("123456789012345678")).toBe("<@123456789012345678>"); expect(normalizeDiscordMention(" 987654321 ")).toBe("<@987654321>"); }); it("passes @everyone/@here and formed mentions through", () => { expect(normalizeDiscordMention("@everyone")).toBe("@everyone"); expect(normalizeDiscordMention("@here")).toBe("@here"); expect(normalizeDiscordMention("<@123456789>")).toBe("<@123456789>"); expect(normalizeDiscordMention("<@&111222333>")).toBe("<@&111222333>"); }); it("returns empty for empty input", () => { expect(normalizeDiscordMention("")).toBe(""); expect(normalizeDiscordMention(" ")).toBe(""); }); }); describe("isNotifyUrlValid", () => { it("accepts http/https URLs", () => { expect(isNotifyUrlValid(WEBHOOK)).toBe(true); expect(isNotifyUrlValid("http://192.168.1.10:8080/hook")).toBe(true); expect(isNotifyUrlValid(` ${WEBHOOK} `)).toBe(true); }); it("rejects empty and non-http values", () => { expect(isNotifyUrlValid("")).toBe(false); expect(isNotifyUrlValid("discord.com/api/webhooks/123/abc")).toBe(false); expect(isNotifyUrlValid("ftp://x")).toBe(false); expect(isNotifyUrlValid("https:// mit leerzeichen")).toBe(false); expect(isNotifyUrlValid("***")).toBe(false); }); }); describe("truncateContent", () => { it("leaves short content untouched", () => { expect(truncateContent("hallo")).toBe("hallo"); }); it("caps at the limit", () => { expect(truncateContent("x".repeat(3000)).length).toBe(2000); }); it("never splits a surrogate pair at the boundary", () => { const emoji = "🏁"; const content = "x".repeat(1999) + emoji; const cut = truncateContent(content); expect(cut.length).toBe(1999); expect(/[\uD800-\uDBFF]$/.test(cut)).toBe(false); }); }); describe("buildNotifyRequest", () => { it("builds a bounded Discord embed with the product username", () => { const req = buildNotifyRequest(` ${WEBHOOK} `, { title: "Paket fehlgeschlagen", message: "Eine Datei konnte nicht verarbeitet werden.", color: 0xe74c3c, fields: [ { name: "Ergebnis", value: "Fehlgeschlagen", inline: true }, { name: "Dateien", value: "0 erfolgreich, 1 fehlgeschlagen", inline: true } ], timestamp: 1000 }); expect(req.url).toBe(WEBHOOK); expect(req.init.method).toBe("POST"); expect(req.init.headers).toMatchObject({ "Content-Type": "application/json" }); const body = JSON.parse(String(req.init.body)); expect(body).toEqual({ username: "Multi-Debrid Downloader", content: "", embeds: [{ title: "Paket fehlgeschlagen", description: "Eine Datei konnte nicht verarbeitet werden.", color: 0xe74c3c, fields: [ { name: "Ergebnis", value: "Fehlgeschlagen", inline: true }, { name: "Dateien", value: "0 erfolgreich, 1 fehlgeschlagen", inline: true } ], timestamp: "1970-01-01T00:00:01.000Z" }] }); }); it("prepends the mention so Discord pings (bare ID gets wrapped)", () => { const req = buildNotifyRequest(WEBHOOK, { title: "T", message: "M", mention: "123456789012345678" }); const body = JSON.parse(String(req.init.body)); expect(body.content).toBe("<@123456789012345678>"); expect(body.embeds[0]).toMatchObject({ title: "T", description: "M" }); }); it("sends no mention prefix when the field is empty", () => { const req = buildNotifyRequest(WEBHOOK, { title: "T", message: "M", mention: "" }); const body = JSON.parse(String(req.init.body)); expect(body.content).toBe(""); }); it("enforces Discord title, description, field, and total embed limits", () => { const req = buildNotifyRequest(WEBHOOK, { title: "T".repeat(400), message: "M".repeat(5000), fields: Array.from({ length: 30 }, (_, index) => ({ name: `${index}-${"N".repeat(300)}`, value: "V".repeat(1400) })) }); const body = JSON.parse(String(req.init.body)); const embed = body.embeds[0] as { title: string; description: string; fields: Array<{ name: string; value: string }> }; expect(embed.title.length).toBeLessThanOrEqual(256); expect(embed.description.length).toBeLessThanOrEqual(4096); expect(embed.fields.length).toBeLessThanOrEqual(25); expect(embed.fields.every((field) => field.name.length <= 256 && field.value.length <= 1024)).toBe(true); const total = embed.title.length + embed.description.length + embed.fields.reduce((sum, field) => sum + field.name.length + field.value.length, 0); expect(total).toBeLessThanOrEqual(6000); }); }); describe("sendNotification", () => { it("returns true on HTTP ok (Discord answers 204 No Content)", async () => { const fetchFn = vi.fn().mockResolvedValue(new Response(null, { status: 204 })); await expect(sendNotification(WEBHOOK, { title: "T", message: "M" }, fetchFn, noSleep)).resolves.toBe(true); expect(fetchFn).toHaveBeenCalledTimes(1); }); it("retries a 429 using Discord's retry_after and then succeeds", async () => { const waits: number[] = []; const sleepSpy = async (ms: number): Promise => { waits.push(ms); }; const fetchFn = vi.fn() .mockResolvedValueOnce(new Response(JSON.stringify({ retry_after: 1.2 }), { status: 429 })) .mockResolvedValueOnce(new Response(null, { status: 204 })); await expect(sendNotification(WEBHOOK, { title: "T", message: "M" }, fetchFn, sleepSpy)).resolves.toBe(true); expect(fetchFn).toHaveBeenCalledTimes(2); expect(waits).toContain(1200); // seconds -> ms }); it("retries transient 5xx and network errors, then gives up", async () => { const fetchFn = vi.fn().mockResolvedValue(new Response("", { status: 502 })); await expect(sendNotification(WEBHOOK, { title: "T", message: "M" }, fetchFn, noSleep)).resolves.toBe(false); expect(fetchFn).toHaveBeenCalledTimes(3); // initial + 2 retries const fetchErr = vi.fn().mockRejectedValue(new Error("offline")); await expect(sendNotification(WEBHOOK, { title: "T", message: "M" }, fetchErr, noSleep)).resolves.toBe(false); expect(fetchErr).toHaveBeenCalledTimes(3); }); it("does not retry a permanent 4xx", async () => { const fetchFn = vi.fn().mockResolvedValue(new Response("", { status: 404 })); await expect(sendNotification(WEBHOOK, { title: "T", message: "M" }, fetchFn, noSleep)).resolves.toBe(false); expect(fetchFn).toHaveBeenCalledTimes(1); }); it("serializes concurrent sends in order (burst protection)", async () => { const order: string[] = []; const fetchFn = vi.fn().mockImplementation(async (_url: string, init: RequestInit) => { order.push(JSON.parse(String(init.body)).embeds[0].title); return new Response(null, { status: 204 }); }); const sends = [ sendNotification(WEBHOOK, { title: "1", message: "" }, fetchFn, noSleep), sendNotification(WEBHOOK, { title: "2", message: "" }, fetchFn, noSleep), sendNotification(WEBHOOK, { title: "3", message: "" }, fetchFn, noSleep) ]; await expect(Promise.all(sends)).resolves.toEqual([true, true, true]); expect(order).toEqual(["1", "2", "3"]); }); it("does not call fetch for an invalid URL", async () => { const fetchFn = vi.fn(); await expect(sendNotification("", { title: "T", message: "M" }, fetchFn, noSleep)).resolves.toBe(false); await expect(sendNotification("kein-url", { title: "T", message: "M" }, fetchFn, noSleep)).resolves.toBe(false); expect(fetchFn).not.toHaveBeenCalled(); }); });