tests added

Author: aditi-khare-mongoDB

src/mongo_client.ts

@@ -367,7 +367,7 @@ export class MongoClient extends TypedEventEmitter<MongoClientEvents> implements
   /** @internal */
   private connectionLock?: Promise<this>;
   /** @internal */
-  private closeLock?: Promise<void>
+  private closeLock?: Promise<void>;
 
   /**
    * The consolidate, parsed, transformed and merged options.
@@ -642,14 +642,14 @@ export class MongoClient extends TypedEventEmitter<MongoClientEvents> implements
   async close(force = false): Promise<void> {
     if (this.closeLock) {
       return await this.closeLock;
-    } 
+    }
 
     try {
       this.closeLock = this._close(force);
       await this.closeLock;
     } finally {
       // release
-      this.connectionLock = undefined;
+      this.closeLock = undefined;
     }
   }
 

test/integration/node-specific/mongo_client.test.ts

@@ -722,6 +722,22 @@ describe('class MongoClient', function () {
         expect(client.s.sessionPool.sessions).to.have.lengthOf(1);
       });
     });
+
+    context('concurrent calls', () => {
+      context('when two concurrent calls to close() occur', () => {
+        it('no error is thrown', async function () {
+          expect(() => Promise.all([client.close(), client.close()])).to.not.throw;
+        });
+      });
+
+      context('when more than two concurrent calls to close() occur', () => {
+        it('no error is thrown', async function () {
+          expect(() =>
+            Promise.all([client.close(), client.close(), client.close(), client.close()])
+          ).to.not.throw;
+        });
+      });
+    });
   });
 
   context('when connecting', function () {

test/unit/mongo_client.test.ts

@@ -1223,4 +1223,22 @@ describe('MongoClient', function () {
       });
     });
   });
+
+  describe('closeLock', function () {
+    let client;
+
+    beforeEach(async function () {
+      client = new MongoClient('mongodb://blah');
+    });
+
+    it('when client.close is pending, client.closeLock is set', () => {
+      client.close();
+      expect(client.closeLock).to.be.instanceOf(Promise);
+    });
+
+    it('when client.close is not pending, client.closeLock is not set', async () => {
+      await client.close();
+      expect(client.closeLock).to.be.undefined;
+    });
+  });
 });