fix: copy connection over to MongooseThenable

Fix #3972
Automattic · Mar 18, 2016 · 1441e6a · 1441e6a
1 parent 12d3b53
commit 1441e6a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -473,6 +473,10 @@ Mongoose.prototype.__defineGetter__('connection', function() {
   return this.connections[0];
 });
 
+Mongoose.prototype.__defineSetter__('connection', function(v) {
+  this.connections[0] = v;
+});
+
 /*!
  * Driver depentend APIs
  */
@@ -731,6 +735,8 @@ function MongooseThenable(mongoose, promise) {
           return mongoose[key].apply(mongoose, arguments);
         };
       })(key);
+    } else if (['connection', 'connections'].indexOf(key) !== -1) {
+      _this[key] = mongoose[key];
     }
   }
   this.$opPromise = promise;