Reset timeout onUnmount

Author: cvle

src/composables/withTimeout.spec.tsx

@@ -97,4 +97,24 @@ describe("withTimeout", () => {
       }, halfPeriod);
     }, halfPeriod);
   });
+
+  it("should not call timeout after unmount", (done) => {
+    const props = {
+      transitionState: {
+        inTransition: false,
+      },
+      transitionInfo: {
+        totalDuration: 20,
+      },
+      timeout: spy(),
+    };
+    const timeoutIn = props.transitionInfo.totalDuration * timeoutMultiplier;
+    const wrapper = mount(<Assembly {...props} />);
+    wrapper.setProps({ transitionState: { inTransition: true } });
+    wrapper.unmount();
+    setTimeout(() => {
+      assert.isFalse(props.timeout.called);
+      done();
+    }, timeoutIn);
+  });
 });

src/composables/withTimeout.ts

@@ -1,4 +1,4 @@
-import { withHandlers, isolate, onWillReceiveProps, debug } from "reassemble";
+import { withHandlers, isolate, onWillReceiveProps, onWillUnmount } from "reassemble";
 
 import { CSSTransitionProps } from "../csstransition";
 import { WithTransitionStateProps } from "./withTransitionState";
@@ -45,4 +45,5 @@ export const withTimeout =
           }
         }
       }),
+    onWillUnmount<PropsUnion>(({ cancel }) => cancel()),
   );