fix: types for onNext callback

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased][]
 
+### Fixed
+
+- Fixing types for `onNext` to cover the callback scenario
+
 ## [1.0.1][] - 2021-10-29
 
 ### Updated
@@ -43,7 +47,5 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 [unreleased]: https://github.com/willmendesneto/react-sweet-wizard/compare/v1.0.0...HEAD
 [1.0.0]: https://github.com/willmendesneto/react-sweet-wizard/tree/v1.0.0
-
-
-[Unreleased]: https://github.com/willmendesneto/react-sweet-wizard/compare/v1.0.1...HEAD
-[1.0.1]: https://github.com/willmendesneto/react-sweet-wizard/tree/v1.0.1
+[unreleased]: https://github.com/willmendesneto/react-sweet-wizard/compare/v1.0.1...HEAD
+[1.0.1]: https://github.com/willmendesneto/react-sweet-wizard/tree/v1.0.1

src/useWizard.test.tsx

@@ -29,6 +29,8 @@ const renderUseWizardHook = (initialStartIndex = 0) =>
   });
 
 describe('useWizard hook', () => {
+  afterEach(jest.resetAllMocks);
+
   it('should throw an error if hooks is NOT using wizard provider as wrapper', () => {
     const { result } = renderHook(() => useWizardContext(), {
       wrapper: props => <div {...props} />,
@@ -90,6 +92,20 @@ describe('useWizard hook', () => {
     });
   });
 
+  it('call should be called if passed on `onNext`', async () => {
+    const { result, waitFor } = renderUseWizardHook();
+    expect(result.current).toBeDefined();
+
+    const handleNext = jest.fn();
+    act(() => {
+      result.current.onNext(handleNext);
+    });
+
+    await waitFor(() => {
+      expect(result.current.activeStepIndex).toBe(1);
+    });
+  });
+
   it('should not break if user goes above or below available steps', () => {
     const { result } = renderUseWizardHook();
     expect(result.current).toBeDefined();

src/useWizard.tsx

@@ -33,7 +33,7 @@ interface WizardStepperContextProps<T = DefaultWizardStepProps> {
   readonly isFirstStep: boolean;
   readonly isLastStep: boolean;
   goTo: (id: number | string) => void;
-  onNext: () => void;
+  onNext: (cb?: () => void) => void;
   getActiveStep: () => T;
   onPrevious: () => void;
   setSteps: (steps: T[] | T) => void;