upstream-yaak/plugins/template-function-regex/tests/regex.test.ts

import type { Context } from "@yaakapp/api";
import { describe, expect, it } from "vite-plus/test";
import { plugin } from "../src";

describe("regex.match", () => {
  const matchFunction = plugin.templateFunctions?.find((f) => f.name === "regex.match");

  it("should exist", () => {
    expect(matchFunction).toBeDefined();
  });

  it("should extract first capture group", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "Hello (\\w+)",
        input: "Hello World",
      },
      purpose: "send",
    });
    expect(result).toBe("World");
  });

  it("should extract named capture group", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "Hello (?<name>\\w+)",
        input: "Hello World",
      },
      purpose: "send",
    });
    expect(result).toBe("World");
  });

  it("should return full match when no capture groups", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "Hello \\w+",
        input: "Hello World",
      },
      purpose: "send",
    });
    expect(result).toBe("Hello World");
  });

  it("should return empty string when no match", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "Goodbye",
        input: "Hello World",
      },
      purpose: "send",
    });
    expect(result).toBe("");
  });

  it("should return empty string when regex is empty", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "",
        input: "Hello World",
      },
      purpose: "send",
    });
    expect(result).toBe("");
  });

  it("should return empty string when input is empty", async () => {
    const result = await matchFunction?.onRender({} as Context, {
      values: {
        regex: "Hello",
        input: "",
      },
      purpose: "send",
    });
    expect(result).toBe("");
  });
});

describe("regex.replace", () => {
  const replaceFunction = plugin.templateFunctions?.find((f) => f.name === "regex.replace");

  it("should exist", () => {
    expect(replaceFunction).toBeDefined();
  });

  it("should replace one occurrence by default", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "o",
        input: "Hello World",
        replacement: "a",
      },
      purpose: "send",
    });
    expect(result).toBe("Hella World");
  });

  it("should replace with capture groups", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "(\\w+) (\\w+)",
        input: "Hello World",
        replacement: "$2 $1",
      },
      purpose: "send",
    });
    expect(result).toBe("World Hello");
  });

  it("should replace with full match reference", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "World",
        input: "Hello World",
        replacement: "[$&]",
      },
      purpose: "send",
    });
    expect(result).toBe("Hello [World]");
  });

  it("should respect flags parameter", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "hello",
        input: "Hello World",
        replacement: "Hi",
        flags: "i",
      },
      purpose: "send",
    });
    expect(result).toBe("Hi World");
  });

  it("should handle empty replacement", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "World",
        input: "Hello World",
        replacement: "",
      },
      purpose: "send",
    });
    expect(result).toBe("Hello ");
  });

  it("should return original input when no match", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "Goodbye",
        input: "Hello World",
        replacement: "Hi",
      },
      purpose: "send",
    });
    expect(result).toBe("Hello World");
  });

  it("should return empty string when regex is empty", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "",
        input: "Hello World",
        replacement: "Hi",
      },
      purpose: "send",
    });
    expect(result).toBe("");
  });

  it("should return empty string when input is empty", async () => {
    const result = await replaceFunction?.onRender({} as Context, {
      values: {
        regex: "Hello",
        input: "",
        replacement: "Hi",
      },
      purpose: "send",
    });
    expect(result).toBe("");
  });

  it("should throw on invalid regex", async () => {
    const fn = replaceFunction?.onRender({} as Context, {
      values: {
        regex: "[",
        input: "Hello World",
        replacement: "Hi",
      },
      purpose: "send",
    });
    await expect(fn).rejects.toThrow(
      "Invalid regular expression: /[/: Unterminated character class",
    );
  });
});