You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
842 B
TypeScript
26 lines
842 B
TypeScript
import { act } from 'react-dom/test-utils';
|
|
import { beforeEach } from 'vitest';
|
|
import { createWithEqualityFn as actualCreate } from 'zustand/traditional';
|
|
|
|
// a variable to hold reset functions for all stores declared in the app
|
|
const storeResetFns = new Set<() => void>();
|
|
|
|
// when creating a store, we get its initial state, create a reset function and add it in the set
|
|
const createImpl = (createState: any) => {
|
|
const store = actualCreate(createState, Object.is);
|
|
const initialState = store.getState();
|
|
storeResetFns.add(() => store.setState(initialState, true));
|
|
return store;
|
|
};
|
|
|
|
// Reset all stores after each test run
|
|
beforeEach(() => {
|
|
act(() => {
|
|
for (const resetFn of storeResetFns) {
|
|
resetFn();
|
|
}
|
|
});
|
|
});
|
|
|
|
export const createWithEqualityFn = (f: any) => (f === undefined ? createImpl : createImpl(f));
|