fix(model): modify discriminator() class in place

Fix #5175
Automattic · Apr 22, 2017 · a2addca · a2addca
1 parent 1237132
commit a2addca
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/model.js b/lib/model.js
@@ -829,8 +829,10 @@ Model.prototype.model = function model(name) {
  */
 
 Model.discriminator = function(name, schema) {
+  var model;
   if (typeof name === 'function') {
-    name = utils.getFunctionName(name);
+    model = name;
+    name = utils.getFunctionName(model);
   }
 
   schema = discriminator(this, name, schema);
@@ -840,7 +842,10 @@ Model.discriminator = function(name, schema) {
 
   schema.$isRootDiscriminator = true;
 
-  this.discriminators[name] = this.db.model(name, schema, this.collection.name);
+  if (!model) {
+    model = this.db.model(name, schema, this.collection.name);
+  }
+  this.discriminators[name] = model;
   var d = this.discriminators[name];
   d.prototype.__proto__ = this.prototype;
   Object.defineProperty(d, 'baseModelName', {