fix(connection): bubble up connected/disconnected events with unified…

… topology
Automattic · Nov 13, 2019 · 4ec852d · 4ec852d
1 parent 98facde
commit 4ec852d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/connection.js b/lib/connection.js
@@ -643,6 +643,12 @@ 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;
+          });
+          server.s.topology.on('serverHeartbeatSucceeded', () => {
+            _handleReconnect();
+          });
           server.s.pool.on('reconnect', () => {
             _handleReconnect();
           });

diff --git a/test/connection.test.js b/test/connection.test.js
@@ -189,7 +189,7 @@ describe('connections:', function() {
         let numReconnected = 0;
         let numReconnect = 0;
         let numClose = 0;
-        const conn = mongoose.createConnection('mongodb://localhost:27000/mongoosetest', {
+        const conn = mongoose.createConnection('mongodb://localhost:27000/mongoosetest?heartbeatfrequencyms=1000', {
           useNewUrlParser: true,
           useUnifiedTopology: true
         });