test(useSetState): add case

alibaba · Jan 9, 2024 · 3e7e7b6 · 3e7e7b6
1 parent dcd1ec5
commit 3e7e7b6
Showing 1 changed file with 53 additions and 2 deletions.
diff --git a/packages/hooks/src/useSetState/__tests__/index.test.ts b/packages/hooks/src/useSetState/__tests__/index.test.ts
@@ -1,13 +1,20 @@
 import { act, renderHook } from '@testing-library/react';
 import useSetState from '../index';
 
+interface IInitialState {
+  hello: string;
+  value: number;
+  foo?: string;
+}
+
 describe('useSetState', () => {
   const setUp = <T extends object>(initialValue: T) =>
     renderHook(() => {
-      const [state, setState] = useSetState<T>(initialValue);
+      const [state, setState, resetState] = useSetState<T>(initialValue);
       return {
         state,
         setState,
+        resetState,
       } as const;
     });
 
@@ -18,8 +25,39 @@ describe('useSetState', () => {
     expect(hook.result.current.state).toEqual({ hello: 'world' });
   });
 
+  it('should support function initialValue', () => {
+    const hook = setUp(() => ({
+      hello: 'world',
+    }));
+    expect(hook.result.current.state).toEqual({ hello: 'world' });
+  });
+
+  it('should keep random initialValue', () => {
+    const random = Math.random();
+    const hook = setUp({
+      count: random,
+    });
+    act(() => {
+      hook.result.current.setState({ count: Math.random() });
+      hook.result.current.resetState();
+    });
+    expect(hook.result.current.state).toEqual({ count: random });
+  });
+
+  it('should keep random function initialValue', () => {
+    const random = Math.random();
+    const hook = setUp(() => ({
+      count: random,
+    }));
+    act(() => {
+      hook.result.current.setState({ count: Math.random() });
+      hook.result.current.resetState();
+    });
+    expect(hook.result.current.state).toEqual({ count: random });
+  });
+
   it('should support object', () => {
-    const hook = setUp<any>({
+    const hook = setUp<Omit<IInitialState, 'value'>>({
       hello: 'world',
     });
     act(() => {
@@ -37,4 +75,17 @@ describe('useSetState', () => {
     });
     expect(hook.result.current.state).toEqual({ count: 1 });
   });
+
+  it('should support resetState', () => {
+    const random = Math.random();
+    const hook = setUp<IInitialState>({
+      hello: '',
+      value: random,
+    });
+    act(() => {
+      hook.result.current.setState({ hello: 'world', value: Math.random(), foo: 'bar' });
+      hook.result.current.resetState();
+    });
+    expect(hook.result.current.state).toEqual({ hello: '', value: random });
+  });
 });