Fix tests; use ifReact in the 15.4 adapter

Author: ljharb

packages/enzyme-adapter-react-15.4/package.json

@@ -34,6 +34,7 @@
   "author": "Leland Richardson <[email protected]>",
   "license": "MIT",
   "dependencies": {
+    "enzyme-adapter-react-helper": "^1.2.2",
     "enzyme-adapter-utils": "^1.3.0",
     "lodash": "^4.17.4",
     "object.assign": "^4.1.0",

packages/enzyme-adapter-react-15.4/src/ReactFifteenFourAdapter.js

@@ -16,8 +16,7 @@ import {
   createMountWrapper,
   propsWithKeysAndRef,
 } from 'enzyme-adapter-utils';
-
-const MINOR_VERSION = React.version.split('.')[1];
+import ifReact from 'enzyme-adapter-react-helper/build/ifReact';
 
 function compositeTypeToNodeType(type) {
   switch (type) {
@@ -257,11 +256,11 @@ class ReactFifteenFourAdapter extends EnzymeAdapter {
 
   invokeSetStateCallback(instance, callback) {
     // React in >= 15.4, and < 16 pass undefined to a setState callback
-    if (MINOR_VERSION >= 4) {
-      callback.call(instance, undefined);
-    } else {
-      super.invokeSetStateCallback(instance, callback);
-    }
+    ifReact(
+      '^15.4',
+      () => { callback.call(instance, undefined); },
+      () => { super.invokeSetStateCallback(instance, callback); },
+    )
   }
 }
 

packages/enzyme-test-suite/package.json

@@ -49,4 +49,4 @@
     "eslint-plugin-jsx-a11y": "^6.0.3",
     "eslint-plugin-react": "^7.5.1"
   }
-}
+}

packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx

@@ -8,16 +8,14 @@ import { ITERATOR_SYMBOL, withSetStateAllowed, sym } from 'enzyme/build/Utils';
 import './_helpers/setupAdapters';
 import { createClass } from './_helpers/react-compat';
 import { describeIf, itIf, itWithData, generateEmptyRenderData } from './_helpers';
-import { REACT013, REACT014, REACT15, REACT150, REACT151, REACT152, REACT153, REACT16, is } from './_helpers/version';
+import { REACT013, REACT014, REACT15, REACT150_4, REACT16, is } from './_helpers/version';
 
 // The shallow renderer in react 16 does not yet support batched updates. When it does,
 // we should be able to go un-skip all of the tests that are skipped with this flag.
 const BATCHING = !REACT16;
 
-
 // some React versions pass undefined as an argument of setState callback.
-const CALLING_SETSTATE_CALLBACK_WITH_UNDEFINED =
-  REACT15 && !REACT150 && !REACT151 && !REACT152 && !REACT153;
+const CALLING_SETSTATE_CALLBACK_WITH_UNDEFINED = REACT15 && !REACT150_4;
 
 const getElementPropSelector = prop => x => x.props[prop];
 const getWrapperPropSelector = prop => x => x.prop(prop);

packages/enzyme-test-suite/test/_helpers/setupAdapters.js

@@ -5,20 +5,26 @@
  * version of React is loaded, and configures enzyme to use the right
  * corresponding adapter.
  */
-const Version = require('./version');
+const {
+  REACT013,
+  REACT014,
+  REACT15,
+  REACT155,
+  REACT16,
+} = require('./version');
 const Enzyme = require('enzyme');
 
 let Adapter = null;
 
-if (Version.REACT013) {
+if (REACT013) {
   Adapter = require('enzyme-adapter-react-13');
-} else if (Version.REACT014) {
+} else if (REACT014) {
   Adapter = require('enzyme-adapter-react-14');
-} else if (Version.REACT155) {
+} else if (REACT155) {
   Adapter = require('enzyme-adapter-react-15');
-} else if (Version.REACT15) {
+} else if (REACT15) {
   Adapter = require('enzyme-adapter-react-15.4');
-} else if (Version.REACT16) {
+} else if (REACT16) {
   Adapter = require('enzyme-adapter-react-16');
 }
 

packages/enzyme-test-suite/test/_helpers/version.js

@@ -8,10 +8,8 @@ const [major, minor] = VERSION.split('.');
 export const REACT013 = VERSION.slice(0, 4) === '0.13';
 export const REACT014 = VERSION.slice(0, 4) === '0.14';
 export const REACT15 = major === '15';
-export const REACT150 = REACT15 && minor === '0';
-export const REACT151 = REACT15 && minor === '1';
-export const REACT152 = REACT15 && minor === '2';
-export const REACT153 = REACT15 && minor === '3';
+export const REACT150_4 = REACT15 && minor < 5;
+export const REACT155 = REACT15 && minor >= 5;
 export const REACT16 = major === '16';
 
 export function gt(v) { return semver.gt(VERSION, v); }