feat(hooks): add feature reset useToggle

chornos13 · Feb 18, 2021 · e2c0c7b · e2c0c7b
1 parent 3eba878
commit e2c0c7b
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 1 deletion.
diff --git a/src/hooks/useToggle/__test__/useToggle.test.ts b/src/hooks/useToggle/__test__/useToggle.test.ts
@@ -117,4 +117,31 @@ describe('basic function', () => {
       ...newState,
     })
   })
+
+  test('should reset state to initial value when call reset function', () => {
+    // arrange
+    const initialState = {
+      anyInitialData: 'anyInitialValue',
+    }
+    const initialToggle = false
+    const { result } = renderHook(() =>
+      useToggle({
+        initialToggle,
+        initialState,
+      }),
+    )
+
+    // act
+    act(() => {
+      result.current.toggle({
+        anyInitialData: 'anyChangedValue',
+      })
+
+      result.current.reset()
+    })
+
+    // assert
+    expect(result.current.state).toEqual(initialState)
+    expect(result.current.isToggled).toEqual(initialToggle)
+  })
 })
diff --git a/src/hooks/useToggle/useToggle.tsx b/src/hooks/useToggle/useToggle.tsx
@@ -40,11 +40,20 @@ function useToggle<T>(configs?: UseToggleConfigs<T>) {
     function untoggle(state?: T | any) {
       setState(TOGGLE_ACTION_TYPE.UNTOGGLE, state)
     }
+
+    function reset() {
+      setState(
+        initialToggle ? TOGGLE_ACTION_TYPE.TOGGLE : TOGGLE_ACTION_TYPE.UNTOGGLE,
+        initialState,
+      )
+    }
+
     return {
       isToggled,
-      state,
+      state: state as T,
       toggle,
       untoggle,
+      reset,
     }
   }, [counter])
 }