diff --git a/extensions/document-extract/document-extractor.test.ts b/extensions/document-extract/document-extractor.test.ts index c9cfcd9d73c..5e9dab77a94 100644 --- a/extensions/document-extract/document-extractor.test.ts +++ b/extensions/document-extract/document-extractor.test.ts @@ -36,6 +36,14 @@ import { createPdfDocumentExtractor } from "./document-extractor.js"; const require = createRequire(import.meta.url); +function requireFirstMockArg(mock: ReturnType, 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;