feat(async/await): Support overwriting native promise with WebDriver'…

…s version

See angular#61

.gitignore

@@ -1,3 +1,3 @@
 *.log
 node_modules
-spec/asyncAwaitSpec.js
+spec/asyncAwaitSpec*.js

index.js

@@ -144,13 +144,21 @@ function wrapInControlFlow(flow, globalFn, fnName) {
  * pass webdriver here instead of using require() in order to ensure Protractor
  * and Jasminews are using the same webdriver instance.
  * @param {Object} flow. The ControlFlow to wrap tests in.
+ * @param {function=} WDPromise The class for promises managed by the
+ *     ControlFlow.  This is an optional parameter.  If you use it, jasminewd
+ *     will overwrite the global `Promise` class with this so that tools like
+ *     async/await work.  This might cause other unexpected problems however.
  */
-function initJasmineWd(flow) {
+function initJasmineWd(flow, WDPromise) {
   if (jasmine.JasmineWdInitialized) {
     throw Error('JasmineWd already initialized when init() was called');
   }
   jasmine.JasmineWdInitialized = true;
 
+  if (WDPromise) {
+    WDPromise.prototype = global.Promise || WDPromise.prototype;
+    global.Promise = WDPromise;
+  }
   global.it = wrapInControlFlow(flow, global.it, 'it');
   global.fit = wrapInControlFlow(flow, global.fit, 'fit');
   global.beforeEach = wrapInControlFlow(flow, global.beforeEach, 'beforeEach');

package.json

@@ -23,7 +23,7 @@
   },
   "main": "index.js",
   "scripts": {
-    "pretest": "node_modules/.bin/jshint index.js spec --exclude spec/asyncAwaitSpec.js; tsc -t ES2015 spec/asyncAwaitSpec.ts",
+    "pretest": "node_modules/.bin/jshint index.js spec --exclude spec/asyncAwaitSpec*.js; tsc -t ES2015 spec/asyncAwaitSpec*.ts",
     "test": "scripts/test.sh"
   },
   "license": "MIT",

scripts/test.sh

@@ -1,5 +1,6 @@
 PASSING_SPECS="spec/support/passing_specs.json"
 FAILING_SPECS="spec/support/failing_specs.json"
+EXPERIMENTAL_SPECS="spec/support/experimental_specs.json"
 CMD_BASE="node node_modules/.bin/jasmine JASMINE_CONFIG_PATH="
 
 echo "### running passing specs"
@@ -18,4 +19,12 @@ results_line=`echo "$res" | tail -2 | head -1`
 echo "result: $results_line"
 [ "$results_line" = "$EXPECTED_RESULTS" ] || exit 1
 
+echo "### running experimental specs"
+CMD=$CMD_BASE$EXPERIMENTAL_SPECS
+echo "### $CMD"
+$CMD
+# We don't care if the experimental specs pass.
+# [ "$?" -eq 0 ] || exit 1
+echo
+
 echo "all pass"

spec/asyncAwaitSpec_experimental.ts

@@ -0,0 +1,28 @@
+/* This is a test for an experimental feature to make async/await work with the
+ * webdriver control flow.  You should not use this. 
+ */
+"use strict";
+
+// Declare globals
+declare var describe;
+declare var it;
+declare var expect;
+declare var require;
+
+let webdriver = require('selenium-webdriver'),
+    flow = webdriver.promise.controlFlow();
+require('../index.js').init(flow, webdriver.promise.Promise);
+
+describe('async function', function() {
+  let sharedVal: any;
+  it('should wait on async functions', async function() {
+    sharedVal = await webdriver.promise.fulfilled('a');
+    expect(sharedVal).toBe('a');
+    flow.timeout(1000); // The control flow needs to work for this.
+    sharedVal = await webdriver.promise.fulfilled('b');
+  });
+
+  it('should have waited until the end of the last it() block', function() {
+    expect(sharedVal).toBe('b');
+  });
+});

spec/support/experimental_specs.json

@@ -0,0 +1,6 @@
+{
+  "spec_dir": "spec",
+  "spec_files": [
+    "asyncAwaitSpec_experimental.js"
+  ]
+}