updated world.js after event to close the browser using promises

Author: John Doherty

package.json

@@ -1,6 +1,6 @@
 {
   "name": "selenium-cucumber-js",
-  "version": "1.2.7",
+  "version": "1.2.8",
   "description": "A debuggable JS BDD framework that uses the official selenium-webdriver for Node and cucumber-js",
   "main": "index.js",
   "scripts": {

runtime/world.js

@@ -173,22 +173,25 @@ module.exports = function () {
     });
 
     // executed after each scenario (always closes the browser to ensure fresh tests)
-    this.After(function (scenario, done) {
+    this.After(function (scenario) {
 
         if (scenario.isFailed()) {
 
             // add a screenshot to the error report
-            driver.takeScreenshot().then(function (screenShot) {
+            return driver.takeScreenshot().then(function (screenShot) {
+
                 scenario.attach(new Buffer(screenShot, 'base64'), 'image/png');
-                driver.close();
-                driver.quit().then(done);
-            });
 
+                return driver.close().then(function(){
+                    return driver.quit();
+                });
+            });
         }
         else {
 
-            driver.close();
-            driver.quit().then(done);
+            return driver.close().then(function(){
+                return driver.quit();
+            });
         }
     });
 };