wix-style-react/src/TableActionCell/test/TableActionCell.private.uni.driver.js

wix-style-react

import { dataHooks } from '../constants';
import { findByHook } from '../../../test/utils/unidriver';
import { tableActionCellUniDriverFactory as publicDriverFactory } from '../TableActionCell.uni.driver';
import { iconButtonDriverFactory } from '../../IconButton/IconButton.uni.driver';

export const tableActionCellPrivateUniDriverFactory = (base, body) => {
  const primaryActionPlaceholderDriver = iconButtonDriverFactory(
    findByHook(base, dataHooks.placeholder),
  );

  return {
    ...publicDriverFactory(base, body),

    /** Whether the primary action placeholder exists */
    primaryActionPlaceholderExists: primaryActionPlaceholderDriver.exists,

    clickSecondaryActions: () =>
      findByHook(base, dataHooks.triggerElement).click(),
  };
};