test: breaking changes

Author: nbbeeken

test/functional/connection.test.js

@@ -3,6 +3,7 @@ const { withClient, setupDatabase } = require('./shared');
 const test = require('./shared').assert;
 const { expect } = require('chai');
 const { ServerHeartbeatStartedEvent } = require('../../src');
+const { Topology } = require('../../src/sdam/topology');
 
 describe('Connection - functional', function () {
   before(function () {
@@ -70,21 +71,16 @@ describe('Connection - functional', function () {
     }
   });
 
-  it('should correctly connect to server using just events', {
-    metadata: {
-      requires: { topology: ['single', 'replicaset', 'sharded', 'ssl', 'heap', 'wiredtiger'] }
-    },
-
-    test: function (done) {
-      var configuration = this.configuration;
-      var client = configuration.newClient({ w: 1 }, { maxPoolSize: 1 });
+  it('should correctly connect to server using just events', function (done) {
+    var configuration = this.configuration;
+    var client = configuration.newClient({ w: 1 }, { maxPoolSize: 1 });
 
-      client.on('open', function () {
-        client.close(done);
-      });
+    client.on('open', topology => {
+      expect(topology).to.be.instanceOf(Topology);
+      client.close(done);
+    });
 
-      client.connect();
-    }
+    client.connect();
   });
 
   it('should correctly connect to server using big connection pool', {

test/unit/sdam/srv_polling.test.js

@@ -425,4 +425,35 @@ describe('Mongos SRV Polling', function () {
       });
     });
   });
+
+  it('should emit a topologyDescriptionChanged event upon receiving an SRV update', function () {
+    const topology = new Topology(['localhost:27017'], { srvHost: SRV_HOST });
+
+    return new Promise(resolve => {
+      topology.on(Topology.TOPOLOGY_DESCRIPTION_CHANGED, ev => {
+        if (ev.topologyId === 'fake') {
+          // the fake description change emitted within this test
+          expect(topology.s.handleSrvPolling).to.be.a('function');
+          expect(ev.newDescription.servers.size).to.equal(0);
+          topology.s.handleSrvPolling(
+            new SrvPollingEvent([srvRecord('localhost', 27017), srvRecord('localhost', 27018)])
+          );
+        } else {
+          // topologyDescriptionChanged event emitted from the handleSrvPolling function
+          expect(ev.topologyId).to.equal(1);
+          expect(ev.newDescription.servers.size).to.equal(2);
+          resolve();
+        }
+      });
+
+      topology.emit(
+        Topology.TOPOLOGY_DESCRIPTION_CHANGED,
+        new sdamEvents.TopologyDescriptionChangedEvent(
+          'fake',
+          new TopologyDescription(TopologyType.Unknown),
+          new TopologyDescription(TopologyType.Sharded)
+        )
+      );
+    });
+  });
 });