Use batchedUpdates for setProps test

enzymejs · Oct 2, 2016 · 4ed9c95 · 4ed9c95
1 parent 5a2fe98
commit 4ed9c95
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/ReactWrapper-spec.jsx b/test/ReactWrapper-spec.jsx
@@ -3,6 +3,7 @@
 import React from 'react';
 import { expect } from 'chai';
 import sinon from 'sinon';
+import { batchedUpdates } from '../src/react-compat';
 
 import {
   describeWithDOM,
@@ -971,8 +972,12 @@ describeWithDOM('mount', () => {
       }
       const wrapper = mount(<Foo id="foo" />);
       expect(wrapper.find('.foo').length).to.equal(1);
-      wrapper.setProps({ id: 'bar', foo: 'bla' }, () => {
-        expect(wrapper.find('.bar').length).to.equal(1);
+
+      batchedUpdates(() => {
+        wrapper.setProps({ id: 'bar', foo: 'bla' }, () => {
+          expect(wrapper.find('.bar').length).to.equal(1);
+        });
+        expect(wrapper.find('.bar').length).to.equal(0);
       });
     });