diff --git a/packages/sanity/src/core/bundles/util/tests/createWrapper.tsx b/packages/sanity/src/core/bundles/util/tests/createWrapper.tsx index 68e827bd8ce..120caff9290 100644 --- a/packages/sanity/src/core/bundles/util/tests/createWrapper.tsx +++ b/packages/sanity/src/core/bundles/util/tests/createWrapper.tsx @@ -1,10 +1,13 @@ import {studioTheme, ThemeProvider} from '@sanity/ui' import {type ReactNode} from 'react' -import {createTestProvider} from '../../../../../test/testUtils/TestProvider' +import { + createTestProvider, + type TestProviderOptions, +} from '../../../../../test/testUtils/TestProvider' -export const createWrapper = async () => { - const TestProvider = await createTestProvider() +export const createWrapper = async (options?: TestProviderOptions) => { + const TestProvider = await createTestProvider(options) return function Wrapper({children}: {children: ReactNode}): JSX.Element { return ( diff --git a/packages/sanity/src/core/releases/tool/__tests__/BundlesOverview.test.tsx b/packages/sanity/src/core/releases/tool/__tests__/BundlesOverview.test.tsx index cd7ecf592ac..bc718a1c98f 100644 --- a/packages/sanity/src/core/releases/tool/__tests__/BundlesOverview.test.tsx +++ b/packages/sanity/src/core/releases/tool/__tests__/BundlesOverview.test.tsx @@ -1,9 +1,8 @@ import {beforeEach, describe, expect, it, jest} from '@jest/globals' import {fireEvent, render, screen, waitFor} from '@testing-library/react' -import {type ReactNode} from 'react' import {queryByDataUi} from '../../../../../test/setup/customQueries' -import {createTestProvider} from '../../../../../test/testUtils/TestProvider' +import {createWrapper} from '../../../bundles/util/tests/createWrapper' import {useBundles} from '../../../store/bundles' import {type BundleDocument} from '../../../store/bundles/types' import {releasesUsEnglishLocaleBundle} from '../../i18n' @@ -23,15 +22,6 @@ jest.mock('sanity/router', () => ({ useRouter: jest.fn().mockReturnValue({state: {}, navigate: jest.fn()}), })) -const createWrapper = async () => { - const TestProvider = await createTestProvider({ - resources: [releasesUsEnglishLocaleBundle], - }) - return function Wrapper({children}: {children: ReactNode}) { - return {children} - } -} - const mockUseBundleStore = useBundles as jest.Mock describe('BundlesOverview', () => { @@ -43,7 +33,9 @@ describe('BundlesOverview', () => { dispatch: jest.fn(), }) - const wrapper = await createWrapper() + const wrapper = await createWrapper({ + resources: [releasesUsEnglishLocaleBundle], + }) return render(, {wrapper}) }) diff --git a/packages/sanity/test/testUtils/TestProvider.tsx b/packages/sanity/test/testUtils/TestProvider.tsx index aff03909472..9fba0d5268f 100644 --- a/packages/sanity/test/testUtils/TestProvider.tsx +++ b/packages/sanity/test/testUtils/TestProvider.tsx @@ -16,7 +16,7 @@ import {studioDefaultLocaleResources} from '../../src/core/i18n/bundles/studio' import {prepareI18n} from '../../src/core/i18n/i18nConfig' import {getMockWorkspace} from './getMockWorkspaceFromConfig' -interface TestProviderOptions { +export interface TestProviderOptions { config?: SingleWorkspace client?: SanityClient resources?: LocaleResourceBundle[]