Validate shape of cacheMap on initialization

src/__tests__/abuse-test.js

@@ -150,4 +150,17 @@ describe('Provides descriptive error messages for API abuse', () => {
     );
   });
 
+  it('Cache should have get, set, delete, and clear methods', async () => {
+    class IncompleteMap {
+      get() {}
+    }
+
+    expect(() => {
+      var incompleteMap = new IncompleteMap();
+      var options = { cacheMap: incompleteMap };
+      new DataLoader(keys => keys, options); // eslint-disable-line no-new
+    }).to.throw(
+      'Custom cacheMap missing methods: set, delete, clear'
+    );
+  });
 });

src/index.js

@@ -53,8 +53,7 @@ export default class DataLoader<K, V> {
     }
     this._batchLoadFn = batchLoadFn;
     this._options = options;
-    this._promiseCache =
-      options && options.cacheMap || (new Map(): Map<K,Promise<V>>);
+    this._promiseCache = getValidCacheMap(options);
     this._queue = [];
   }
 
@@ -305,6 +304,24 @@ function failedDispatch<K, V>(
   });
 }
 
+function getValidCacheMap<K, V>(
+  options: ?Options<K, V>
+): CacheMap<K, Promise<V>> {
+  var cacheMap = options && options.cacheMap;
+  if (!cacheMap) {
+    return new Map();
+  }
+  var cacheFunctions = [ 'get', 'set', 'delete', 'clear' ];
+  var missingFunctions = cacheFunctions
+    .filter(fnName => cacheMap && typeof cacheMap[fnName] !== 'function');
+  if (missingFunctions.length !== 0) {
+    throw new TypeError(
+      'Custom cacheMap missing methods: ' + missingFunctions.join(', ')
+    );
+  }
+  return cacheMap;
+}
+
 // Private
 type LoaderQueue<K, V> = Array<{
   key: K;