From 553f124de313b5cc4aabf555579c32c1cda6e651 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bj=C3=B8rge=20N=C3=A6ss?= <bjoerge@gmail.com>
Date: Tue, 15 Oct 2024 15:29:33 +0200
Subject: [PATCH] test(sanity): migrate tests to vitest

---
 .../core/preview/utils/replayLatest.test.ts   |  2 +-
 .../__tests__/ReleasesMenu.test.tsx           | 30 +++++-----
 .../__tests__/ReleaseDetailsDialog.test.tsx   | 49 ++++++++-------
 .../dialog/__tests__/ReleaseForm.test.tsx     | 26 ++++----
 .../ReleaseIconEditorPicker.test.tsx          | 13 ++--
 .../__tests__/useDocumentVersions.test.tsx    | 32 +++++-----
 .../__tests__/ReleaseMenuButton.test.tsx      | 23 +++----
 .../detail/__tests__/ReleaseDetail.test.tsx   | 60 ++++++++++---------
 .../detail/__tests__/ReleaseReview.test.tsx   | 33 ++++------
 .../detail/__tests__/ReleaseSummary.test.tsx  | 35 ++++-------
 .../__tests__/ReleasesOverview.test.tsx       | 40 +++++++------
 .../src/core/releases/util/util.test.ts       |  2 +-
 .../utils/fetchFeatureToggle.test.ts          |  3 +-
 .../__tests__/resolvePerspectives.test.ts     |  2 +-
 .../structureTool/StructureTitle.test.tsx     | 15 ++---
 .../__tests__/DeletedDocumentBanners.test.tsx | 30 +++++-----
 .../__tests__/DocumentHeaderTitle.test.tsx    | 37 +++++-------
 .../DocumentPerspectiveMenu.test.tsx          | 39 ++++++------
 .../__tests__/PaneContainer.test.tsx          | 16 +++--
 .../__tests__/DocumentSheetListPane.test.tsx  |  4 +-
 .../__tests__/useDocumentSheetColumns.test.ts |  4 +-
 21 files changed, 235 insertions(+), 260 deletions(-)

diff --git a/packages/sanity/src/core/preview/utils/replayLatest.test.ts b/packages/sanity/src/core/preview/utils/replayLatest.test.ts
index 03796d77b38..5b0b52f3f21 100644
--- a/packages/sanity/src/core/preview/utils/replayLatest.test.ts
+++ b/packages/sanity/src/core/preview/utils/replayLatest.test.ts
@@ -1,6 +1,6 @@
-import {expect, test} from '@jest/globals'
 import {concat, from, lastValueFrom, of, share, timer} from 'rxjs'
 import {concatMap, delay, mergeMap, take, toArray} from 'rxjs/operators'
+import {expect, test} from 'vitest'
 
 import {shareReplayLatest} from './shareReplayLatest'
 
diff --git a/packages/sanity/src/core/releases/components/__tests__/ReleasesMenu.test.tsx b/packages/sanity/src/core/releases/components/__tests__/ReleasesMenu.test.tsx
index 27a8816fb9e..5e98b37808b 100644
--- a/packages/sanity/src/core/releases/components/__tests__/ReleasesMenu.test.tsx
+++ b/packages/sanity/src/core/releases/components/__tests__/ReleasesMenu.test.tsx
@@ -1,8 +1,8 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen, within} from '@testing-library/react'
 import userEvent from '@testing-library/user-event'
 import {act} from 'react'
 import {type BundleDocument, useBundles} from 'sanity'
