test(fix): fix broken test after deps upgrade

affects: @tao.js/react
zzyzxlab · Feb 5, 2022 · efe3c37 · efe3c37
1 parent a63d50f
commit efe3c37
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/packages/react-tao/src/Reactor.js b/packages/react-tao/src/Reactor.js
@@ -23,7 +23,7 @@ class Reactor extends Component {
 
   componentWillUnmount() {
     const { adapter } = this.props;
-    adapter.unregisterReactor(this);
+    adapter && adapter.unregisterReactor && adapter.unregisterReactor(this);
   }
 
   componentDidUpdate(prevProps) {

diff --git a/packages/react-tao/test/Reactor.spec.js b/packages/react-tao/test/Reactor.spec.js
@@ -91,6 +91,8 @@ describe('Reactor exports a React Component for reacting to TAO App Contexts', (
       wrapper.instance(),
       expect.any(Function)
     );
+    registerMock.mockReset();
+    registerMock.mockRestore();
   });
 
   it('should unregister from the Adapter when unmounting', () => {
@@ -106,6 +108,8 @@ describe('Reactor exports a React Component for reacting to TAO App Contexts', (
     // Assert
     expect(unregisterMock).toHaveBeenCalledTimes(1);
     expect(unregisterMock).toHaveBeenCalledWith(wrapperInstance);
+    unregisterMock.mockReset();
+    unregisterMock.mockRestore();
   });
 
   it('should not render a component without an AC being triggered', () => {