Ensuring iteratee follows Immutable strategy. Fixes #846

Author: davidkpiano

src/utils/iteratee.js

@@ -1,34 +1,44 @@
 import identity from './identity';
+import _get from '../utils/get';
 
-function matcher(object) {
-  return (compareObject) => {
-    if (compareObject === object) return true;
+const defaultStrategy = {
+  get: _get,
+};
 
-    return Object.keys(object)
-      .every((key) => object[key] === compareObject[key]);
-  };
-}
 
-function propChecker(prop) {
-  return (object) => object && !!object[prop];
-}
+export function createIteratee(s = defaultStrategy) {
+  function matcher(object) {
+    return (compareObject) => {
+      if (compareObject === object) return true;
 
-export default function iteratee(value) {
-  if (typeof value === 'function') {
-    return value;
+      return Object.keys(object)
+        .every((key) => s.get(object, key) === s.get(compareObject, key));
+    };
   }
 
-  if (value === null) {
-    return identity;
+  function propChecker(prop) {
+    return (object) => object && !!s.get(object, prop);
   }
 
-  if (typeof value === 'object') {
-    return matcher(value);
-  }
+  return (value) => {
+    if (typeof value === 'function') {
+      return value;
+    }
 
-  return propChecker(value);
+    if (value === null) {
+      return identity;
+    }
+
+    if (typeof value === 'object') {
+      return matcher(value);
+    }
+
+    return propChecker(value);
+  };
 }
 
+const iteratee = createIteratee();
+
 export function iterateeValue(data, value) {
   if (typeof value === 'function') {
     return value(data);
@@ -42,3 +52,5 @@ export function iterateeValue(data, value) {
 
   return iteratee(value)(data);
 }
+
+export default iteratee;

src/utils/track.js

@@ -1,13 +1,15 @@
 import findKey from '../utils/find-key';
 import _get from '../utils/get';
-import iteratee from '../utils/iteratee';
+import { createIteratee } from '../utils/iteratee';
 import isMulti from '../utils/is-multi';
 
 const defaultStrategy = {
   get: _get,
 };
 
 function createTrack(s = defaultStrategy) {
+  const iteratee = createIteratee(s);
+
   return function track(model, ...predicates) {
     const isPartial = model[0] === '.';