+import {beforeEach, describe, expect, it, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../test/testUtils/TestProvider'
 import {Button} from '../../../../ui-components'
@@ -10,25 +10,25 @@ import {usePerspective} from '../../hooks/usePerspective'
 import {LATEST} from '../../util/const'
 import {ReleasesMenu} from '../ReleasesMenu'
 
-jest.mock('../../hooks/usePerspective', () => ({
-  usePerspective: jest.fn().mockReturnValue({
+vi.mock('../../hooks/usePerspective', () => ({
+  usePerspective: vi.fn().mockReturnValue({
     currentGlobalBundle: {},
-    setPerspective: jest.fn(),
+    setPerspective: vi.fn(),
   }),
 }))
 
-jest.mock('../../util/util', () => ({
-  isDraftOrPublished: jest.fn(),
+vi.mock('../../util/util', () => ({
+  isDraftOrPublished: vi.fn(),
 }))
 
-jest.mock('../../../store/bundles/useBundles', () => ({
-  useBundles: jest.fn().mockReturnValue({deletedBundles: {}}),
+vi.mock('../../../store/bundles/useBundles', () => ({
+  useBundles: vi.fn().mockReturnValue({deletedBundles: {}}),
 }))
 
-const mockUseBundles = useBundles as jest.Mock<typeof useBundles>
+const mockUseBundles = useBundles as Mock<typeof useBundles>
 
 describe('ReleasesMenu', () => {
-  const mockUsePerspective = usePerspective as jest.Mock
+  const mockUsePerspective = usePerspective as Mock
   const ButtonTest = <Button text="Button Test" />
   const mockBundles: BundleDocument[] = [
     {
@@ -123,7 +123,7 @@ describe('ReleasesMenu', () => {
   it('should render latest bundle menu item as selected when currentGlobalBundle is LATEST', async () => {
     mockUsePerspective.mockReturnValue({
       currentGlobalBundle: LATEST,
-      setPerspective: jest.fn(),
+      setPerspective: vi.fn(),
     })
 
     const wrapper = await createTestProvider()
@@ -142,7 +142,7 @@ describe('ReleasesMenu', () => {
   it('should render bundle as selected when currentGlobalBundle is that bundle', async () => {
     mockUsePerspective.mockReturnValue({
       currentGlobalBundle: mockBundles[0],
-      setPerspective: jest.fn(),
+      setPerspective: vi.fn(),
     })
 
     const wrapper = await createTestProvider()
@@ -174,7 +174,7 @@ describe('ReleasesMenu', () => {
   })
 
   it('should call setPerspective when a bundle menu item is clicked', async () => {
-    const setPerspective = jest.fn()
+    const setPerspective = vi.fn()
     mockUsePerspective.mockReturnValue({
       currentGlobalBundle: LATEST,
       setPerspective,
@@ -213,7 +213,7 @@ describe('ReleasesMenu', () => {
 
   it('should not show deleted bundles when not included in the list', async () => {
     mockUseBundles.mockReturnValue({
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       loading: false,
       data: [],
       deletedBundles: {
@@ -238,7 +238,7 @@ describe('ReleasesMenu', () => {
 
   it('should show deleted bundles that are included in the list', async () => {
     mockUseBundles.mockReturnValue({
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       loading: false,
       data: [],
       deletedBundles: {
diff --git a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseDetailsDialog.test.tsx b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseDetailsDialog.test.tsx
index 64e664f0cea..9da58d82e99 100644
--- a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseDetailsDialog.test.tsx
+++ b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseDetailsDialog.test.tsx
@@ -1,44 +1,47 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen, waitFor, within} from '@testing-library/react'
 import {type BundleDocument, useBundles} from 'sanity'
+import {afterEach, beforeEach, describe, expect, it, type Mock, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
 import {useBundleOperations} from '../../../../store/bundles/useBundleOperations'
 import {usePerspective} from '../../../hooks/usePerspective'
 import {ReleaseDetailsDialog} from '../ReleaseDetailsDialog'
 
-/*jest.mock('../../../../../core/hooks/useDateTimeFormat', () => ({
-  useDateTimeFormat: jest.fn(),
+/*vi.mock('../../../../../core/hooks/useDateTimeFormat', () => ({
+  useDateTimeFormat: vi.fn(),
 }))*/
 
-jest.mock('../../../../store/bundles', () => ({
-  useBundles: jest.fn(),
+vi.mock('../../../../store/bundles', () => ({
+  useBundles: vi.fn(),
 }))
 
-jest.mock('../../../../store/bundles/useBundleOperations', () => ({
-  useBundleOperations: jest.fn().mockReturnValue({
-    createBundle: jest.fn(),
-    updateBundle: jest.fn(),
+vi.mock('../../../../store/bundles/useBundleOperations', () => ({
+  useBundleOperations: vi.fn().mockReturnValue({
+    createBundle: vi.fn(),
+    updateBundle: vi.fn(),
   }),
 }))
 
-jest.mock('../../../hooks/usePerspective', () => ({
-  usePerspective: jest.fn().mockReturnValue({
-    setPerspective: jest.fn(),
+vi.mock('../../../hooks/usePerspective', () => ({
+  usePerspective: vi.fn().mockReturnValue({
+    setPerspective: vi.fn(),
   }),
 }))
 
-const mockUseBundleStore = useBundles as jest.Mock<typeof useBundles>
-//const mockUseDateTimeFormat = useDateTimeFormat as jest.Mock
+const mockUseBundleStore = useBundles as Mock<typeof useBundles>
+//const mockUseDateTimeFormat = useDateTimeFormat as Mock
 
 describe('ReleaseDetailsDialog', () => {
   beforeEach(() => {
-    jest.clearAllMocks()
+    vi.clearAllMocks()
+  })
+  afterEach(() => {
+    vi.clearAllMocks()
   })
 
   describe('when creating a new bundle', () => {
-    const onCancelMock = jest.fn()
-    const onSubmitMock = jest.fn()
+    const onCancelMock = vi.fn()
+    const onSubmitMock = vi.fn()
 
     beforeEach(async () => {
       onCancelMock.mockClear()
@@ -47,12 +50,12 @@ describe('ReleaseDetailsDialog', () => {
       mockUseBundleStore.mockReturnValue({
         data: [],
         loading: true,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         error: undefined,
         deletedBundles: {},
       })
 
-      //mockUseDateTimeFormat.mockReturnValue({format: jest.fn().mockReturnValue('Mocked date')})
+      //mockUseDateTimeFormat.mockReturnValue({format: vi.fn().mockReturnValue('Mocked date')})
 
       const wrapper = await createTestProvider()
       render(<ReleaseDetailsDialog onCancel={onCancelMock} onSubmit={onSubmitMock} />, {wrapper})
@@ -96,8 +99,8 @@ describe('ReleaseDetailsDialog', () => {
   })
 
   describe('when updating an existing bundle', () => {
-    const onCancelMock = jest.fn()
-    const onSubmitMock = jest.fn()
+    const onCancelMock = vi.fn()
+    const onSubmitMock = vi.fn()
     const existingBundleValue: BundleDocument = {
       _id: 'existing-bundle',
       _type: 'release',
@@ -118,12 +121,12 @@ describe('ReleaseDetailsDialog', () => {
       mockUseBundleStore.mockReturnValue({
         data: [],
         loading: true,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         error: undefined,
         deletedBundles: {},
       })
 
-      //mockUseDateTimeFormat.mockReturnValue({format: jest.fn().mockReturnValue('Mocked date')})
+      //mockUseDateTimeFormat.mockReturnValue({format: vi.fn().mockReturnValue('Mocked date')})
 
       const wrapper = await createTestProvider()
       render(
diff --git a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseForm.test.tsx b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseForm.test.tsx
index 4514bad23bb..2121e02f6a0 100644
--- a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseForm.test.tsx
+++ b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseForm.test.tsx
@@ -1,25 +1,25 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen} from '@testing-library/react'
 import {type BundleDocument, type FormBundleDocument, useDateTimeFormat} from 'sanity'
+import {beforeEach, describe, expect, it, type Mock, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
 import {useBundles} from '../../../../store/bundles'
 import {ReleaseForm} from '../ReleaseForm'
 
-jest.mock('../../../../../core/hooks/useDateTimeFormat', () => ({
-  useDateTimeFormat: jest.fn(),
+vi.mock('../../../../../core/hooks/useDateTimeFormat', () => ({
+  useDateTimeFormat: vi.fn(),
 }))
 
-jest.mock('../../../../store/bundles', () => ({
-  useBundles: jest.fn(),
+vi.mock('../../../../store/bundles', () => ({
+  useBundles: vi.fn(),
 }))
 
-const mockUseBundleStore = useBundles as jest.Mock<typeof useBundles>
-const mockUseDateTimeFormat = useDateTimeFormat as jest.Mock
+const mockUseBundleStore = useBundles as Mock<typeof useBundles>
+const mockUseDateTimeFormat = useDateTimeFormat as Mock
 
 describe('ReleaseForm', () => {
-  const onChangeMock = jest.fn()
-  const onErrorMock = jest.fn()
+  const onChangeMock = vi.fn()
+  const onErrorMock = vi.fn()
   const valueMock: FormBundleDocument = {
     _id: 'very-random',
     _type: 'release',
@@ -54,12 +54,12 @@ describe('ReleaseForm', () => {
       mockUseBundleStore.mockReturnValue({
         data: mockData,
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         error: undefined,
         deletedBundles: {},
       })
 
-      mockUseDateTimeFormat.mockReturnValue({format: jest.fn().mockReturnValue('Mocked date')})
+      mockUseDateTimeFormat.mockReturnValue({format: vi.fn().mockReturnValue('Mocked date')})
 
       const wrapper = await createTestProvider()
       render(<ReleaseForm onChange={onChangeMock} value={valueMock} />, {
@@ -139,11 +139,11 @@ describe('ReleaseForm', () => {
       mockUseBundleStore.mockReturnValue({
         data: mockData,
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {} as Record<string, BundleDocument>,
       })
 
-      mockUseDateTimeFormat.mockReturnValue({format: jest.fn().mockReturnValue('Mocked date')})
+      mockUseDateTimeFormat.mockReturnValue({format: vi.fn().mockReturnValue('Mocked date')})
 
       const wrapper = await createTestProvider()
       render(<ReleaseForm onChange={onChangeMock} value={existingBundleValue} />, {
diff --git a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseIconEditorPicker.test.tsx b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseIconEditorPicker.test.tsx
index 4a409e99621..7f31adc1cef 100644
--- a/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseIconEditorPicker.test.tsx
+++ b/packages/sanity/src/core/releases/components/dialog/__tests__/ReleaseIconEditorPicker.test.tsx
@@ -1,5 +1,5 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen} from '@testing-library/react'
+import {beforeEach, describe, expect, it, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
 import {
@@ -7,12 +7,15 @@ import {
   type ReleaseIconEditorPickerValue,
 } from '../ReleaseIconEditorPicker'
 
-jest.mock('sanity', () => ({
-  useTranslation: jest.fn().mockReturnValue({t: jest.fn()}),
-}))
+vi.mock('sanity', () => {
+  return {
+    SANITY_VERSION: '0.0.0',
+    useTranslation: vi.fn().mockReturnValue({t: vi.fn()}),
+  }
+})
 
 describe('BundleIconEditorPicker', () => {
-  const onChangeMock = jest.fn()
+  const onChangeMock = vi.fn()
   const valueMock: ReleaseIconEditorPickerValue = {
     hue: 'gray',
     icon: 'cube',
diff --git a/packages/sanity/src/core/releases/hooks/__tests__/useDocumentVersions.test.tsx b/packages/sanity/src/core/releases/hooks/__tests__/useDocumentVersions.test.tsx
index 4ca2e78f7bd..e3e779b9218 100644
--- a/packages/sanity/src/core/releases/hooks/__tests__/useDocumentVersions.test.tsx
+++ b/packages/sanity/src/core/releases/hooks/__tests__/useDocumentVersions.test.tsx
@@ -1,6 +1,6 @@
-import {describe, expect, it, jest} from '@jest/globals'
 import {renderHook} from '@testing-library/react'
 import {of} from 'rxjs'
+import {describe, expect, it, type Mock, vi} from 'vitest'
 
 import {type DocumentPreviewStore} from '../../../preview'
 import {type BundleDocument} from '../../../store'
@@ -8,17 +8,15 @@ import {type PublishedId} from '../../../util/draftUtils'
 import {useDocumentVersions} from '../useDocumentVersions'
 
 // Mock the entire module
-jest.mock('../../../studio/source')
+vi.mock('../../../studio/source')
 
-jest.mock('sanity', () => {
-  const actual = jest.requireActual('sanity')
-  return Object.assign({}, actual, {
-    useClient: jest.fn(),
-    useBundles: jest.fn(() => ({data: {}})),
-    getPublishedId: jest.fn(),
-    useDocumentPreviewStore: jest.fn(),
-  })
-})
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useClient: vi.fn(),
+  useBundles: vi.fn(() => ({data: {}})),
+  getPublishedId: vi.fn(),
+  useDocumentPreviewStore: vi.fn(),
+}))
 
 const mockBundles = [
   {
@@ -57,25 +55,23 @@ async function setupMocks({
 }) {
   const sanityModule = await import('sanity')
 
-  const useBundles = sanityModule.useBundles as jest.Mock<typeof sanityModule.useBundles>
-  const useDocumentPreviewStore = sanityModule.useDocumentPreviewStore as jest.Mock<
+  const useBundles = sanityModule.useBundles as Mock<typeof sanityModule.useBundles>
+  const useDocumentPreviewStore = sanityModule.useDocumentPreviewStore as Mock<
     typeof sanityModule.useDocumentPreviewStore
   >
-  const getPublishedId = sanityModule.getPublishedId as jest.Mock<
-    typeof sanityModule.getPublishedId
-  >
+  const getPublishedId = sanityModule.getPublishedId as Mock<typeof sanityModule.getPublishedId>
 
   useBundles.mockReturnValue({
     data: bundles,
     loading: false,
-    dispatch: jest.fn(),
+    dispatch: vi.fn(),
     deletedBundles: {},
   })
 
   getPublishedId.mockReturnValue('document-1' as PublishedId)
 
   useDocumentPreviewStore.mockReturnValue({
-    unstable_observeDocumentIdSet: jest
+    unstable_observeDocumentIdSet: vi
       .fn<DocumentPreviewStore['unstable_observeDocumentIdSet']>()
       .mockReturnValue(of({status: 'connected', documentIds: versionIds})),
   } as unknown as DocumentPreviewStore)
diff --git a/packages/sanity/src/core/releases/tool/components/ReleaseMenuButton/__tests__/ReleaseMenuButton.test.tsx b/packages/sanity/src/core/releases/tool/components/ReleaseMenuButton/__tests__/ReleaseMenuButton.test.tsx
index cf45f419b83..83fbe9df2aa 100644
--- a/packages/sanity/src/core/releases/tool/components/ReleaseMenuButton/__tests__/ReleaseMenuButton.test.tsx
+++ b/packages/sanity/src/core/releases/tool/components/ReleaseMenuButton/__tests__/ReleaseMenuButton.test.tsx
@@ -1,7 +1,7 @@
-import {beforeEach, describe, expect, jest, test} from '@jest/globals'
 import {fireEvent, render, screen} from '@testing-library/react'
 import {act} from 'react'
 import {useRouter} from 'sanity/router'
+import {beforeEach, describe, expect, test, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../../test/testUtils/TestProvider'
 import {type BundleDocument} from '../../../../../store/bundles/types'
@@ -9,20 +9,21 @@ import {useBundleOperations} from '../../../../../store/bundles/useBundleOperati
 import {releasesUsEnglishLocaleBundle} from '../../../../i18n'
 import {ReleaseMenuButton, type ReleaseMenuButtonProps} from '../ReleaseMenuButton'
 
-jest.mock('sanity', () => ({
-  useTranslation: jest.fn().mockReturnValue({t: jest.fn()}),
+vi.mock('sanity', () => ({
+  SANITY_VERSION: '0.0.0',
+  useTranslation: vi.fn().mockReturnValue({t: vi.fn()}),
 }))
 
-jest.mock('../../../../../store/bundles/useBundleOperations', () => ({
-  useBundleOperations: jest.fn().mockReturnValue({
-    deleteBundle: jest.fn(),
-    updateBundle: jest.fn(),
+vi.mock('../../../../../store/bundles/useBundleOperations', () => ({
+  useBundleOperations: vi.fn().mockReturnValue({
+    deleteBundle: vi.fn(),
+    updateBundle: vi.fn(),
   }),
 }))
 
-jest.mock('sanity/router', () => ({
-  ...(jest.requireActual('sanity/router') || {}),
-  useRouter: jest.fn().mockReturnValue({state: {}, navigate: jest.fn()}),
+vi.mock('sanity/router', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useRouter: vi.fn().mockReturnValue({state: {}, navigate: vi.fn()}),
 }))
 
 const renderTest = async ({
@@ -41,7 +42,7 @@ const renderTest = async ({
 
 describe('BundleMenuButton', () => {
   beforeEach(() => {
-    jest.clearAllMocks()
+    vi.clearAllMocks()
   })
 
   test('will archive an unarchived bundle', async () => {
diff --git a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseDetail.test.tsx b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseDetail.test.tsx
index 380eccd5e1d..c38351352f9 100644
--- a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseDetail.test.tsx
+++ b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseDetail.test.tsx
@@ -1,6 +1,6 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen} from '@testing-library/react'
 import {route, RouterProvider} from 'sanity/router'
+import {beforeEach, describe, expect, it, type Mock, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
 import {useBundles} from '../../../../store/bundles'
@@ -9,48 +9,52 @@ import {releasesUsEnglishLocaleBundle} from '../../../i18n'
 import {ReleaseDetail} from '../ReleaseDetail'
 import {useBundleDocuments} from '../useBundleDocuments'
 
-jest.mock('../../../../store/bundles', () => ({
-  useBundles: jest.fn().mockReturnValue({data: [], loading: false, deletedBundles: {}}),
+vi.mock('../../../../store/bundles', () => ({
+  useBundles: vi.fn().mockReturnValue({data: [], loading: false, deletedBundles: {}}),
 }))
 
-jest.mock('../../../../store/bundles/useBundleOperations', () => ({
-  useBundleOperations: jest.fn().mockReturnValue({
-    publishBundle: jest.fn(),
+vi.mock('../../../../store/bundles/useBundleOperations', () => ({
+  useBundleOperations: vi.fn().mockReturnValue({
+    publishBundle: vi.fn(),
   }),
 }))
 
-jest.mock('../useBundleDocuments', () => ({
-  useBundleDocuments: jest
+vi.mock('../useBundleDocuments', () => ({
+  useBundleDocuments: vi
     .fn<typeof useBundleDocuments>()
     .mockReturnValue({loading: true, results: []}),
 }))
 
-jest.mock('sanity', () => ({
-  LoadingBlock: () => <div data-testid="mocked-loading-block" />,
-  useClient: jest.fn().mockReturnValue({getUrl: jest.fn(), config: jest.fn().mockReturnValue({})}),
-  useCurrentUser: jest.fn().mockReturnValue({id: 'test-user-id'}),
-  useTranslation: jest.fn().mockReturnValue({t: jest.fn()}),
-}))
+vi.mock('sanity', async (importOriginal) => {
+  return {
+    ...(await importOriginal()),
+    SANITY_VERSION: '0.0.0',
+    LoadingBlock: () => <div data-testid="mocked-loading-block" />,
+    useClient: vi.fn().mockReturnValue({getUrl: vi.fn(), config: vi.fn().mockReturnValue({})}),
+    useCurrentUser: vi.fn().mockReturnValue({id: 'test-user-id'}),
+    useTranslation: vi.fn().mockReturnValue({t: vi.fn()}),
+  }
+})
 
-jest.mock('../../components/ReleasePublishAllButton/useObserveDocumentRevisions', () => ({
-  useObserveDocumentRevisions: jest.fn().mockReturnValue({
+vi.mock('../../components/ReleasePublishAllButton/useObserveDocumentRevisions', () => ({
+  useObserveDocumentRevisions: vi.fn().mockReturnValue({
     '123': 'mock revision id',
   }),
 }))
 
-jest.mock('../ReleaseSummary', () => ({
+vi.mock('../ReleaseSummary', () => ({
   ReleaseSummary: () => <div data-testid="mocked-release-summary" />,
 }))
 
-jest.mock('../documentTable/useReleaseHistory', () => ({
-  useReleaseHistory: jest.fn().mockReturnValue({
+vi.mock('../documentTable/useReleaseHistory', () => ({
+  useReleaseHistory: vi.fn().mockReturnValue({
     documentsHistory: new Map(),
   }),
 }))
 
-const mockUseBundles = useBundles as jest.Mock<typeof useBundles>
-const mockUseBundleDocuments = useBundleDocuments as jest.Mock<typeof useBundleDocuments>
-const mockRouterNavigate = jest.fn()
+const mockUseBundles = useBundles as Mock<typeof useBundles>
+const mockUseBundleDocuments = useBundleDocuments as Mock<typeof useBundleDocuments>
+const mockRouterNavigate = vi.fn()
 
 const renderTest = async () => {
   const wrapper = await createTestProvider({
@@ -90,7 +94,7 @@ describe('ReleaseDetail', () => {
       mockUseBundles.mockReturnValue({
         data: [],
         loading: true,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
       await renderTest()
@@ -127,7 +131,7 @@ describe('ReleaseDetail', () => {
           },
         ],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
       await renderTest()
@@ -168,7 +172,7 @@ describe('after bundles have loaded', () => {
           },
         ],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
     })
@@ -353,7 +357,7 @@ describe('after bundles have loaded', () => {
           },
         ],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
 
@@ -381,7 +385,7 @@ describe('after bundles have loaded', () => {
       mockUseBundles.mockReturnValue({
         data: [],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {
           'test-bundle-id': {
             title: 'Test bundle',
@@ -417,7 +421,7 @@ describe('after bundles have loaded', () => {
       mockUseBundles.mockReturnValue({
         data: [],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
       await renderTest()
diff --git a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseReview.test.tsx b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseReview.test.tsx
index 9d9bba3b76e..fa2a653bdac 100644
--- a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseReview.test.tsx
+++ b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseReview.test.tsx
@@ -1,7 +1,7 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {act, fireEvent, render, screen, within} from '@testing-library/react'
 import {type ReactNode} from 'react'
-import {ColorSchemeProvider, type DefaultPreview, UserColorManagerProvider} from 'sanity'
+import {ColorSchemeProvider, UserColorManagerProvider} from 'sanity'
+import {beforeEach, describe, expect, it, type Mock, vi} from 'vitest'
 
 import {queryByDataUi} from '../../../../../../test/setup/customQueries'
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
@@ -122,37 +122,26 @@ const MOCKED_PROPS = {
   },
 }
 
-jest.mock('sanity/router', () => ({
-  ...(jest.requireActual('sanity/router') || {}),
-  IntentLink: jest.fn().mockImplementation((props: any) => <a> {props.children}</a>),
-  useRouter: jest.fn().mockReturnValue({
+vi.mock('sanity/router', async (importOriginal) => ({
+  ...(await importOriginal()),
+  IntentLink: vi.fn().mockImplementation((props: any) => <a> {props.children}</a>),
+  useRouter: vi.fn().mockReturnValue({
     state: {releaseId: 'differences'},
-    navigate: jest.fn(),
+    navigate: vi.fn(),
   }),
 }))
 
-jest.mock('../../../components/ReleasesMenu', () => ({
+vi.mock('../../../components/ReleasesMenu', () => ({
   ReleasesMenu: () => <div>ReleasesMenu</div>,
 }))
 
-jest.mock('../../../../preview/useObserveDocument', () => {
+vi.mock('../../../../preview/useObserveDocument', () => {
   return {
-    useObserveDocument: jest.fn(),
+    useObserveDocument: vi.fn(),
   }
 })
 
-jest.mock('../../../../components', () => {
-  const {DefaultPreview: actualDefaultPreview} = jest.requireActual(
-    '../../../../components/previews/general/DefaultPreview',
-  ) as {DefaultPreview: typeof DefaultPreview}
-
-  return {
-    ...(jest.requireActual('../../../../components') || {}),
-    DefaultPreview: actualDefaultPreview,
-  }
-})
-
-const mockedUseObserveDocument = useObserveDocument as jest.Mock<typeof useObserveDocument>
+const mockedUseObserveDocument = useObserveDocument as Mock<typeof useObserveDocument>
 
 async function createReleaseReviewWrapper() {
   const wrapper = await createTestProvider({
diff --git a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseSummary.test.tsx b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseSummary.test.tsx
index ac065f40193..e331da1df62 100644
--- a/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseSummary.test.tsx
+++ b/packages/sanity/src/core/releases/tool/detail/__tests__/ReleaseSummary.test.tsx
@@ -1,7 +1,7 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen, within} from '@testing-library/react'
-import {type BundleDocument, type DefaultPreview, defineType} from 'sanity'
+import {type BundleDocument, defineType} from 'sanity'
 import {route, RouterProvider} from 'sanity/router'
+import {beforeEach, describe, expect, it, vi} from 'vitest'
 
 import {getAllByDataUi, getByDataUi} from '../../../../../../test/setup/customQueries'
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
@@ -10,34 +10,23 @@ import {type DocumentHistory} from '../documentTable/useReleaseHistory'
 import {ReleaseSummary, type ReleaseSummaryProps} from '../ReleaseSummary'
 import {type DocumentInBundleResult} from '../useBundleDocuments'
 
-jest.mock('../../../../studio/addonDataset/useAddonDataset', () => ({
-  useAddonDataset: jest.fn().mockReturnValue({client: {}}),
+vi.mock('../../../../studio/addonDataset/useAddonDataset', () => ({
+  useAddonDataset: vi.fn().mockReturnValue({client: {}}),
 }))
 
-jest.mock('../../../../store', () => ({
-  ...(jest.requireActual('../../../../store') || {}),
-  useUser: jest.fn().mockReturnValue([{}]),
+vi.mock('../../../../store', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useUser: vi.fn().mockReturnValue([{}]),
 }))
 
-jest.mock('../../../../user-color', () => ({
-  useUserColor: jest.fn().mockReturnValue('red'),
+vi.mock('../../../../user-color', () => ({
+  useUserColor: vi.fn().mockReturnValue('red'),
 }))
 
-jest.mock('../../../components/ReleasesMenu', () => ({
+vi.mock('../../../components/ReleasesMenu', () => ({
   ReleasesMenu: () => <div>ReleasesMenu</div>,
 }))
 
-jest.mock('../../../../components', () => {
-  const {DefaultPreview: actualDefaultPreview} = jest.requireActual(
-    '../../../../components/previews/general/DefaultPreview',
-  ) as {DefaultPreview: typeof DefaultPreview}
-
-  return {
-    ...(jest.requireActual('../../../../components') || {}),
-    DefaultPreview: actualDefaultPreview,
-  }
-})
-
 const timeNow = new Date()
 
 const releaseDocuments: DocumentInBundleResult[] = [
@@ -122,7 +111,7 @@ const renderTest = async (props: Partial<ReleaseSummaryProps>) => {
   return render(
     <RouterProvider
       state={{}}
-      onNavigate={jest.fn()}
+      onNavigate={vi.fn()}
       router={route.create('/test', [route.intents('/intent')])}
     >
       <ReleaseSummary
@@ -158,7 +147,7 @@ const renderTest = async (props: Partial<ReleaseSummaryProps>) => {
 
 describe('ReleaseSummary', () => {
   beforeEach(async () => {
-    jest.clearAllMocks()
+    vi.clearAllMocks()
 
     await renderTest({})
   })
diff --git a/packages/sanity/src/core/releases/tool/overview/__tests__/ReleasesOverview.test.tsx b/packages/sanity/src/core/releases/tool/overview/__tests__/ReleasesOverview.test.tsx
index 6e81b23d90d..f8b65e6cae1 100644
--- a/packages/sanity/src/core/releases/tool/overview/__tests__/ReleasesOverview.test.tsx
+++ b/packages/sanity/src/core/releases/tool/overview/__tests__/ReleasesOverview.test.tsx
@@ -1,6 +1,6 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {fireEvent, render, screen, waitFor, within} from '@testing-library/react'
 import {useRouter} from 'sanity/router'
+import {beforeEach, describe, expect, it, type Mock, vi} from 'vitest'
 
 import {queryByDataUi} from '../../../../../../test/setup/customQueries'
 import {createTestProvider} from '../../../../../../test/testUtils/TestProvider'
@@ -10,27 +10,31 @@ import {releasesUsEnglishLocaleBundle} from '../../../i18n'
 import {type BundlesMetadata, useBundlesMetadata} from '../../useBundlesMetadata'
 import {ReleasesOverview} from '../ReleasesOverview'
 
-jest.mock('../../useBundlesMetadata', () => ({
-  useBundlesMetadata: jest.fn(),
-}))
+vi.mock('sanity', () => {
+  return {
+    SANITY_VERSION: '0.0.0',
+    Translate: vi.fn(),
+    useCurrentUser: vi.fn().mockReturnValue({user: {id: 'user-id'}}),
+    useTranslation: vi.fn().mockReturnValue({t: vi.fn()}),
+  }
+})
 
-jest.mock('../../../../store', () => ({
-  ...(jest.requireActual('../../../../store') || {}),
-  useBundles: jest.fn(),
+vi.mock('../../useBundlesMetadata', () => ({
+  useBundlesMetadata: vi.fn(),
 }))
 
-jest.mock('sanity', () => ({
-  useCurrentUser: jest.fn().mockReturnValue({user: {id: 'user-id'}}),
-  useTranslation: jest.fn().mockReturnValue({t: jest.fn()}),
+vi.mock('../../../../store', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useBundles: vi.fn(),
 }))
 
-jest.mock('sanity/router', () => ({
-  ...(jest.requireActual('sanity/router') || {}),
-  useRouter: jest.fn().mockReturnValue({state: {}, navigate: jest.fn()}),
+vi.mock('sanity/router', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useRouter: vi.fn().mockReturnValue({state: {}, navigate: vi.fn()}),
 }))
 
-const mockUseBundles = useBundles as jest.Mock<typeof useBundles>
-const mockUseBundlesMetadata = useBundlesMetadata as jest.Mock<typeof useBundlesMetadata>
+const mockUseBundles = useBundles as Mock<typeof useBundles>
+const mockUseBundlesMetadata = useBundlesMetadata as Mock<typeof useBundlesMetadata>
 
 describe('ReleasesOverview', () => {
   describe('when loading bundles', () => {
@@ -38,7 +42,7 @@ describe('ReleasesOverview', () => {
       mockUseBundles.mockReturnValue({
         data: null,
         loading: true,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
       mockUseBundlesMetadata.mockReturnValue({
@@ -78,7 +82,7 @@ describe('ReleasesOverview', () => {
       mockUseBundles.mockReturnValue({
         data: [],
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {},
       })
       mockUseBundlesMetadata.mockReturnValue({
@@ -150,7 +154,7 @@ describe('ReleasesOverview', () => {
       mockUseBundles.mockReturnValue({
         data: bundles,
         loading: false,
-        dispatch: jest.fn(),
+        dispatch: vi.fn(),
         deletedBundles: {
           'deleted-bundle-id': {
             title: 'Deleted Bundle',
diff --git a/packages/sanity/src/core/releases/util/util.test.ts b/packages/sanity/src/core/releases/util/util.test.ts
index 5ec07a26821..a730849d7e0 100644
--- a/packages/sanity/src/core/releases/util/util.test.ts
+++ b/packages/sanity/src/core/releases/util/util.test.ts
@@ -1,4 +1,4 @@
-import {describe, expect, it} from '@jest/globals'
+import {describe, expect, it} from 'vitest'
 
 import {getDocumentIsInPerspective} from './util'
 
diff --git a/packages/sanity/src/core/store/_legacy/document/document-pair/utils/fetchFeatureToggle.test.ts b/packages/sanity/src/core/store/_legacy/document/document-pair/utils/fetchFeatureToggle.test.ts
index 4f84c281348..537a6d58e77 100644
--- a/packages/sanity/src/core/store/_legacy/document/document-pair/utils/fetchFeatureToggle.test.ts
+++ b/packages/sanity/src/core/store/_legacy/document/document-pair/utils/fetchFeatureToggle.test.ts
@@ -3,7 +3,8 @@ import {describe, expect, it, vi} from 'vitest'
 
 import {type ActionsFeatureToggle, mapResponse} from './fetchFeatureToggle'
 
-vi.mock('sanity', () => ({
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
   SANITY_VERSION: '3.47.0',
 }))
 
diff --git a/packages/sanity/src/core/util/__tests__/resolvePerspectives.test.ts b/packages/sanity/src/core/util/__tests__/resolvePerspectives.test.ts
index 812c2bf6df9..a08bae45024 100644
--- a/packages/sanity/src/core/util/__tests__/resolvePerspectives.test.ts
+++ b/packages/sanity/src/core/util/__tests__/resolvePerspectives.test.ts
@@ -1,4 +1,4 @@
-import {describe, expect, it} from '@jest/globals'
+import {describe, expect, it} from 'vitest'
 
 import {resolveBundlePerspective, resolvePerspectiveOptions} from '../resolvePerspective'
 
diff --git a/packages/sanity/src/structure/components/structureTool/StructureTitle.test.tsx b/packages/sanity/src/structure/components/structureTool/StructureTitle.test.tsx
index 9adc87bf10e..bd07059e908 100644
--- a/packages/sanity/src/structure/components/structureTool/StructureTitle.test.tsx
+++ b/packages/sanity/src/structure/components/structureTool/StructureTitle.test.tsx
@@ -10,15 +10,12 @@ import {type Panes} from '../../structureResolvers'
 import * as USE_STRUCTURE_TOOL from '../../useStructureTool'
 import {StructureTitle} from './StructureTitle'
 
-vi.mock('sanity', async () => {
-  const actual = await vi.importActual('sanity')
-  return {
-    ...actual,
-    useEditState: vi.fn(),
-    useSchema: vi.fn(),
-    unstable_useValuePreview: vi.fn(),
-  }
-})
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
+  useEditState: vi.fn(),
+  useSchema: vi.fn(),
+  unstable_useValuePreview: vi.fn(),
+}))
 
 function createWrapperComponent(client: SANITY.SanityClient) {
   const config = SANITY.defineConfig({
diff --git a/packages/sanity/src/structure/panes/document/documentPanel/banners/__tests__/DeletedDocumentBanners.test.tsx b/packages/sanity/src/structure/panes/document/documentPanel/banners/__tests__/DeletedDocumentBanners.test.tsx
index 6397f3622a2..5ce5193ea6e 100644
--- a/packages/sanity/src/structure/panes/document/documentPanel/banners/__tests__/DeletedDocumentBanners.test.tsx
+++ b/packages/sanity/src/structure/panes/document/documentPanel/banners/__tests__/DeletedDocumentBanners.test.tsx
@@ -1,31 +1,31 @@
-import {describe, expect, it, jest} from '@jest/globals'
 import {render, screen} from '@testing-library/react'
 import {type BundleDocument, LATEST, useBundles, usePerspective} from 'sanity'
 import {useDocumentPane} from 'sanity/structure'
+import {describe, expect, it, type Mock, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../../test/testUtils/TestProvider'
 import {structureUsEnglishLocaleBundle} from '../../../../../i18n'
 import {DeletedDocumentBanners} from '../DeletedDocumentBanners'
 
-jest.mock('../../../useDocumentPane', () => ({
-  useDocumentPane: jest.fn(),
+vi.mock('../../../useDocumentPane', () => ({
+  useDocumentPane: vi.fn(),
 }))
 
-jest.mock('../../../../../../core/releases/hooks/usePerspective', () => ({
-  usePerspective: jest.fn(),
+vi.mock('../../../../../../core/releases/hooks/usePerspective', () => ({
+  usePerspective: vi.fn(),
 }))
 
-jest.mock('../../../../../../core/store/bundles/useBundles', () => ({
-  useBundles: jest.fn(),
+vi.mock('../../../../../../core/store/bundles/useBundles', () => ({
+  useBundles: vi.fn(),
 }))
 
-jest.mock('../../../../../../core/store/_legacy/history/useTimelineSelector', () => ({
-  useTimelineSelector: jest.fn(),
+vi.mock('../../../../../../core/store/_legacy/history/useTimelineSelector', () => ({
+  useTimelineSelector: vi.fn(),
 }))
 
-const mockUseDocumentPane = useDocumentPane as jest.Mock<typeof useDocumentPane>
-const mockUseBundles = useBundles as jest.Mock<typeof useBundles>
-const mockUsePerspective = usePerspective as jest.Mock<typeof usePerspective>
+const mockUseDocumentPane = useDocumentPane as Mock<typeof useDocumentPane>
+const mockUseBundles = useBundles as Mock<typeof useBundles>
+const mockUsePerspective = usePerspective as Mock<typeof usePerspective>
 
 const renderTest = async () => {
   const wrapper = await createTestProvider({resources: [structureUsEnglishLocaleBundle]})
@@ -41,7 +41,7 @@ describe('DeletedDocumentBanners', () => {
     mockUseBundles.mockReturnValue({
       data: [],
       deletedBundles: {},
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       loading: false,
     })
     mockUseDocumentPane.mockReturnValue({
@@ -63,7 +63,7 @@ describe('DeletedDocumentBanners', () => {
     mockUseBundles.mockReturnValue({
       data: [mockBundleDocument],
       deletedBundles: {test: mockBundleDocument},
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       loading: false,
     })
     mockUseDocumentPane.mockReturnValue({
@@ -88,7 +88,7 @@ describe('DeletedDocumentBanners', () => {
     mockUseBundles.mockReturnValue({
       data: [mockBundleDocument],
       deletedBundles: {test: mockBundleDocument},
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       loading: false,
     })
     mockUseDocumentPane.mockReturnValue({
diff --git a/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentHeaderTitle.test.tsx b/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentHeaderTitle.test.tsx
index 19441ffdf71..966c2ff686a 100644
--- a/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentHeaderTitle.test.tsx
+++ b/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentHeaderTitle.test.tsx
@@ -1,4 +1,3 @@
-import {jest} from '@jest/globals'
 import {render, waitFor} from '@testing-library/react'
 import {
   defineConfig,
@@ -8,7 +7,7 @@ import {
   useDocumentVersions,
 } from 'sanity'
 import {useRouter} from 'sanity/router'
-import {beforeEach, describe, expect, it, vi} from 'vitest'
+import {beforeEach, describe, expect, it, type Mock, type MockedFunction, vi} from 'vitest'
 
 import {createMockSanityClient} from '../../../../../../../../test/mocks/mockSanityClient'
 import {createTestProvider} from '../../../../../../../../test/testUtils/TestProvider'
@@ -30,33 +29,29 @@ function createWrapperComponent(client: SanityClient) {
   })
 }
 
-jest.mock('../../../../useDocumentPane')
+vi.mock('../../../../useDocumentPane')
 
-jest.mock('sanity', () => {
-  // eslint-disable-next-line @typescript-eslint/consistent-type-imports
-  const actual = vi.importActual<typeof import('sanity')>('sanity')
+vi.mock('sanity', async (importOriginal) => {
   return {
-    ...actual,
-    unstable_useValuePreview: jest.fn(),
-    useDocumentVersions: jest.fn(),
+    ...(await importOriginal()),
+    unstable_useValuePreview: vi.fn(),
+    useDocumentVersions: vi.fn(),
   }
 })
 
-jest.mock('sanity/router')
+vi.mock('sanity/router')
 
-jest.mock('../../../../../../../core/store/bundles/useBundles', () => ({
-  useBundles: jest.fn(),
+vi.mock('../../../../../../../core/store/bundles/useBundles', () => ({
+  useBundles: vi.fn(),
 }))
 
-const mockUseBundles = useBundles as jest.Mock<typeof useBundles>
-const mockUseDocumentVersions = useDocumentVersions as jest.MockedFunction<
-  typeof useDocumentVersions
->
+const mockUseBundles = useBundles as Mock<typeof useBundles>
+const mockUseDocumentVersions = useDocumentVersions as MockedFunction<typeof useDocumentVersions>
 
 describe('DocumentHeaderTitle', () => {
-  const mockUseDocumentPane = useDocumentPane as jest.MockedFunction<typeof useDocumentPane>
-  const mockUseValuePreview = useValuePreview as jest.MockedFunction<typeof useValuePreview>
-  const mockUseRouter = useRouter as jest.MockedFunction<typeof useRouter>
+  const mockUseDocumentPane = useDocumentPane as MockedFunction<typeof useDocumentPane>
+  const mockUseValuePreview = useValuePreview as MockedFunction<typeof useValuePreview>
+  const mockUseRouter = useRouter as MockedFunction<typeof useRouter>
   const defaultProps = {
     connectionState: 'connected',
     schemaType: {title: 'Test Schema', name: 'testSchema'},
@@ -72,12 +67,12 @@ describe('DocumentHeaderTitle', () => {
     mockUseValuePreview.mockReturnValue({...defaultValue, error: undefined, value: undefined})
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
     // @ts-ignore
-    mockUseRouter.mockReturnValue({stickyParams: {}, state: {}, navigate: jest.fn()})
+    mockUseRouter.mockReturnValue({stickyParams: {}, state: {}, navigate: vi.fn()})
     mockUseDocumentVersions.mockReturnValue({data: [], loading: false, error: undefined})
     mockUseBundles.mockReturnValue({
       data: [],
       loading: false,
-      dispatch: jest.fn(),
+      dispatch: vi.fn(),
       deletedBundles: {},
     })
   })
diff --git a/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentPerspectiveMenu.test.tsx b/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentPerspectiveMenu.test.tsx
index 987fe8588c0..3b7049c1a89 100644
--- a/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentPerspectiveMenu.test.tsx
+++ b/packages/sanity/src/structure/panes/document/documentPanel/header/perspective/__tests__/DocumentPerspectiveMenu.test.tsx
@@ -1,27 +1,22 @@
-import {beforeEach, describe, expect, it, jest} from '@jest/globals'
 import {render, screen} from '@testing-library/react'
 import {type HTMLProps} from 'react'
 import {type BundleDocument, usePerspective} from 'sanity'
 import {type IntentLinkProps} from 'sanity/router'
+import {beforeEach, describe, expect, it, type Mock, type MockedFunction, vi} from 'vitest'
 
 import {createTestProvider} from '../../../../../../../../test/testUtils/TestProvider'
 import {type DocumentPaneContextValue} from '../../../../DocumentPaneContext'
 import {useDocumentPane} from '../../../../useDocumentPane'
 import {DocumentPerspectiveMenu} from '../DocumentPerspectiveMenu'
 
-jest.mock('sanity', () => {
-  // eslint-disable-next-line @typescript-eslint/consistent-type-imports
-  const actual = jest.requireActual<typeof import('sanity')>('sanity')
-
-  return {
-    ...actual,
-    usePerspective: jest.fn().mockReturnValue({
-      currentGlobalBundle: {},
-      setPerspective: jest.fn(),
-    }),
-    useTranslation: jest.fn().mockReturnValue({t: jest.fn()}),
-  }
-})
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
+  usePerspective: vi.fn().mockReturnValue({
+    currentGlobalBundle: {},
+    setPerspective: vi.fn(),
+  }),
+  useTranslation: vi.fn().mockReturnValue({t: vi.fn()}),
+}))
 
 const IntentLinkMock = (props: IntentLinkProps & HTMLProps<HTMLAnchorElement>) => {
   const {params = {}, intent, ...rest} = props
@@ -34,25 +29,25 @@ const IntentLinkMock = (props: IntentLinkProps & HTMLProps<HTMLAnchorElement>) =
   return <a {...rest} href={`/intent/${intent}/${stringParams}`} />
 }
 
-jest.mock('sanity/router', () => {
+vi.mock('sanity/router', () => {
   return {
-    useRouter: jest.fn().mockReturnValue({
+    useRouter: vi.fn().mockReturnValue({
       stickyParams: {},
     }),
     route: {
-      create: jest.fn(),
+      create: vi.fn(),
     },
     IntentLink: IntentLinkMock,
   }
 })
 
-jest.mock('../../../../useDocumentPane')
+vi.mock('../../../../useDocumentPane')
 
-const mockUseDocumentPane = useDocumentPane as jest.MockedFunction<
+const mockUseDocumentPane = useDocumentPane as MockedFunction<
   () => Partial<DocumentPaneContextValue>
 >
 
-const mockUsePerspective = usePerspective as jest.Mock
+const mockUsePerspective = usePerspective as Mock
 
 describe('DocumentPerspectiveMenu', () => {
   const mockCurrent: BundleDocument = {
@@ -69,10 +64,10 @@ describe('DocumentPerspectiveMenu', () => {
   }
 
   beforeEach(() => {
-    jest.clearAllMocks()
+    vi.clearAllMocks()
     mockUsePerspective.mockReturnValue({
       currentGlobalBundle: mockCurrent,
-      setPerspective: jest.fn(),
+      setPerspective: vi.fn(),
     })
 
     mockUseDocumentPane.mockReturnValue({
diff --git a/packages/sanity/src/structure/panes/documentList/__tests__/PaneContainer.test.tsx b/packages/sanity/src/structure/panes/documentList/__tests__/PaneContainer.test.tsx
index 6987c410c9c..1df35ee5052 100644
--- a/packages/sanity/src/structure/panes/documentList/__tests__/PaneContainer.test.tsx
+++ b/packages/sanity/src/structure/panes/documentList/__tests__/PaneContainer.test.tsx
@@ -23,15 +23,13 @@ vi.mock('../sheetList/useDocumentSheetList', () => ({
   useDocumentSheetList: vi.fn().mockReturnValue({data: [], isLoading: false}),
 }))
 
-vi.mock('sanity', async () => {
-  const actual = await vi.importActual('sanity')
-  return {
-    ...actual,
-    useSearchState: vi.fn(),
-  }
-})
-jest.mock('sanity/router', () => ({
-  ...(jest.requireActual('sanity/router') || {}),
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
+
+  useSearchState: vi.fn(),
+}))
+vi.mock('sanity/router', () => ({
+  ...(vi.requireActual('sanity/router') || {}),
   useRouter: jest.fn().mockReturnValue({stickyParams: {}, state: {}, navigate: jest.fn()}),
 }))
 
diff --git a/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/DocumentSheetListPane.test.tsx b/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/DocumentSheetListPane.test.tsx
index 2e4364988cb..db182381639 100644
--- a/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/DocumentSheetListPane.test.tsx
+++ b/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/DocumentSheetListPane.test.tsx
@@ -29,8 +29,8 @@ vi.mock('../useDocumentSheetList', () => ({
   }),
 }))
 
-vi.mock('sanity', async () => ({
-  ...((await vi.importActual('sanity')) || {}),
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
   useDocumentPreviewStore: vi.fn().mockReturnValue({
     observeForPreview: vi.fn().mockReturnValue([]),
   }),
diff --git a/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/useDocumentSheetColumns.test.ts b/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/useDocumentSheetColumns.test.ts
index 5ecef6757ca..2c120471b12 100644
--- a/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/useDocumentSheetColumns.test.ts
+++ b/packages/sanity/src/structure/panes/documentList/sheetList/__tests__/useDocumentSheetColumns.test.ts
@@ -4,8 +4,8 @@ import {describe, expect, it, vi} from 'vitest'
 
 import {useDocumentSheetColumns} from '../useDocumentSheetColumns'
 
-vi.mock('sanity', () => ({
-  ...(vi.importActual('sanity') || {}),
+vi.mock('sanity', async (importOriginal) => ({
+  ...(await importOriginal()),
   useDocumentPreviewStore: vi.fn().mockReturnValue({}),
 }))