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;