support options for changesStream.close

Author: W-A-James

src/legacy_wrappers/change_stream.js

@@ -7,8 +7,15 @@ Object.defineProperty(module.exports, '__esModule', { value: true });
 
 module.exports.makeLegacyChangeStream = function (baseClass) {
   class LegacyChangeStream extends baseClass {
-    close(callback) {
-      return maybeCallback(super.close(), callback);
+    close(options, callback) {
+      callback =
+        typeof callback === 'function'
+          ? callback
+          : typeof options === 'function'
+            ? options
+            : undefined;
+      options = typeof options !== 'function' ? options : undefined;
+      return maybeCallback(super.close(options), callback);
     }
     hasNext(callback) {
       return maybeCallback(super.hasNext(), callback);

test/unit/legacy_wrappers/change_streams.test.js

@@ -0,0 +1,36 @@
+'use strict';
+
+const { MongoClient: LegacyMongoClient } = require('../../../src/index');
+const mongodbDriver = require('mongodb');
+const sinon = require('sinon');
+const { expect } = require('chai');
+
+const iLoveJs = 'mongodb://iLoveJavascript';
+
+describe('legacy_wrappers/change_streams.js', () => {
+  let client;
+  let changeStream;
+  beforeEach(async function () {
+    client = new LegacyMongoClient(iLoveJs);
+    changeStream = client.db('test').collection('test').watch();
+  });
+
+  afterEach(async function () {
+    sinon.restore();
+    await client.close();
+  });
+
+  context('close', function () {
+    it('correctly handles parameters when options are provided', function () {
+      const spy = sinon.spy(mongodbDriver.ChangeStream.prototype, 'close');
+      const opts = { timeoutMS: 100 };
+      changeStream.close(opts, () => {});
+      expect(spy).to.be.calledWithExactly(opts);
+    });
+    it('correctly handles parameters when options are not provided', function () {
+      const spy = sinon.spy(mongodbDriver.ChangeStream.prototype, 'close');
+      changeStream.close(() => {});
+      expect(spy).to.be.calledWithExactly(undefined);
+    });
+  });
+});