update msmState factory

canonical · Jun 12, 2024 · 5cefb34 · 5cefb34
1 parent d003e62
commit 5cefb34
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/src/testing/factories/msm.ts b/src/testing/factories/msm.ts
@@ -13,5 +13,6 @@ const msmStatus = define<MsmStatus>({
 export const msm = define<MsmState>({
   status: msmStatus(),
   loading: false,
+  loaded: false,
   errors: null,
 });
diff --git a/src/testing/factories/state.ts b/src/testing/factories/state.ts
@@ -433,6 +433,7 @@ export const msmStatus = define<MsmStatus>({
 export const msmState = define<MsmState>({
   status: msmStatus(),
   loading: false,
+  loaded: false,
   errors: null,
 });
 

diff --git a/src/testing/utils.tsx b/src/testing/utils.tsx
@@ -188,9 +188,14 @@ export const renderWithMockStore = (
   rerender: (ui: React.ReactNode, newOptions?: WithStoreRenderOptions) => void;
 } => {
   const { state, store, ...renderOptions } = options ?? {};
+  const initialState =
+    typeof state === "function"
+      ? produce(rootStateFactory(), state)
+      : state || rootStateFactory();
+
   const rendered = render(ui, {
     wrapper: (props) => (
-      <WithMockStoreProvider {...props} state={state} store={store} />
+      <WithMockStoreProvider {...props} state={initialState} store={store} />
     ),
     ...renderOptions,
   });