[Tests] ensure that the setState callback is actually called

enzymejs · Jan 6, 2018 · c262c69 · c262c69
1 parent 847b59b
commit c262c69
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -1381,11 +1381,14 @@ describe('shallow', () => {
       }
       const wrapper = shallow(<Foo />);
       expect(wrapper.state()).to.eql({ id: 'foo' });
-      wrapper.setState({ id: 'bar' }, function callback(...args) {
-        expect(wrapper.state()).to.eql({ id: 'bar' });
-        expect(this.state).to.eql({ id: 'bar' });
-        expect(wrapper.find('div').prop('className')).to.eql('bar');
-        expect(args).to.eql(CALLING_SETSTATE_CALLBACK_WITH_UNDEFINED ? [undefined] : []);
+      return new Promise((resolve) => {
+        wrapper.setState({ id: 'bar' }, function callback(...args) {
+          expect(wrapper.state()).to.eql({ id: 'bar' });
+          expect(this.state).to.eql({ id: 'bar' });
+          expect(wrapper.find('div').prop('className')).to.eql('bar');
+          expect(args).to.eql(CALLING_SETSTATE_CALLBACK_WITH_UNDEFINED ? [undefined] : []);
+          resolve();
+        });
       });
     });