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

wix-style-react

import { badgeDriverFactory } from 'wix-ui-backoffice/dist/src/components/Badge/Badge.driver';

import dropdownLayoutDriverFactory from '../DropdownLayout/DropdownLayout.driver';
import popoverDriverFactory from '../Popover/Popover.driver';

var badgeSelectDriverFactory = function badgeSelectDriverFactory(_ref) {
  var element = _ref.element,
      eventTrigger = _ref.eventTrigger;

  var popoverDriver = popoverDriverFactory({ element: element, eventTrigger: eventTrigger });

  var badgeDriver = badgeDriverFactory({
    element: popoverDriver.getTargetElement().childNodes[0],
    eventTrigger: eventTrigger
  });

  var driver = {
    /** Returns 'true' wether the element exists */
    exists: function exists() {
      return !!element;
    },

    /** Click on an option */
    clickAtOption: function clickAtOption(index) {
      if (!popoverDriver.isContentElementExists()) {
        badgeDriver.click();
      }

      // Since the popover may be closed, we need to create the DropdownLayout driver every time
      var dropdownLayoutDriver = dropdownLayoutDriverFactory({
        element: element.querySelector('[data-hook=badgeSelect-dropdownLayout]')
      });

      if (dropdownLayoutDriver.exists()) {
        dropdownLayoutDriver.clickAtOption(index);
      }
    }
  };

  return driver;
};

export default badgeSelectDriverFactory;