angular/modules/benchpress/src/webdriver/selenium_webdriver_adapter.es6

import { Promise, PromiseWrapper } from 'angular2/src/facade/async';
import { bind } from 'angular2/di';
import { WebDriverAdapter } from '../web_driver_adapter';

import webdriver from 'selenium-webdriver';

/**
 * Adapter for the selenium-webdriver.
 */
export class SeleniumWebDriverAdapter extends WebDriverAdapter {
  _driver:any;

  constructor(driver) {
    super();
    this._driver = driver;
  }

  _convertPromise(thenable) {
    var completer = PromiseWrapper.completer();
    thenable.then(completer.complete, completer.reject);
    return completer.promise;
  }

  waitFor(callback):Promise {
    return this._convertPromise(this._driver.controlFlow().execute(callback));
  }

  executeScript(script:string):Promise {
    return this._convertPromise(this._driver.executeScript(script));
  }

  capabilities():Promise {
    return this._convertPromise(this._driver.getCapabilities());
  }

  logs(type:string):Promise {
    // Needed as selenium-webdriver does not forward
    // performance logs in the correct way via manage().logs
    return this._convertPromise(this._driver.schedule(
      new webdriver.Command(webdriver.CommandName.GET_LOG).
          setParameter('type', type),
      'WebDriver.manage().logs().get(' + type + ')').then( (logs) => {
        // Need to convert the Array into an instance of an Array
        // as selenium-webdriver uses an own Node.js context!
        return [].slice.call(logs);
      }));
  }

}