added some minor tests for single server/replicaset for disconnect

Author: christkv

lib/collection.js

@@ -513,9 +513,7 @@ Collection.prototype.insertMany = function(docs, options, callback) {
 
   // Return a Promise
   return new this.s.promiseLibrary(function(resolve, reject) {
-    // console.log("########## insertMany 0")
     bulkWrite(self, operations, options, function(err, r) {
-      // console.log("########## insertMany 1")
       if(err) return reject(err);
       resolve(mapInserManyResults(docs, r));
     });
@@ -2478,8 +2476,10 @@ var findAndRemove = function(self, query, sort, options, callback) {
 define.classMethod('findAndRemove', {callback: true, promise:true});
 
 function decorateWithWriteConcern(command, self, options) {
+  // Do we support collation 3.4 and higher
+  var capabilities = self.s.topology.capabilities();
   // Do we support write concerns 3.4 and higher
-  if(self.s.topology.capabilities().commandsTakeWriteConcern) {
+  if(capabilities && capabilities.commandsTakeWriteConcern) {
     // Get the write concern settings
     var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
     // Add the write concern to the command
@@ -2491,7 +2491,9 @@ function decorateWithWriteConcern(command, self, options) {
 
 function decorateWithCollation(command, self, options) {
   // Do we support collation 3.4 and higher
-  if(self.s.topology.capabilities().commandsTakeCollation) {
+  var capabilities = self.s.topology.capabilities();
+  // Do we support write concerns 3.4 and higher
+  if(capabilities && capabilities.commandsTakeCollation) {
     if(options.collation && typeof options.collation == 'object') {
       command.collation = options.collation;
     } else if(self.s.collation && typeof self.s.collation == 'object') {

test/functional/disconnect_handler_tests.js

@@ -0,0 +1,33 @@
+/**
+ * @ignore
+ */
+exports['Should correctly recover when bufferMaxEntries: -1 and restart'] = {
+  metadata: { requires: { topology: ['single', 'replicaset'] }, ignore: { travis:true } },
+
+  // The actual test we wish to run
+  test: function(configuration, test) {
+    var MongoClient = configuration.require.MongoClient
+      , f = require('util').format;
+
+    var done = false;
+
+    MongoClient.connect(configuration.url(), {}, function(err, db) {
+      test.equal(null, err);
+
+      configuration.manager.stop(9).then(function() {
+        db.collection('disconnect_handler_tests').update({a:1}, {$set: {b:1}}, function(err, r) {
+          test.equal(null, err);
+          test.equal(0, r.result.nModified);
+
+          db.close();
+          test.done();
+        });
+
+        setTimeout(function() {
+          configuration.manager.restart(9, {waitMS: 5000}).then(function() {
+          });
+        }, 5000)
+      });
+    });
+  }
+}

test/runner.js

@@ -298,7 +298,8 @@ var testFiles = [
   , '/test/functional/mongo_client_options_tests.js'
   , '/test/functional/decimal128_tests.js'
   , '/test/functional/find_and_modify_tests.js'
-  , '/test/functional/hang_tests.js'
+  , '/test/functional/hang_tests.js',
+  , '/test/functional/disconnect_handler_tests.js',
 
   // Replicaset tests
   , '/test/functional/replset_read_preference_tests.js'