import { UniDriver } from '@wix/wix-ui-test-utils/unidriver';
import { BulkAssignTagsPopoverUniDriver } from '../Tags/BulkAssignTagsPopover/BulkAssignTagsPopover.uni.driver';
import { DisabledTooltipButtonUniDriver } from '../DisabledTooltip/DisabledTooltipButton.uni.driver';

export default (base: UniDriver, body: UniDriver) => {
  const button = DisabledTooltipButtonUniDriver(base, body, {
    dataHook: 'tags-assign-button',
  });

  return {
    ...button,
    popover: () => BulkAssignTagsPopoverUniDriver(base, body),
  };
};
