diff --git a/packages/enzyme-test-suite/test/shared/hooks/useState.jsx b/packages/enzyme-test-suite/test/shared/hooks/useState.jsx
index 91523f32c..2f5e87ef0 100644
--- a/packages/enzyme-test-suite/test/shared/hooks/useState.jsx
+++ b/packages/enzyme-test-suite/test/shared/hooks/useState.jsx
@@ -55,6 +55,33 @@ export default function describeUseState({
expect(wrapper.find('.counter').text()).to.equal(String(initialCount - 1));
});
+ it('handles useState', () => {
+ function ComponentUsingStateHook() {
+ const [count] = useState(0);
+ return
{count}
;
+ }
+
+ const wrapper = Wrap();
+
+ expect(wrapper.find('div').length).to.equal(1);
+ expect(wrapper.find('div').text()).to.equal('0');
+ });
+
+ it('handles setState returned from useState', () => {
+ function ComponentUsingStateHook() {
+ const [count, setCount] = useState(0);
+ return setCount(count + 1)}>{count}
;
+ }
+
+ const wrapper = Wrap();
+ const div = wrapper.find('div');
+ const setCount = div.prop('onClick');
+ setCount();
+ wrapper.update();
+
+ expect(wrapper.find('div').text()).to.equal('1');
+ });
+
describe('useState with willReceive prop effect / simulate getDerivedStateFromProp', () => {
const newPropCount = 10;