twenty/packages/twenty-front/src/testing/jest/getJestMetadataAndApolloMocksWrapper.tsx

import { MockedProvider, type MockedResponse } from '@apollo/client/testing';
import { Provider as JotaiProvider } from 'jotai';
import { type ReactNode } from 'react';
import { RecoilRoot, type MutableSnapshot } from 'recoil';

import { ContextStoreComponentInstanceContext } from '@/context-store/states/contexts/ContextStoreComponentInstanceContext';

import { type ObjectMetadataItem } from '@/object-metadata/types/ObjectMetadataItem';
import { RecordComponentInstanceContextsWrapper } from '@/object-record/components/RecordComponentInstanceContextsWrapper';
import { SnackBarComponentInstanceContext } from '@/ui/feedback/snack-bar-manager/contexts/SnackBarComponentInstanceContext';
import { jotaiStore } from '@/ui/utilities/state/jotai/jotaiStore';
import { ViewComponentInstanceContext } from '@/views/states/contexts/ViewComponentInstanceContext';
import { type InMemoryCache } from '@apollo/client';
import { JestContextStoreSetter } from '~/testing/jest/JestContextStoreSetter';
import { JestObjectMetadataItemSetter } from '~/testing/jest/JestObjectMetadataItemSetter';

export const getJestMetadataAndApolloMocksWrapper = ({
  apolloMocks,
  cache,
  onInitializeRecoilSnapshot,
  objectMetadataItems,
}: {
  cache?: InMemoryCache;
  apolloMocks?:
    | readonly MockedResponse<Record<string, any>, Record<string, any>>[]
    | undefined;
  onInitializeRecoilSnapshot?: (snapshot: MutableSnapshot) => void;
  objectMetadataItems?: ObjectMetadataItem[];
}) => {
  return ({ children }: { children: ReactNode }) => (
    <JotaiProvider store={jotaiStore}>
      <RecoilRoot initializeState={onInitializeRecoilSnapshot}>
        <SnackBarComponentInstanceContext.Provider
          value={{ instanceId: 'snack-bar-manager' }}
        >
          <MockedProvider mocks={apolloMocks} addTypename={false} cache={cache}>
            <RecordComponentInstanceContextsWrapper componentInstanceId="instanceId">
              <ViewComponentInstanceContext.Provider
                value={{ instanceId: 'instanceId' }}
              >
                <JestObjectMetadataItemSetter
                  objectMetadataItems={objectMetadataItems}
                >
                  <ContextStoreComponentInstanceContext.Provider
                    value={{ instanceId: 'instanceId' }}
                  >
                    <JestContextStoreSetter>{children}</JestContextStoreSetter>
                  </ContextStoreComponentInstanceContext.Provider>
                </JestObjectMetadataItemSetter>
              </ViewComponentInstanceContext.Provider>
            </RecordComponentInstanceContextsWrapper>
          </MockedProvider>
        </SnackBarComponentInstanceContext.Provider>
      </RecoilRoot>
    </JotaiProvider>
  );
};