LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-13 15:47:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: guard document extract mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 00:19:06 +01:00
parent 00453a9607
commit 225a374d5e
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/document-extract/document-extractor.test.ts
View File

@@ -36,6 +36,14 @@ import { createPdfDocumentExtractor } from "./document-extractor.js";
const require = createRequire(import.meta.url);
function requireFirstMockArg(mock: ReturnType<typeof vi.fn>, label: string) {
const [call] = mock.mock.calls;
if (!call) {
throw new Error(`Expected ${label}`);
}
return call[0];
}
describe("PDF document extractor", () => {
afterAll(() => {
vi.doUnmock("pdfjs-dist/legacy/build/pdf.mjs");
@@ -92,7 +100,7 @@ describe("PDF document extractor", () => {
});
expect(getDocumentMock).toHaveBeenCalledTimes(1);
const [params] = getDocumentMock.mock.calls[0] ?? [];
const params = requireFirstMockArg(getDocumentMock, "pdfjs getDocument call");
const { data, standardFontDataUrl, ...stableParams } = params as {
data: Uint8Array;
disableWorker: boolean;