Adding some tests

Author: mikecousins

packages/react-pdf-js/package.json

@@ -29,7 +29,8 @@
   },
   "scripts": {
     "build": "tsup src/index.tsx --dts --format esm,cjs",
-    "lint": "eslint"
+    "lint": "eslint",
+    "test": "vitest"
   },
   "prettier": {
     "printWidth": 80,
@@ -38,15 +39,18 @@
     "trailingComma": "es5"
   },
   "devDependencies": {
+    "@testing-library/jest-dom": "^6.6.3",
     "@testing-library/react": "^16.2.0",
-    "@types/jest": "^29.5.14",
     "@types/react": "^19.0.12",
     "@types/react-dom": "^19.0.4",
     "@vitejs/plugin-react": "^4.3.4",
-    "jest-canvas-mock": "^2.5.2",
+    "jsdom": "^26.0.0",
     "pdfjs-dist": "^5.0.375",
+    "react": "^19.0.0",
+    "react-dom": "^19.0.0",
     "tsup": "^8.4.0",
-    "typescript": "^5.8.2"
+    "typescript": "^5.8.2",
+    "vitest": "^3.2.0"
   },
   "peerDependencies": {
     "pdfjs-dist": "^5",

packages/react-pdf-js/test/index.test.tsx

@@ -0,0 +1,49 @@
+import { vi, expect } from 'vitest';
+import { DocumentInitParameters } from 'pdfjs-dist/types/src/display/api';
+import * as matchers from '@testing-library/jest-dom/matchers';
+
+expect.extend(matchers);
+
+// Mock canvas context
+HTMLCanvasElement.prototype.getContext = vi.fn((contextType: string) => {
+  if (contextType === '2d') {
+    return {
+      scale: vi.fn(),
+    };
+  }
+  return null;
+});
+
+// Mock devicePixelRatio
+Object.defineProperty(window, 'devicePixelRatio', {
+  writable: true,
+  value: 1,
+});
+
+// Mock pdfjs-dist
+vi.mock('pdfjs-dist', () => ({
+  version: '1.0',
+  GlobalWorkerOptions: {
+    workerSrc: '',
+  },
+  getDocument: vi.fn((config: DocumentInitParameters) => ({
+    promise: config.url?.includes('fail_document')
+      ? Promise.reject(new Error('Document load failed'))
+      : Promise.resolve({
+          getPage: vi.fn((pageNum: number) =>
+            config.url?.includes('fail_page')
+              ? Promise.reject(new Error('Page load failed'))
+              : Promise.resolve({
+                  rotate: 0,
+                  getViewport: vi.fn(() => ({ width: 100, height: 100 })),
+                  render: vi.fn(() => ({
+                    promise: config.url?.includes('fail_render')
+                      ? Promise.reject(new Error('Render failed'))
+                      : Promise.resolve(),
+                    cancel: vi.fn(),
+                  })),
+                })
+          ),
+        }),
+  })),
+}));