wix-style-react/src/Swatches/Swatches.uni.driver.js

wix-style-react

import { fillPreviewDriverFactory } from '../FillPreview/FillPreview.uni.driver';
import { dataHooks } from './constants';

export const swatchesDriverFactory = base => {
  const getSwatch = async index =>
    base.$$('[data-hook="color-swatches-swatch"]').get(index);

  const getEmptySwatch = async () => base.$(`[data-hook="${dataHooks.empty}"]`);

  return {
    exists: async () => base.exists(),

    /** Get swatch count */
    getSwatchCount: async () =>
      base.$$('[data-hook="color-swatches-swatch"]').count(),

    /** Get swatch at given index */
    getSwatch: async index => fillPreviewDriverFactory(await getSwatch(index)),

    /** Click empty swatch */
    clickEmptySwatch: async () =>
      fillPreviewDriverFactory(await getEmptySwatch()).click(),

    /** Test if swatch is selected at given index */
    isSwatchSelectedAt: async index => {
      return fillPreviewDriverFactory(await getSwatch(index)).isSelected();
    },

    /** Test if empty swatch is selected */
    isEmptySwatchSelected: async () => {
      return fillPreviewDriverFactory(await getEmptySwatch()).isSelected();
    },

    addButtonExists: async () =>
      base.$$('[data-hook="color-preview-add-button"]').count() === 1,
  };
};