✨ handle useLayoutEffect react hook

FBerthelot · Apr 19, 2019 · b6ffc7a · b6ffc7a
1 parent 72ed91d
commit b6ffc7a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/packages/component-test-utils-react/src/dispatcher/index.js b/packages/component-test-utils-react/src/dispatcher/index.js
@@ -127,6 +127,10 @@ class Dispatcher {
       ...objectBuilder()
     };
   }
+
+  useLayoutEffect(...args) {
+    return this.useEffect(...args);
+  }
 }
 
 exports.createDispatcher = shallowedComponent => {

diff --git a/packages/component-test-utils-react/src/dispatcher/index.spec.js b/packages/component-test-utils-react/src/dispatcher/index.spec.js
@@ -265,4 +265,17 @@ describe('dispatcher', () => {
       expect(objectBuilder).toHaveBeenCalledTimes(2);
     });
   });
+
+  describe('useLayoutEffect', () => {
+    it('should call useEffect', () => {
+      dispatcher.useEffect = jest.fn(() => 4);
+
+      const args = [jest.fn(), [1, 4]];
+
+      const res = dispatcher.useLayoutEffect(...args);
+
+      expect(dispatcher.useEffect).toHaveBeenCalledWith(...args);
+      expect(res).toBe(4);
+    });
+  });
 });