common: support checking custom types (#1529)

stephenplusplus · callmehiphop · commit 77c83a15cbd7 · 2016-08-25T20:03:15.000-04:00
* common: support checking custom types

* test parent module matching

* simplification
diff --git a/packages/common/src/util.js b/packages/common/src/util.js
@@ -599,3 +599,29 @@ function createLimiter(makeRequestFn, options) {
 }
 
 util.createLimiter = createLimiter;
+
+function isCustomType(unknown, module) {
+  function getConstructorName(obj) {
+    return obj.constructor && obj.constructor.name.toLowerCase();
+  }
+
+  var moduleNameParts = module.split('/');
+
+  var parentModuleName = moduleNameParts[0] && moduleNameParts[0].toLowerCase();
+  var subModuleName = moduleNameParts[1] && moduleNameParts[1].toLowerCase();
+
+  if (subModuleName && getConstructorName(unknown) !== subModuleName) {
+    return false;
+  }
+
+  var walkingModule = unknown;
+  do {
+    if (getConstructorName(walkingModule) === parentModuleName) {
+      return true;
+    }
+  } while ((walkingModule = walkingModule.parent));
+
+  return false;
+}
+
+util.isCustomType = isCustomType;
diff --git a/packages/common/test/util.js b/packages/common/test/util.js
@@ -1370,4 +1370,35 @@ describe('common/util', function() {
       });
     });
   });
+
+  describe('isCustomType', function() {
+    function PubSub() {}
+
+    function MiddleLayer() {
+      this.parent = new PubSub();
+    }
+
+    function Subscription() {
+      this.parent = new MiddleLayer();
+    }
+
+    var pubsub = new PubSub();
+    var subscription = new Subscription();
+
+    it('should match a Service type by constructor names', function() {
+      assert(util.isCustomType(pubsub, 'pubsub'));
+    });
+
+    it('should match a ServiceObject type by constructor names', function() {
+      assert(util.isCustomType(subscription, 'pubsub'));
+      assert(util.isCustomType(subscription, 'pubsub/subscription'));
+
+      assert(util.isCustomType(subscription, 'middlelayer'));
+      assert(util.isCustomType(subscription, 'middlelayer/subscription'));
+    });
+
+    it('should support any casing', function() {
+      assert(util.isCustomType(subscription, 'PubSub/Subscription'));
+    });
+  });
 });
