156 lines
3.8 KiB
TypeScript
156 lines
3.8 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
||
import type { WebContents } from "electron";
|
||
import {
|
||
createPagedDocumentPayload,
|
||
defaultExportConfig,
|
||
type DocxMediaCapturePlan
|
||
} from "@md-to-pdf/core";
|
||
import { captureDocxMediaWithElectronWebContents } from "../src/electron-docx-media-capture.js";
|
||
|
||
const plan: DocxMediaCapturePlan = {
|
||
targets: [
|
||
{
|
||
id: "docx-media-1",
|
||
kind: "image",
|
||
ordinal: 1,
|
||
kindOrdinal: 1,
|
||
altText: "图片",
|
||
alignment: "center",
|
||
displayWidthPx: 200,
|
||
displayHeightPx: 100,
|
||
captureX: 12,
|
||
captureY: 24,
|
||
captureWidthPx: 200,
|
||
captureHeightPx: 100,
|
||
rasterScale: 3.125
|
||
}
|
||
],
|
||
echartsErrors: [],
|
||
mermaidErrors: []
|
||
};
|
||
|
||
const request = {
|
||
payload: createPagedDocumentPayload({
|
||
document: {
|
||
rendererVersion: 1,
|
||
articleHtml: '<article id="write"></article>',
|
||
bodyHtml: "",
|
||
metadata: {
|
||
title: "",
|
||
author: "",
|
||
subject: "",
|
||
keywords: [],
|
||
language: "zh-CN"
|
||
},
|
||
semanticDocument: {
|
||
schemaVersion: 1,
|
||
titlePolicy: {
|
||
metadataTitle: "emit",
|
||
firstBodyHeading: "keep"
|
||
},
|
||
regions: []
|
||
},
|
||
features: [],
|
||
warnings: []
|
||
},
|
||
fileName: "test.md",
|
||
themeCss: "",
|
||
exportConfig: defaultExportConfig
|
||
}),
|
||
dimensions: {
|
||
contentWidthPx: 600,
|
||
contentHeightPx: 900
|
||
}
|
||
};
|
||
|
||
describe("Electron DOCX 媒体捕获", () => {
|
||
it("复用隐藏 Chromium 并在捕获后释放调试会话", async () => {
|
||
let attached = false;
|
||
const attach = vi.fn(() => {
|
||
attached = true;
|
||
});
|
||
const detach = vi.fn(() => {
|
||
attached = false;
|
||
});
|
||
const sendCommand = vi.fn(async () => ({
|
||
data: Buffer.from("png").toString("base64")
|
||
}));
|
||
const webContents = {
|
||
executeJavaScript: vi.fn(async () => plan),
|
||
debugger: {
|
||
isAttached: vi.fn(() => attached),
|
||
attach,
|
||
detach,
|
||
sendCommand
|
||
}
|
||
} as unknown as WebContents;
|
||
|
||
const result = await captureDocxMediaWithElectronWebContents(
|
||
webContents,
|
||
request
|
||
);
|
||
|
||
expect(result.captures[0]).toEqual({
|
||
id: "docx-media-1",
|
||
png: Buffer.from("png")
|
||
});
|
||
expect(attach).toHaveBeenCalledWith("1.3");
|
||
expect(sendCommand).toHaveBeenCalledWith(
|
||
"Page.captureScreenshot",
|
||
expect.objectContaining({
|
||
format: "png",
|
||
clip: expect.objectContaining({ scale: 3.125 })
|
||
})
|
||
);
|
||
expect(detach).toHaveBeenCalledOnce();
|
||
});
|
||
|
||
it("媒体布局永久不返回时报告局部超时", async () => {
|
||
const webContents = {
|
||
executeJavaScript: vi.fn(() => new Promise(() => undefined)),
|
||
debugger: {
|
||
isAttached: vi.fn(() => false)
|
||
}
|
||
} as unknown as WebContents;
|
||
|
||
await expect(
|
||
captureDocxMediaWithElectronWebContents(
|
||
webContents,
|
||
request,
|
||
undefined,
|
||
{ renderMs: 10, screenshotMs: 10 }
|
||
)
|
||
).rejects.toThrow("桌面 DOCX 媒体布局超过 10ms");
|
||
});
|
||
|
||
it("单个 PNG 永久不返回时报告媒体编号和类型", async () => {
|
||
let attached = false;
|
||
const detach = vi.fn(() => {
|
||
attached = false;
|
||
});
|
||
const webContents = {
|
||
executeJavaScript: vi.fn(async () => plan),
|
||
debugger: {
|
||
isAttached: vi.fn(() => attached),
|
||
attach: vi.fn(() => {
|
||
attached = true;
|
||
}),
|
||
detach,
|
||
sendCommand: vi.fn(() => new Promise(() => undefined))
|
||
}
|
||
} as unknown as WebContents;
|
||
|
||
await expect(
|
||
captureDocxMediaWithElectronWebContents(
|
||
webContents,
|
||
request,
|
||
undefined,
|
||
{ renderMs: 10, screenshotMs: 10 }
|
||
)
|
||
).rejects.toThrow(
|
||
"桌面 DOCX 第 1/1 个媒体（image）PNG 捕获超过 10ms"
|
||
);
|
||
expect(detach).toHaveBeenCalledOnce();
|
||
});
|
||
});
|