update tests for react 16

enzymejs · Sep 26, 2017 · 9930f3e · 9930f3e
1 parent 664a99b
commit 9930f3e
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
@@ -3475,7 +3475,7 @@ describe('shallow', () => {
           },
         );
         wrapper.setState({ foo: 'baz' });
-        expect(spy.args).to.deep.equal([
+        const expected = [
           [
             'render',
           ],
@@ -3494,13 +3494,16 @@ describe('shallow', () => {
           [
             'render',
           ],
-          [
+        ];
+        if (!REACT16) {
+          expected.push([
             'componentDidUpdate',
             { foo: 'props' }, { foo: 'props' },
             { foo: 'bar' }, { foo: 'baz' },
-            REACT16 ? undefined : { foo: 'context' },
-          ],
-        ]);
+            { foo: 'context' },
+          ]);
+        }
+        expect(spy.args).to.deep.equal(expected);
       });
 
       it('should cancel rendering when Component returns false in shouldComponentUpdate', () => {