@dgit/react-data-grid-addons/src/draggable/__tests__/DragDropContainer.spec.js

A set of addons for react-data-grid

import React from 'react';
import { shallow } from 'enzyme';
import DragDropContainer from '../DragDropContainer';
import jasmineEnzyme from 'jasmine-enzyme';
import RowDragLayer from '../RowDragLayer';
import DraggableHeaderCell from '../DraggableHeaderCell';

describe('<DragDropContainer />', () => {
  let wrapper;

  const childProps = {
    rowsCount: 2,
    rowGetter: index => index
  };

  const GridStub = () => <div/>;

  function render(props = {}) {
    let ComponentUnderTest = DragDropContainer.DecoratedComponent;
    wrapper = shallow(
      <ComponentUnderTest {...props} >
        <GridStub {...childProps} />
      </ComponentUnderTest>
    );
  }

  beforeEach(() => {
    jasmineEnzyme();
    render();
  });

  it('should render a RowDragLayer', () => {
    let rowDragLayer = wrapper.find(RowDragLayer);
    expect(rowDragLayer).toBePresent();
    expect(rowDragLayer).toHaveProp('rows', [0, 1]);
  });

  it('should correctly render child grid component when passed in', () => {
    let gridStub = wrapper.find(GridStub);
    expect(gridStub).toBePresent();
    expect(gridStub).toHaveProp('draggableHeaderCell', DraggableHeaderCell);
  });

  it('getDragPreviewRow should override rowGetter to provide rows to rowDragLayer', () => {
    const props = {
      getDragPreviewRow: index => index + 'a'
    };
    render(props);
    let rowDragLayer = wrapper.find(RowDragLayer);
    expect(rowDragLayer).toBePresent();
    expect(rowDragLayer).toHaveProp('rows', ['0a', '1a']);
  });
});