1 files changed, 13 insertions, 15 deletions
diff --git a/jest/test-utils.tsx b/jest/test-utils.tsx
index a5946ed06..c84ee637e 100644
--- a/jest/test-utils.tsx
+++ b/jest/test-utils.tsx
@@ -1,29 +1,27 @@
 import React from 'react'
-import RN from 'react-native'
 import {render} from '@testing-library/react-native'
 import {GestureHandlerRootView} from 'react-native-gesture-handler'
 import {RootSiblingParent} from 'react-native-root-siblings'
 import {SafeAreaProvider} from 'react-native-safe-area-context'
-import {DEFAULT_SERVICE, RootStoreModel, RootStoreProvider} from '../src/state'
-import {SessionServiceClient} from '../src/third-party/api/src'
-import {sessionClient as AtpApi} from '../src/third-party/api'
+import {RootStoreProvider} from '../src/state'
+import {mockedRootStore} from '../__mocks__/state-mock'
 
-const WrappedComponent = ({children}: any) => {
-  const api = AtpApi.service(DEFAULT_SERVICE) as SessionServiceClient
-  const rootStore = new RootStoreModel(api)
-  return (
+const customRender = (ui: any, storeMock?: any) =>
+  render(
+    // eslint-disable-next-line react-native/no-inline-styles
     <GestureHandlerRootView style={{flex: 1}}>
       <RootSiblingParent>
-        <RootStoreProvider value={rootStore}>
-          <SafeAreaProvider>{children}</SafeAreaProvider>
+        <RootStoreProvider
+          value={
+            storeMock != null
+              ? {...mockedRootStore, ...storeMock}
+              : mockedRootStore
+          }>
+          <SafeAreaProvider>{ui}</SafeAreaProvider>
         </RootStoreProvider>
       </RootSiblingParent>
-    </GestureHandlerRootView>
+    </GestureHandlerRootView>,
   )
-}
-
-const customRender = (ui: any, options?: any) =>
-  render(ui, {wrapper: WrappedComponent, ...options})
 
 // re-export everything
 export * from '@testing-library/react-native'