[Tests]: add failed useState test case in shallow for hooks

enzymejs · Feb 28, 2019 · dee77bd · dee77bd
1 parent 9faac9d
commit dee77bd
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -31,6 +31,7 @@ import {
   forwardRef,
   memo,
   PureComponent,
+  useState,
 } from './_helpers/react-compat';
 import {
   describeIf,
@@ -1534,6 +1535,35 @@ describe('shallow', () => {
       });
     });
 
+    describeIf(is('>= 16.8'), 'hooks', () => {
+      it('handles useState', () => {
+        const ComponentUsingStateHook = () => {
+          const [count] = useState(0);
+          return <div>{count}</div>;
+        };
+
+        const wrapper = shallow(<ComponentUsingStateHook />);
+
+        expect(wrapper.find('div').length).to.equal(1);
+        expect(wrapper.find('div').text()).to.equal('0');
+      });
+
+      it('handles setState returned from useState', () => {
+        const ComponentUsingStateHook = () => {
+          const [count, setCount] = useState(0);
+          return <div onClick={() => setCount(count + 1)}>{count}</div>;
+        };
+
+        const wrapper = shallow(<ComponentUsingStateHook />);
+        const div = wrapper.find('div');
+        const setCount = div.prop('onClick');
+        setCount();
+        wrapper.update();
+
+        expect(wrapper.find('div').text()).to.equal('1');
+      });
+    });
+
     describeWithDOM('find DOM elements by constructor', () => {
       const { elements, all } = getData();