lib/registry: fix findModel for model ctor

Fix `registry.findModel(arg)` to support the case when `arg` is already a model constructor.
studykik · Nov 9, 2015 · 36cd5a7 · 36cd5a7
1 parent e633617
commit 36cd5a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/registry.js b/lib/registry.js
@@ -284,7 +284,7 @@ Registry.prototype._defineRemoteMethods = function(ModelCtor, methods) {
  * @header loopback.findModel(modelName)
  */
 Registry.prototype.findModel = function(modelName) {
-  if (typeof modelType === 'function') return modelName;
+  if (typeof modelName === 'function') return modelName;
   return this.modelBuilder.models[modelName];
 };
 

diff --git a/test/loopback.test.js b/test/loopback.test.js
@@ -245,6 +245,7 @@ describe('loopback', function() {
         assert(loopback.getModel('MyModel') === MyModel);
         assert(loopback.getModel('MyCustomModel') === MyCustomModel);
         assert(loopback.findModel('Invalid') === undefined);
+        assert(loopback.getModel(MyModel) === MyModel);
       });
       it('should be able to get model by type', function() {
         var MyModel = loopback.createModel('MyModel', {}, {