import { mkdtemp, rm, writeFile } from "node:fs/promises"; import os from "node:os"; import path from "node:path"; import { describe, expect, it } from "vitest"; import { findMarkdownFileArgument, isMarkdownFilePath, readMarkdownDocument } from "../src/markdown-file.js"; describe("桌面 Markdown 文件参数", () => { it("识别两种 Markdown 扩展名且忽略大小写", () => { expect(isMarkdownFilePath("C:\\docs\\周报.md")).toBe(true); expect(isMarkdownFilePath("C:\\docs\\说明.MARKDOWN")).toBe(true); expect(isMarkdownFilePath("C:\\docs\\说明.txt")).toBe(false); }); it("从启动参数中提取绝对 Markdown 路径", () => { expect( findMarkdownFileArgument( [ "C:\\Program Files\\md-to-pdf\\md-to-pdf.exe", "--original-process-start-time=1", "D:\\文档\\项目周报.md" ], "C:\\Windows" ) ).toBe(path.normalize("D:\\文档\\项目周报.md")); }); it("根据第二实例的工作目录解析相对路径", () => { expect( findMarkdownFileArgument( ["md-to-pdf.exe", "docs\\readme.markdown"], "D:\\项目" ) ).toBe(path.resolve("D:\\项目", "docs\\readme.markdown")); }); it("忽略非 Markdown 文件及 Chromium 参数", () => { expect( findMarkdownFileArgument( ["md-to-pdf.exe", "--inspect", "notes.txt"], "C:\\项目" ) ).toBeUndefined(); }); it("统一读取 Markdown 并限制解码后的字符数", async () => { const directory = await mkdtemp( path.join(os.tmpdir(), "md-to-pdf-markdown-") ); const filePath = path.join(directory, "示例.md"); try { await writeFile(filePath, "# 示例", "utf8"); await expect( readMarkdownDocument(filePath, 20) ).resolves.toEqual({ markdown: "# 示例", fileName: "示例.md" }); await expect( readMarkdownDocument(filePath, 2) ).rejects.toThrow("Markdown 文件过大"); } finally { await rm(directory, { recursive: true, force: true }); } }); it("拒绝非 Markdown 扩展名", async () => { await expect( readMarkdownDocument("C:\\docs\\说明.txt", 100) ).rejects.toThrow("只允许打开 .md 或 .markdown 文件"); }); });