Add memoization to useAwaitAction (#1)

* Add memoization to useAwaitAction

* Remove type assertion

* Add test for memoized hook

* Add space to jsdoc comment

src/useAwaitAction.spec.tsx

@@ -1,4 +1,4 @@
-import React, { useEffect } from 'react';
+import React, { useEffect, useState } from 'react';
 import { connect, DispatchProp, Provider } from 'react-redux';
 import { act } from 'react-dom/test-utils';
 import { render } from '@testing-library/react';
@@ -213,4 +213,38 @@ describe('useAwaitAction', () => {
 
         console.error = errorHandler; // eslint-disable-line no-console
     });
+
+    it('Should not re-initialize when storeAwait when component re-renders', async () => {
+        const renderCounter = jest.fn();
+
+        /**
+         * -
+         * @returns TestComponent
+         */
+        const TestComponent = (): any => {
+            const storeAwait = useAwaitAction();
+            const [, triggerRender] = useState<null>();
+
+            useEffect(() => {
+                triggerRender(null);
+            });
+
+            useEffect(() => {
+                renderCounter();
+            }, [storeAwait]);
+
+            return null;
+        };
+
+        act(() => {
+            render(
+                <Provider store={store}>
+                    <StoreAwaitProvider emitter={eventEmitter}>
+                        <TestComponent />
+                    </StoreAwaitProvider>
+                </Provider>,
+            );
+        });
+        expect(renderCounter).toHaveBeenCalledTimes(1);
+    });
 });

src/useAwaitAction.ts

@@ -1,5 +1,5 @@
+import { useMemo, useContext } from 'react';
 import { Action } from 'redux';
-import { useContext } from 'react';
 import { useStore } from 'react-redux';
 import { AwaitEventEmitterContext } from './context/awaitEventEmitterContext';
 import { ActionTypes, createAwaitAction } from './createAwaitAction';
@@ -15,5 +15,10 @@ export function useAwaitAction<S extends Action = Action>(): (s: ActionTypes, e?
 
     if (!eventEmitter) throw new Error('You need to access the await inside the <StoreAwaitProvider> component');
 
-    return createAwaitAction(store, eventEmitter);
+    const awaitAction: (s: ActionTypes, e?: ActionTypes | undefined) => Promise<S> = useMemo(
+        () => createAwaitAction(store, eventEmitter),
+        [store, eventEmitter],
+    );
+
+    return awaitAction;
 }