Updated tests to support new template parameter

Author: m-messer

index.test.ts

@@ -62,6 +62,21 @@ describe("schema", () => {
     const data = [{ userId: "u1", fileName: "doc", typeOfFile: "PDF", markdown: "text", variables: { lang: 42 } }];
     expect(schema.safeParse(data).success).toBe(false);
   });
+
+  it("validates a request with template \"default\"", () => {
+    const data = [{ userId: "u1", fileName: "doc", typeOfFile: "PDF", markdown: "text", template: "default" }];
+    expect(schema.safeParse(data).success).toBe(true);
+  });
+
+  it("validates a request with template \"cjk\"", () => {
+    const data = [{ userId: "u1", fileName: "doc", typeOfFile: "PDF", markdown: "text", template: "cjk" }];
+    expect(schema.safeParse(data).success).toBe(true);
+  });
+
+  it("rejects an unknown template value", () => {
+    const data = [{ userId: "u1", fileName: "doc", typeOfFile: "PDF", markdown: "text", template: "korean" }];
+    expect(schema.safeParse(data).success).toBe(false);
+  });
 });
 
 describe("handler", () => {
@@ -117,6 +132,28 @@ describe("handler", () => {
     expect(calledArgs).toContain("--variable=CJKmainfont:Noto Sans CJK KR");
   });
 
+  it("uses the default template when template is omitted", async () => {
+    const executeMock = vi.fn().mockResolvedValue("");
+    vi.mocked(PdcTs).mockImplementationOnce(() => ({ Execute: executeMock }) as any);
+
+    const event = [{ userId: "user1", fileName: "test-doc", typeOfFile: "TEX", markdown: "# Hello" }];
+    await handler(event as any);
+
+    const calledArgs: string[] = executeMock.mock.calls[0]?.[0]?.pandocArgs ?? [];
+    expect(calledArgs).toContain("--template=./templates/default.latex");
+  });
+
+  it("uses the cjk template when template is \"cjk\"", async () => {
+    const executeMock = vi.fn().mockResolvedValue("");
+    vi.mocked(PdcTs).mockImplementationOnce(() => ({ Execute: executeMock }) as any);
+
+    const event = [{ userId: "user1", fileName: "test-doc", typeOfFile: "TEX", markdown: "# Hello", template: "cjk" }];
+    await handler(event as any);
+
+    const calledArgs: string[] = executeMock.mock.calls[0]?.[0]?.pandocArgs ?? [];
+    expect(calledArgs).toContain("--template=./templates/cjk.latex");
+  });
+
   it("returns 500 when Pandoc execution fails", async () => {
     vi.mocked(PdcTs).mockImplementationOnce(() => ({
       Execute: vi.fn()

src/compile.test.ts

@@ -8,18 +8,19 @@ import { PdcTs } from "pdc-ts";
 // These run the full production pipeline: markdown → Pandoc + template.latex → xelatex → PDF.
 // They are intentionally slow (~5-15s per compile).
 
-// Absolute path so Pandoc can locate the template regardless of its working directory
-const TEMPLATE = path.resolve(__dirname, "template.latex");
+// Absolute paths so Pandoc can locate the templates regardless of its working directory
+const DEFAULT_TEMPLATE = path.resolve(__dirname, "templates/default.latex");
+const CJK_TEMPLATE = path.resolve(__dirname, "templates/cjk.latex");
 
 const pendingPdfs: string[] = [];
 
-const compileToPdf = async (markdown: string, id: string) => {
+const compileToPdf = async (markdown: string, id: string, template: string = DEFAULT_TEMPLATE) => {
   const tmpPath = `/tmp/compile-test-${id}.pdf`;
   pendingPdfs.push(tmpPath);
   await new PdcTs().Execute({
     from: "markdown",
     to: "latex",
-    pandocArgs: ["--pdf-engine=xelatex", `--template=${TEMPLATE}`],
+    pandocArgs: ["--pdf-engine=xelatex", `--template=${template}`],
     spawnOpts: { argv0: "+RTS -M512M -RTS" },
     outputToFile: true,
     sourceText: markdown,
@@ -84,4 +85,22 @@ describe("PDF compile (end-to-end pipeline)", () => {
     },
     { timeout: 60_000 }
   );
+
+  it(
+    "renders Korean text via the cjk template",
+    async () => {
+      const pdf = await compileToPdf(
+        "# 수학 문서\n\n이차 방정식의 판별식은 $\\Delta = b^2 - 4ac$ 입니다.",
+        "korean",
+        CJK_TEMPLATE
+      );
+      const text = extractText(pdf);
+      // pdftotext collapses inter-word spacing between Hangul syllable blocks,
+      // so match the individual words rather than the spaced phrase
+      expect(text).toContain("수학");
+      expect(text).toContain("문서");
+      expect(text).toContain("판별식");
+    },
+    { timeout: 60_000 }
+  );
 });