move jest to vitest

Author: yuming-long

Makefile

@@ -34,12 +34,12 @@ check:
 ## test-unit:					run unit tests
 .PHONY: test-unit
 test-unit:
-	npx jest --verbose --detectOpenHandles --config jest.config.js test/unit
+	npx vitest --dir test/unit --run --reporter verbose --config vitest.config.ts
 
 ## test-integration:			run integration tests
 .PHONY: test-integration
 test-integration:
-	npx jest --verbose --detectOpenHandles --config jest.config.js test/integration --forceExit
+	npx vitest --dir test/integration --run --reporter verbose --config vitest.config.ts
 
 ## test:					run all tests
 .PHONY: test

jest.config.js

@@ -3,6 +3,7 @@ import { readFileSync } from "fs";
 import { UnstructuredClient } from "../../src";
 import { PartitionResponse } from "../../src/sdk/models/operations";
 import { PartitionParameters, Strategy } from "../../src/sdk/models/shared";
+import { describe, it, expect} from 'vitest';
 
 const localServer = "http://localhost:8000"
 

test/integration/HttpsCheckHook.test.ts

@@ -3,14 +3,15 @@ import { readFileSync } from "fs";
 import { UnstructuredClient } from "../../src";
 import { PartitionResponse } from "../../src/sdk/models/operations";
 import { PartitionParameters, Strategy } from "../../src/sdk/models/shared";
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
 
 const localServer = "http://localhost:8000";
 
 describe("SplitPdfHook integration tests check splitted file is same as not splitted", () => {
   const FAKE_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
-  const consoleInfoSpy = jest.spyOn(console, "info");
-  const consoleWarnSpy = jest.spyOn(console, "warn");
-  const consoleErrorSpy = jest.spyOn(console, "error");
+  const consoleInfoSpy = vi.spyOn(console, "info");
+  const consoleWarnSpy = vi.spyOn(console, "warn");
+  const consoleErrorSpy = vi.spyOn(console, "error");
 
   let client = new UnstructuredClient({
     serverURL: localServer,
@@ -273,9 +274,9 @@ describe("SplitPdfHook integration tests check splitted file is same as not spli
 
 describe("SplitPdfHook integration tests page range parameter", () => {
   const FAKE_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
-  const consoleInfoSpy = jest.spyOn(console, "info");
-  const consoleWarnSpy = jest.spyOn(console, "warn");
-  const consoleErrorSpy = jest.spyOn(console, "error");
+  const consoleInfoSpy = vi.spyOn(console, "info");
+  const consoleWarnSpy = vi.spyOn(console, "warn");
+  const consoleErrorSpy = vi.spyOn(console, "error");
 
   let client = new UnstructuredClient({
     serverURL: localServer,

test/integration/SplitPdfHook.test.ts

@@ -1,8 +1,9 @@
 import { HttpsCheckHook } from "../../src/hooks/custom/HttpsCheckHook";
 import { HTTPClient } from "../../src/lib/http";
+import { describe, it, expect, vi, afterEach } from 'vitest';
 
 describe("HttpsCheckHook", () => {
-  const consoleSpy = jest.spyOn(global.console, "warn");
+  const consoleSpy = vi.spyOn(global.console, "warn");
 
   afterEach(() => {
     consoleSpy.mockClear();

test/unit/HttpsCheckHook.test.ts

@@ -1,5 +1,6 @@
 import { PDFDocument } from "pdf-lib";
 import { readFileSync } from "node:fs";
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
 
 import {
   loadPdf,
@@ -20,14 +21,14 @@ describe("Pdf utility functions", () => {
   });
 
   afterEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
   });
 
   describe("pdfPagesToBlob", () => {
     it("should convert range of pages to a Blob object", async () => {
-      const copyMock = jest.spyOn(PDFDocument.prototype, "copyPages");
-      const saveMock = jest.spyOn(PDFDocument.prototype, "save");
-      const addMock = jest.spyOn(PDFDocument.prototype, "addPage");
+      const copyMock = vi.spyOn(PDFDocument.prototype, "copyPages");
+      const saveMock = vi.spyOn(PDFDocument.prototype, "save");
+      const addMock = vi.spyOn(PDFDocument.prototype, "addPage");
 
       // Call the method
       const result = await pdfPagesToBlob(pdf, 4, 8);
@@ -106,7 +107,7 @@ describe("Pdf utility functions", () => {
     it("should return true, null, and 0 if the file is not a PDF", async () => {
       const file = {
         name: "document.txt",
-        content: jest.fn().mockResolvedValue(new ArrayBuffer(0)),
+        content: vi.fn().mockResolvedValue(new ArrayBuffer(0)),
       };
 
       const result = await loadPdf(file as any);
@@ -118,7 +119,7 @@ describe("Pdf utility functions", () => {
     it("should return true, null, and 0 if the file is not a PDF without basing on file extension", async () => {
       const file = {
         name: "uuid1234",
-        content: jest.fn().mockResolvedValue(new ArrayBuffer(0)),
+        content: vi.fn().mockResolvedValue(new ArrayBuffer(0)),
       };
 
       const result = await loadPdf(file as any);
@@ -131,7 +132,7 @@ describe("Pdf utility functions", () => {
     it("should return true, null, and 0 if there is an error while loading the PDF", async () => {
       const file = {
         name: "document.pdf",
-        arrayBuffer: jest.fn().mockRejectedValue(new ArrayBuffer(0)),
+        arrayBuffer: vi.fn().mockRejectedValue(new ArrayBuffer(0)),
       };
 
       const result = await loadPdf(file as any);
@@ -147,7 +148,7 @@ describe("Pdf utility functions", () => {
         arrayBuffer: () => file.buffer,
       };
 
-      const loadMock = jest.spyOn(PDFDocument, "load");
+      const loadMock = vi.spyOn(PDFDocument, "load");
 
       const [error, _, pages] = await loadPdf(f as any);
 
@@ -164,8 +165,8 @@ describe("Pdf utility functions", () => {
         arrayBuffer: () => file.buffer,
       };
 
-      jest.clearAllMocks(); // Reset Mocks Between Tests
-      const loadMock = jest.spyOn(PDFDocument, "load");
+      vi.clearAllMocks(); // Reset Mocks Between Tests
+      const loadMock = vi.spyOn(PDFDocument, "load");
 
       const [error, _, pages] = await loadPdf(f as any);
 

test/unit/utils/pdf.test.ts

@@ -0,0 +1,17 @@
+/// <reference types="vitest" />
+
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  esbuild: {
+    // Transpile all files with ESBuild to remove comments from code coverage.
+    // Required for `test.coverage.ignoreEmptyLines` to work:
+    include: ['**/*.js', '**/*.jsx', '**/*.mjs', '**/*.ts', '**/*.tsx'],
+  },
+  test: {
+    coverage: {
+      provider: 'v8',
+      ignoreEmptyLines: true,
+    },
+  },
+})