ant-design/components/table/__tests__/Table.expand.test.js

/* eslint-disable react/no-multi-comp */
import React from 'react';
import { mount } from 'enzyme';
import Table from '..';

const columns = [
  {
    title: 'Name',
    key: 'name',
    dataIndex: 'name',
  },
];

const data = [
  {
    key: '1',
    firstName: 'John',
    lastName: 'Brown',
    age: 32,

    children: [
      {
        key: '2',
        firstName: 'Jim',
        lastName: 'Green',
        age: 42,
      },
    ],
  },
];

describe('Table.expand', () => {
  it('click to expand', () => {
    const wrapper = mount(<Table columns={columns} dataSource={data} />);
    wrapper
      .find('.ant-table-row-expand-icon')
      .last()
      .simulate('click');
    expect(wrapper.render()).toMatchSnapshot();
  });

  it('should support expandIconColumnIndex', () => {
    const wrapper = mount(<Table columns={[]} dataSource={data} expandIconColumnIndex={1} />);
    expect(wrapper.render()).toMatchSnapshot();
  });
});