Merge pull request #1056 from 18F/ab/functional-web-element-fix

Allow BaseElement to accept selector or WebElement
cloud-gov · Apr 17, 2017 · 2f1e20e · 2f1e20e
2 parents b7f1589 + 4f89f06
commit 2f1e20e
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/static_src/test/functional/pageobjects/base.element.js b/static_src/test/functional/pageobjects/base.element.js
@@ -10,9 +10,18 @@
  * to deal with a single component.
  **/
 
+import assert from 'assert';
+
 export default class BaseElement {
-  constructor(browser, webElement) {
+  constructor(browser, webElementOrSelector) {
     this.browser = browser;
+
+    let webElement = webElementOrSelector;
+    if (typeof webElementOrSelector === 'string') {
+      webElement = browser.element(webElementOrSelector);
+    }
+
+    assert(webElement.value, `Element '${webElement.selector}' does not exist in the DOM.`);
     this.webElementId = webElement.value.ELEMENT;
   }