✅ test: update useStore

ant-design · Aug 14, 2023 · 9995505 · 9995505
1 parent dc508f0
commit 9995505
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/__mocks__/zustand.ts b/__mocks__/zustand.ts
@@ -1,6 +1,6 @@
 import { act } from 'react-dom/test-utils';
 import { beforeEach } from 'vitest';
-import { createWithEqualityFn as actualCreate } from 'zustand/traditional';
+import { createWithEqualityFn as actualCreate, useStoreWithEqualityFn } from 'zustand/traditional';
 
 // a variable to hold reset functions for all stores declared in the app
 const storeResetFns = new Set<() => void>();
@@ -26,4 +26,4 @@ export const create = (f) => (f === undefined ? createImpl : createImpl(f));
 
 export default create;
 
-export { useStore } from 'zustand';
+export const useStore = (store, selector) => useStoreWithEqualityFn(store, selector, Object.is);