wix-style-react/dist/es/src/BadgeSelect/BadgeSelect.protractor.driver.js

wix-style-react

import _asyncToGenerator from "@babel/runtime/helpers/asyncToGenerator";
import _regeneratorRuntime from "@babel/runtime/regenerator";
import popoverDriverFactory from '../Popover/Popover.protractor.driver';

var badgeSelectDriverFactory = function badgeSelectDriverFactory(component) {
  var popoverDriver = popoverDriverFactory(component);

  var getBadgeElement = function getBadgeElement() {
    return popoverDriver.getTargetElement();
  };

  var getDropdownItem = function getDropdownItem(index) {
    return component.$$("[data-hook=\"dropdown-layout-options\"] div").get(index);
  };

  return {
    /** Returns the element */
    element: function element() {
      return component;
    },

    /** Click the badge */
    clickBadge: function clickBadge() {
      return getBadgeElement().click();
    },

    /** Select a specific option */
    selectOption: function selectOption(index) {
      var _this = this;

      return _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
        return _regeneratorRuntime.wrap(function _callee$(_context) {
          while (1) {
            switch (_context.prev = _context.next) {
              case 0:
                _context.next = 2;
                return _this.clickBadge();

              case 2:
                _context.next = 4;
                return getDropdownItem(index).click();

              case 4:
              case "end":
                return _context.stop();
            }
          }
        }, _callee);
      }))();
    }
  };
};

export default badgeSelectDriverFactory;