wix-style-react/src/utils/DragAndDrop/dragAndDropTestUtils.js

wix-style-react

import ReactTestUtils from 'react-dom/test-utils';

const findInstance = (wrapper, cb) => {
  let itemInstance = null;
  ReactTestUtils.findAllInRenderedTree(wrapper, ins => {
    const portalRoot = ins && (ins.portal || ins.tooltipContent);
    if (portalRoot && ReactTestUtils.isCompositeComponent(portalRoot)) {
      return ReactTestUtils.findAllInRenderedTree(portalRoot, insInPortal => {
        if (cb(insInPortal)) {
          itemInstance = insInPortal;
        }
        return Boolean(itemInstance);
      });
    } else if (cb(ins)) {
      itemInstance = ins;
    }
    return Boolean(itemInstance);
  });
  return itemInstance;
};

export const getInstanceOfDraggableProvider = wrapper =>
  findInstance(wrapper, ins => ins?.props?.backend);

export const getInstanceOfDraggableSource = (wrapper, itemId) =>
  findInstance(
    wrapper,
    ins => ins?.handler?.monitor.sourceId && ins.props.id === itemId,
  );

export const getInstanceOfDraggableTarget = (wrapper, itemId) =>
  findInstance(
    wrapper,
    ins => ins?.handler?.monitor.targetId && ins.props.id === itemId,
  );