Ensure webdriver session is closed before exiting

niklasvh · Sep 9, 2014 · 645fcd6 · 645fcd6
1 parent 0325a9b
commit 645fcd6
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/selenium.js b/tests/selenium.js
@@ -172,11 +172,13 @@
             browserStream.push(result.browser);
         });
 
-        resultStream.onEnd(function() {
-            browser.quit();
+        return resultStream.fold([], pushToArray).flatMap(function(value) {
+            return Bacon.fromCallback(function(callback) {
+                browser.quit(function() {
+                    callback(value);
+                });
+            });
         });
-
-        return resultStream.fold([], pushToArray);
     }
 
     function createImages(data) {