Add callback in dispose

Author: appurva21

lib/postman-sandbox-fleet.js

@@ -159,13 +159,24 @@ class PostmanSandboxFleet {
     /**
      * Dispose off all initialized sandbox instances from the fleet
      *
+     * @param {Function} [callback] -
      * @returns {void}
      */
-    disposeAll () {
+    disposeAll (callback) {
+        let disposedCount = 0;
+
+        if (typeof callback !== 'function') {
+            callback = _.noop;
+        }
+
         this.fleet.forEach((context, templateName) => {
-            context.dispose();
+            context.dispose(() => {
+                this.fleet.delete(templateName);
 
-            this.fleet.delete(templateName);
+                if (++disposedCount === this.fleet.size) {
+                    return callback();
+                }
+            });
         });
     }
 }

lib/postman-sandbox.js

@@ -132,7 +132,7 @@ class PostmanSandbox extends UniversalVM {
         });
     }
 
-    dispose () {
+    dispose (callback) {
         this.once('dispose', () => {
             _.forEach(this._executing, (irq, id) => {
                 irq && clearTimeout(irq);
@@ -149,6 +149,8 @@ class PostmanSandbox extends UniversalVM {
             this.removeAllListeners(ASSERTION_EVENT_NAME);
             this.removeAllListeners(ERROR_EVENT_NAME);
             this.disconnect();
+
+            typeof callback === 'function' && callback();
         });
 
         this.dispatch('dispose');