From 03318c8952f902f9abe10a15c6653b571cff9340 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=B3=E4=B9=99=E5=B1=B1?= <pipio1994@gmail.com>
Date: Thu, 7 Feb 2019 17:15:12 +0800
Subject: [PATCH] [Tests]: add useState test case in shallow for hooks

---
 .../test/ShallowWrapper-spec.jsx              | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
index 3947654d2..403f7b1d9 100644
--- a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
+++ b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -2012,6 +2012,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();