@workos-inc/authkit-nextjs 2.13.0 → 2.15.0

package/src/authkit-callback-route.spec.ts

@@ -3,19 +3,21 @@ import { handleAuth } from './authkit-callback-route.js';
 import { getSessionFromCookie, saveSession } from './session.js';
 import { NextRequest, NextResponse } from 'next/server';
 
-// Mocked in jest.setup.ts
+// Mocked in vitest.setup.ts
 import { cookies, headers } from 'next/headers';
 
 // Mock dependencies
-const fakeWorkosInstance = {
-  userManagement: {
-    authenticateWithCode: jest.fn(),
-    getJwksUrl: jest.fn(() => 'https://api.workos.com/sso/jwks/client_1234567890'),
+const { fakeWorkosInstance } = vi.hoisted(() => ({
+  fakeWorkosInstance: {
+    userManagement: {
+      authenticateWithCode: vi.fn(),
+      getJwksUrl: vi.fn(() => 'https://api.workos.com/sso/jwks/client_1234567890'),
+    },
   },
-};
+}));
 
-jest.mock('../src/workos', () => ({
-  getWorkOS: jest.fn(() => fakeWorkosInstance),
+vi.mock('../src/workos', () => ({
+  getWorkOS: vi.fn(() => fakeWorkosInstance),
 }));
 
 describe('authkit-callback-route', () => {
@@ -51,12 +53,12 @@ describe('authkit-callback-route', () => {
 
     beforeAll(() => {
       // Silence console.error during tests
-      jest.spyOn(console, 'error').mockImplementation(() => {});
+      vi.spyOn(console, 'error').mockImplementation(() => {});
     });
 
     beforeEach(async () => {
       // Reset all mocks
-      jest.clearAllMocks();
+      vi.clearAllMocks();
 
       // Create a new request with searchParams
       request = new NextRequest(new URL('http://example.com/callback'));
@@ -72,7 +74,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should handle successful authentication', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -89,7 +91,7 @@ describe('authkit-callback-route', () => {
 
     it('should handle authentication failure', async () => {
       // Mock authentication failure
-      (workos.userManagement.authenticateWithCode as jest.Mock).mockRejectedValue(new Error('Auth failed'));
+      (workos.userManagement.authenticateWithCode as Mock).mockRejectedValue(new Error('Auth failed'));
 
       request.nextUrl.searchParams.set('code', 'invalid-code');
 
@@ -103,7 +105,7 @@ describe('authkit-callback-route', () => {
 
     it('should handle authentication failure if a non-Error object is thrown', async () => {
       // Mock authentication failure
-      jest.mocked(workos.userManagement.authenticateWithCode).mockRejectedValue('Auth failed');
+      vi.mocked(workos.userManagement.authenticateWithCode).mockRejectedValue('Auth failed');
 
       request.nextUrl.searchParams.set('code', 'invalid-code');
 
@@ -117,7 +119,7 @@ describe('authkit-callback-route', () => {
 
     it('should handle authentication failure with custom onError handler', async () => {
       // Mock authentication failure
-      jest.mocked(workos.userManagement.authenticateWithCode).mockRejectedValue('Auth failed');
+      vi.mocked(workos.userManagement.authenticateWithCode).mockRejectedValue('Auth failed');
       request.nextUrl.searchParams.set('code', 'invalid-code');
 
       const handler = handleAuth({
@@ -145,7 +147,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should respect custom returnPathname', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       request.nextUrl.searchParams.set('code', 'test-code');
 
@@ -156,7 +158,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should handle state parameter with returnPathname', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       const state = btoa(JSON.stringify({ returnPathname: '/custom-path' }));
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -169,7 +171,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should extract custom search params from returnPathname', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       const state = btoa(JSON.stringify({ returnPathname: '/custom-path?foo=bar&baz=qux' }));
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -182,7 +184,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should handle full URL in returnPathname by extracting only the pathname', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       const state = btoa(JSON.stringify({ returnPathname: 'https://example.com/invite/k0123456789' }));
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -200,7 +202,7 @@ describe('authkit-callback-route', () => {
       const originalRedirect = NextResponse.redirect;
       (NextResponse as Partial<typeof NextResponse>).redirect = undefined;
 
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -232,7 +234,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should use baseURL if provided', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -248,7 +250,7 @@ describe('authkit-callback-route', () => {
         user: { id: 'user_123' },
       };
 
-      (workos.userManagement.authenticateWithCode as jest.Mock).mockResolvedValue(mockAuthResponse);
+      (workos.userManagement.authenticateWithCode as Mock).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -260,12 +262,12 @@ describe('authkit-callback-route', () => {
     });
 
     it('should call onSuccess if provided', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
 
-      const onSuccess = jest.fn();
+      const onSuccess = vi.fn();
       const handler = handleAuth({ onSuccess: onSuccess });
       await handler(request);
 
@@ -276,7 +278,7 @@ describe('authkit-callback-route', () => {
 
     it('should allow onSuccess to update session', async () => {
       const newAccessToken = 'new-access-token';
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Set up request with code
       request.nextUrl.searchParams.set('code', 'test-code');
@@ -293,7 +295,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should pass custom state data to onSuccess callback', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Create state with new format: internal.user
       const internalState = btoa(JSON.stringify({ returnPathname: '/dashboard' }))
@@ -305,7 +307,7 @@ describe('authkit-callback-route', () => {
       request.nextUrl.searchParams.set('code', 'test-code');
       request.nextUrl.searchParams.set('state', state);
 
-      const onSuccess = jest.fn();
+      const onSuccess = vi.fn();
       const handler = handleAuth({ onSuccess });
       await handler(request);
 
@@ -323,7 +325,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should handle state without custom data', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // State with only returnPathname
       const state = btoa(JSON.stringify({ returnPathname: '/profile' }));
@@ -331,7 +333,7 @@ describe('authkit-callback-route', () => {
       request.nextUrl.searchParams.set('code', 'test-code');
       request.nextUrl.searchParams.set('state', state);
 
-      const onSuccess = jest.fn();
+      const onSuccess = vi.fn();
       const handler = handleAuth({ onSuccess });
       await handler(request);
 
@@ -345,7 +347,7 @@ describe('authkit-callback-route', () => {
     });
 
     it('should handle backward compatibility with old state format', async () => {
-      jest.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
+      vi.mocked(workos.userManagement.authenticateWithCode).mockResolvedValue(mockAuthResponse);
 
       // Old format: just returnPathname
       const state = btoa(JSON.stringify({ returnPathname: '/old-path' }));

package/src/components/authkit-provider.spec.tsx

@@ -10,17 +10,17 @@ import {
   switchToOrganizationAction,
 } from '../actions.js';
 
-jest.mock('../actions', () => ({
-  checkSessionAction: jest.fn(),
-  getAuthAction: jest.fn(),
-  refreshAuthAction: jest.fn(),
-  handleSignOutAction: jest.fn(),
-  switchToOrganizationAction: jest.fn(),
+vi.mock('../actions', () => ({
+  checkSessionAction: vi.fn(),
+  getAuthAction: vi.fn(),
+  refreshAuthAction: vi.fn(),
+  handleSignOutAction: vi.fn(),
+  switchToOrganizationAction: vi.fn(),
 }));
 
 describe('AuthKitProvider', () => {
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
   });
 
   it('should render children', async () => {
@@ -134,7 +134,7 @@ describe('AuthKitProvider', () => {
   });
 
   it('should call getAuthAction when initialAuth is not provided', async () => {
-    (getAuthAction as jest.Mock).mockResolvedValueOnce({
+    (getAuthAction as Mock).mockResolvedValueOnce({
       user: { email: 'test@example.com' },
       sessionId: 'test-session',
     });
@@ -151,7 +151,7 @@ describe('AuthKitProvider', () => {
   });
 
   it('should do nothing if onSessionExpired is false', async () => {
-    jest.spyOn(window, 'addEventListener');
+    vi.spyOn(window, 'addEventListener');
 
     await act(async () => {
       render(
@@ -166,8 +166,8 @@ describe('AuthKitProvider', () => {
   });
 
   it('should call onSessionExpired when session is expired', async () => {
-    (checkSessionAction as jest.Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
-    const onSessionExpired = jest.fn();
+    (checkSessionAction as Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
+    const onSessionExpired = vi.fn();
 
     render(
       <AuthKitProvider onSessionExpired={onSessionExpired}>
@@ -186,8 +186,8 @@ describe('AuthKitProvider', () => {
   });
 
   it('should only call onSessionExpired once if multiple visibility changes occur', async () => {
-    (checkSessionAction as jest.Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
-    const onSessionExpired = jest.fn();
+    (checkSessionAction as Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
+    const onSessionExpired = vi.fn();
 
     render(
       <AuthKitProvider onSessionExpired={onSessionExpired}>
@@ -207,9 +207,9 @@ describe('AuthKitProvider', () => {
   });
 
   it('should pass through if checkSessionAction does not throw "Failed to fetch"', async () => {
-    (checkSessionAction as jest.Mock).mockResolvedValueOnce(false);
+    (checkSessionAction as Mock).mockResolvedValueOnce(false);
 
-    const onSessionExpired = jest.fn();
+    const onSessionExpired = vi.fn();
 
     render(
       <AuthKitProvider onSessionExpired={onSessionExpired}>
@@ -234,7 +234,7 @@ describe('AuthKitProvider', () => {
       originalLocation = window.location;
       // @ts-expect-error - deleting window.location to mock it
       delete window.location;
-      window.location = { reload: jest.fn() } as unknown as Location;
+      window.location = { reload: vi.fn() } as unknown as Location;
     });
 
     afterEach(() => {
@@ -242,7 +242,7 @@ describe('AuthKitProvider', () => {
     });
 
     it('should reload the page when session is expired and no onSessionExpired handler is provided', async () => {
-      (checkSessionAction as jest.Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
+      (checkSessionAction as Mock).mockRejectedValueOnce(new Error('Failed to fetch'));
 
       render(
         <AuthKitProvider>
@@ -261,8 +261,8 @@ describe('AuthKitProvider', () => {
     });
 
     it('should not call onSessionExpired or reload the page if session is valid', async () => {
-      (checkSessionAction as jest.Mock).mockResolvedValueOnce(true);
-      const onSessionExpired = jest.fn();
+      (checkSessionAction as Mock).mockResolvedValueOnce(true);
+      const onSessionExpired = vi.fn();
 
       render(
         <AuthKitProvider onSessionExpired={onSessionExpired}>
@@ -285,12 +285,12 @@ describe('AuthKitProvider', () => {
 
 describe('useAuth', () => {
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
   });
 
   it('should call getAuth when a user is not returned when ensureSignedIn is true', async () => {
     // First and second calls return no user, second call returns a user
-    (getAuthAction as jest.Mock)
+    (getAuthAction as Mock)
       .mockResolvedValueOnce({ user: null, loading: true })
       .mockResolvedValueOnce({ user: { email: 'test@example.com' }, loading: false });
 
@@ -319,7 +319,7 @@ describe('useAuth', () => {
     };
 
     // Suppress console.error for this test since we expect an error
-    const consoleSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
+    const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => {});
 
     expect(() => {
       render(<TestComponent />);
@@ -329,7 +329,7 @@ describe('useAuth', () => {
   });
 
   it('should provide auth context values when used within AuthKitProvider', async () => {
-    (getAuthAction as jest.Mock).mockResolvedValueOnce({
+    (getAuthAction as Mock).mockResolvedValueOnce({
       user: { email: 'test@example.com' },
       sessionId: 'test-session',
       organizationId: 'test-org',
@@ -377,8 +377,8 @@ describe('useAuth', () => {
       sessionId: 'test-session',
     };
 
-    (getAuthAction as jest.Mock).mockResolvedValueOnce(mockAuth);
-    (refreshAuthAction as jest.Mock).mockResolvedValueOnce({
+    (getAuthAction as Mock).mockResolvedValueOnce(mockAuth);
+    (refreshAuthAction as Mock).mockResolvedValueOnce({
       ...mockAuth,
       sessionId: 'new-session',
     });
@@ -420,10 +420,10 @@ describe('useAuth', () => {
       organizationId: 'new-org',
     };
 
-    (getAuthAction as jest.Mock)
+    (getAuthAction as Mock)
       .mockResolvedValue(mockAuth)
       .mockResolvedValueOnce({ ...mockAuth, organizationId: 'old-org' });
-    (switchToOrganizationAction as jest.Mock).mockResolvedValueOnce(mockAuth);
+    (switchToOrganizationAction as Mock).mockResolvedValueOnce(mockAuth);
 
     const TestComponent = () => {
       const auth = useAuth();
@@ -456,7 +456,7 @@ describe('useAuth', () => {
   });
 
   it('should receive an error when refreshAuth fails with an error', async () => {
-    (refreshAuthAction as jest.Mock).mockRejectedValueOnce(new Error('Refresh failed'));
+    (refreshAuthAction as Mock).mockRejectedValueOnce(new Error('Refresh failed'));
 
     let error: string | undefined;
 
@@ -493,7 +493,7 @@ describe('useAuth', () => {
   });
 
   it('should receive an error when refreshAuth fails with a string error', async () => {
-    (refreshAuthAction as jest.Mock).mockRejectedValueOnce('Refresh failed');
+    (refreshAuthAction as Mock).mockRejectedValueOnce('Refresh failed');
 
     let error: string | undefined;
 
@@ -530,7 +530,7 @@ describe('useAuth', () => {
   });
 
   it('should call handleSignOutAction when signOut is called', async () => {
-    (handleSignOutAction as jest.Mock).mockResolvedValueOnce({});
+    (handleSignOutAction as Mock).mockResolvedValueOnce({});
 
     const TestComponent = () => {
       const auth = useAuth();
@@ -556,7 +556,7 @@ describe('useAuth', () => {
   });
 
   it('should pass returnTo parameter to handleSignOutAction', async () => {
-    (handleSignOutAction as jest.Mock).mockResolvedValueOnce({});
+    (handleSignOutAction as Mock).mockResolvedValueOnce({});
 
     const TestComponent = () => {
       const auth = useAuth();

package/src/components/button.spec.tsx

@@ -24,12 +24,10 @@ describe('Button', () => {
     const { getByRole } = render(<Button style={{ backgroundColor: 'red' }}>Click me</Button>);
     const button = getByRole('button');
 
-    expect(button).toHaveStyle({
-      backgroundColor: 'red',
-      display: 'inline-flex',
-      alignItems: 'center',
-      justifyContent: 'center',
-    });
+    expect(button.style.backgroundColor).toBe('red');
+    expect(button.style.display).toBe('inline-flex');
+    expect(button.style.alignItems).toBe('center');
+    expect(button.style.justifyContent).toBe('center');
   });
 
   it('should pass through additional props', () => {

package/src/components/impersonation.spec.tsx

@@ -7,23 +7,23 @@ import * as React from 'react';
 import { handleSignOutAction } from '../actions.js';
 
 // Mock the useAuth hook
-jest.mock('./authkit-provider', () => ({
-  useAuth: jest.fn(),
+vi.mock('./authkit-provider', () => ({
+  useAuth: vi.fn(),
 }));
 
 // Mock the getOrganizationAction
-jest.mock('../actions', () => ({
-  getOrganizationAction: jest.fn(),
-  handleSignOutAction: jest.fn(),
+vi.mock('../actions', () => ({
+  getOrganizationAction: vi.fn(),
+  handleSignOutAction: vi.fn(),
 }));
 
 describe('Impersonation', () => {
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
   });
 
   it('should return null if not impersonating', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: null,
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -34,7 +34,7 @@ describe('Impersonation', () => {
   });
 
   it('should render impersonation banner when impersonating', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -45,13 +45,13 @@ describe('Impersonation', () => {
   });
 
   it('should render with organization info when organizationId is provided', async () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: 'org_123',
     });
 
-    (getOrganizationAction as jest.Mock).mockResolvedValue({
+    (getOrganizationAction as Mock).mockResolvedValue({
       id: 'org_123',
       name: 'Test Org',
     });
@@ -64,7 +64,7 @@ describe('Impersonation', () => {
   });
 
   it('should render at the bottom by default', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -76,7 +76,7 @@ describe('Impersonation', () => {
   });
 
   it('should render at the top when side prop is "top"', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -88,7 +88,7 @@ describe('Impersonation', () => {
   });
 
   it('should merge custom styles with default styles', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -97,11 +97,11 @@ describe('Impersonation', () => {
     const customStyle = { backgroundColor: 'red' };
     const { container } = render(<Impersonation style={customStyle} />);
     const root = container.querySelector('[data-workos-impersonation-root]');
-    expect(root).toHaveStyle({ backgroundColor: 'red' });
+    expect((root as HTMLElement).style.backgroundColor).toBe('red');
   });
 
   it('should should sign out when the Stop button is called', async () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -114,7 +114,7 @@ describe('Impersonation', () => {
   });
 
   it('should pass returnTo prop to handleSignOutAction when provided', async () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -129,7 +129,7 @@ describe('Impersonation', () => {
   });
 
   it('should not call getOrganizationAction when organizationId is not provided', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: { id: '123', email: 'user@example.com' },
       organizationId: null,
@@ -140,7 +140,7 @@ describe('Impersonation', () => {
   });
 
   it('should not call getOrganizationAction when impersonator is not present', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: null,
       user: { id: '123', email: 'user@example.com' },
       organizationId: 'org_123',
@@ -151,7 +151,7 @@ describe('Impersonation', () => {
   });
 
   it('should not call getOrganizationAction when user is not present', () => {
-    (useAuth as jest.Mock).mockReturnValue({
+    (useAuth as Mock).mockReturnValue({
       impersonator: { email: 'admin@example.com' },
       user: null,
       organizationId: 'org_123',
@@ -167,10 +167,10 @@ describe('Impersonation', () => {
       name: 'Test Org',
     };
 
-    (getOrganizationAction as jest.Mock).mockResolvedValue(mockOrg);
+    (getOrganizationAction as Mock).mockResolvedValue(mockOrg);
 
     const { rerender } = await act(async () => {
-      (useAuth as jest.Mock).mockReturnValue({
+      (useAuth as Mock).mockReturnValue({
         impersonator: { email: 'admin@example.com' },
         user: { id: '123', email: 'user@example.com' },
         organizationId: 'org_123',
@@ -188,7 +188,7 @@ describe('Impersonation', () => {
 
     // Rerender with the same organizationId
     await act(async () => {
-      (useAuth as jest.Mock).mockReturnValue({
+      (useAuth as Mock).mockReturnValue({
         impersonator: { email: 'admin@example.com' },
         user: { id: '123', email: 'user@example.com' },
         organizationId: 'org_123',
@@ -212,10 +212,10 @@ describe('Impersonation', () => {
       name: 'Test Org 2',
     };
 
-    (getOrganizationAction as jest.Mock).mockResolvedValueOnce(mockOrg1).mockResolvedValueOnce(mockOrg2);
+    (getOrganizationAction as Mock).mockResolvedValueOnce(mockOrg1).mockResolvedValueOnce(mockOrg2);
 
     const { rerender } = await act(async () => {
-      (useAuth as jest.Mock).mockReturnValue({
+      (useAuth as Mock).mockReturnValue({
         impersonator: { email: 'admin@example.com' },
         user: { id: '123', email: 'user@example.com' },
         organizationId: 'org_123',
@@ -234,7 +234,7 @@ describe('Impersonation', () => {
 
     // Rerender with a different organizationId
     await act(async () => {
-      (useAuth as jest.Mock).mockReturnValue({
+      (useAuth as Mock).mockReturnValue({
         impersonator: { email: 'admin@example.com' },
         user: { id: '123', email: 'user@example.com' },
         organizationId: 'org_456',

package/src/components/min-max-button.spec.tsx

@@ -14,7 +14,7 @@ describe('MinMaxButton', () => {
   afterEach(() => {
     // Clean up after each test
     document.body.innerHTML = '';
-    jest.restoreAllMocks();
+    vi.restoreAllMocks();
   });
 
   it('sets minimized value when clicked', () => {
@@ -34,7 +34,7 @@ describe('MinMaxButton', () => {
     const root = document.querySelector('[data-workos-impersonation-root]');
 
     // Mock querySelector to return null for this test
-    jest.spyOn(document, 'querySelector').mockReturnValue(null);
+    vi.spyOn(document, 'querySelector').mockReturnValue(null);
 
     act(() => {
       getByRole('button').click();