test: fix one test re: #8383

Automattic · Nov 26, 2019 · e6616d7 · e6616d7
1 parent 68cdc33
commit e6616d7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/connection.js b/lib/connection.js
@@ -670,6 +670,7 @@ Connection.prototype.openUri = function(uri, options, callback) {
       if (options.useUnifiedTopology) {
         if (type === 'Single') {
           const server = Array.from(db.s.topology.s.servers.values())[0];
+
           server.s.pool.on('close', () => {
             _this.readyState = STATES.disconnected;
           });
@@ -682,6 +683,11 @@ Connection.prototype.openUri = function(uri, options, callback) {
           server.s.pool.on('timeout', () => {
             _this.emit('timeout');
           });
+          server.s.pool.on('drain', err => {
+            if (err && err.name === 'MongoNetworkError') {
+              _this.emit('timeout');
+            }
+          });
         } else if (type.startsWith('ReplicaSet')) {
           db.on('close', function() {
             const type = get(db, 's.topology.s.description.type', '');