nightwatch/lib/element/locate/elements-by-recursion.js

Easy to use Node.js based end-to-end testing solution for web applications using the W3C WebDriver API.

const RecursiveLookupBase = require('./recursive-lookup.js');

/**
 * Search for multiple elements on the page, starting with the first element of the array,
 *  and where each element in the passed array is nested under the previous one.
 * The located element will be returned as a WebElement JSON object.
 *
 * @param {Array} elements An array of ancestor element objects containing selector and locateStrategy properties
 * @param {function} [callback] Optional callback function to be called when the command finishes.
 * @api protocol
 */

class MultipleElementsByRecursion extends RecursiveLookupBase {
  get transportAction() {
    return 'locateMultipleElementsByElementId';
  }

  get commandName() {
    return 'elementIdElements';
  }

  locateElements({element, returnSingleElement}) {
    this.allElements = Array.isArray(element.selector) ? element.selector.slice() : [
      element
    ];
    const nextElement = this.getNextElement();

    this.elementLocator
      .findElement({element: nextElement, cacheElementId: false, returnSingleElement: this.shouldReturnSingleElement(returnSingleElement)})
      .then(result => {
        if (result.value === null) {
          this.deferred.resolve(result);
        } else {
          this.recursiveElementLookup({result, returnSingleElement});
        }
      });

    return this.deferred.promise;
  }
}


module.exports = MultipleElementsByRecursion;