feat: allow to fake dispatch

atheck · Aug 11, 2022 · ca335ae · ca335ae
1 parent 8e87592
commit ca335ae
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 15 deletions.
diff --git a/src/Testing/fakeInitResult.ts b/src/Testing/fakeInitResult.ts
@@ -1,22 +1,34 @@
 import { InitResult, MessageBase, Nullable } from "../Types";
+import { Dispatch } from "../Cmd";
 
 type FakeInitResult = Nullable<[unknown, unknown?]>;
 
-let initResult: FakeInitResult;
+let currentFakeInitResult: FakeInitResult;
+let currentFakeDispatch: Nullable<Dispatch<MessageBase>>;
 
-function setFakeInitResult (model: FakeInitResult): void {
-    initResult = model;
+function setFakes (fakeInitResult: FakeInitResult, fakeDispatch?: Nullable<Dispatch<MessageBase>>): void {
+    currentFakeInitResult = fakeInitResult;
+    currentFakeDispatch = fakeDispatch ?? null;
 }
 
 function getFakeInitResultOnce<TModel, TMessage extends MessageBase> (): Nullable<InitResult<TModel, TMessage>> {
-    const temp = initResult as Nullable<InitResult<TModel, TMessage>>;
+    const temp = currentFakeInitResult as Nullable<InitResult<TModel, TMessage>>;
 
-    initResult = null;
+    currentFakeInitResult = null;
+
+    return temp;
+}
+
+function getFakeDispatchOnce<TMessage extends MessageBase> (): Nullable<Dispatch<TMessage>> {
+    const temp = currentFakeDispatch;
+
+    currentFakeDispatch = null;
 
     return temp;
 }
 
 export {
-    setFakeInitResult,
+    setFakes,
     getFakeInitResultOnce,
+    getFakeDispatchOnce,
 };
diff --git a/src/Testing/index.ts b/src/Testing/index.ts
@@ -1,7 +1,7 @@
+import { Cmd, Dispatch } from "../Cmd";
 import { InitResult, MessageBase, Nullable, UpdateMap, UpdateReturnType } from "../Types";
 import { callUpdateMap } from "../useElmish";
-import { Cmd } from "../Cmd";
-import { setFakeInitResult } from "./fakeInitResult";
+import { setFakes } from "./fakeInitResult";
 
 /**
  * Extracts the messages out of a command.
@@ -87,12 +87,12 @@ function createUpdateArgsFactory <TProps, TModel, TMessage extends MessageBase>
     };
 }
 
-function renderWithFakeInit<TModel, TMessage extends MessageBase, TResult = unknown> (initResult: InitResult<TModel, TMessage>, render: () => TResult): TResult {
-    setFakeInitResult(initResult);
+function renderWithFake<TModel, TMessage extends MessageBase, TResult = unknown> (initResult: InitResult<TModel, TMessage>, render: () => TResult, fakeDispatch?: Dispatch<TMessage>): TResult {
+    setFakes(initResult, (fakeDispatch ?? null) as Nullable<Dispatch<unknown>>);
 
     const result = render();
 
-    setFakeInitResult(null);
+    setFakes(null, null);
 
     return result;
 }
@@ -106,5 +106,5 @@ export {
     execCmd,
     getUpdateFn,
     createUpdateArgsFactory,
-    renderWithFakeInit,
+    renderWithFake,
 };
diff --git a/src/useElmish.ts b/src/useElmish.ts
@@ -1,8 +1,8 @@
 /* eslint-disable react-hooks/exhaustive-deps */
 import { Cmd, Dispatch } from "./Cmd";
+import { getFakeDispatchOnce, getFakeInitResultOnce } from "./Testing/fakeInitResult";
 import { InitFunction, MessageBase, Nullable, UpdateFunction, UpdateMap, UpdateReturnType } from "./Types";
 import { useCallback, useEffect, useRef, useState } from "react";
-import { getFakeInitResultOnce } from "./Testing/fakeInitResult";
 import { Services } from "./Init";
 
 type SubscriptionResult<TMessage> = [Cmd<TMessage>, (() => void)?];
@@ -47,7 +47,8 @@ function useElmish<TProps, TModel, TMessage extends MessageBase> ({ name, props,
         });
     }, []);
 
-    const dispatch = useCallback((msg: TMessage): void => {
+    const fakeDispatch = getFakeDispatchOnce();
+    const dispatch = useCallback(fakeDispatch ?? ((msg: TMessage): void => {
         if (!initializedModel) {
             return;
         }
@@ -100,7 +101,7 @@ function useElmish<TProps, TModel, TMessage extends MessageBase> ({ name, props,
                 });
             }
         }
-    }, []);
+    }), []);
 
     if (!initializedModel) {
         const [initModel, initCmd] = getFakeInitResultOnce<TModel, TMessage>() ?? init(props);