import React from 'react'; import { render } from '../../../tests/utils'; import Tree from '../index'; import type { AntTreeNodeProps } from '../Tree'; const { TreeNode } = Tree; describe('Tree', () => { it('icon and switcherIcon of Tree with showLine should render correctly', () => { const { asFragment } = render( , ); expect(asFragment().firstChild).toMatchSnapshot(); }); it('switcherIcon in Tree should not render at leaf nodes', () => { const { container } = render( } defaultExpandAll> , ); expect(container.querySelectorAll('.switcherIcon').length).toBe(1); }); it('switcherIcon in Tree could be string', () => { const { asFragment } = render( , ); expect(asFragment().firstChild).toMatchSnapshot(); }); it('switcherIcon should be loading icon when loadData', () => { const onLoadData = () => new Promise(resolve => { setTimeout(() => { resolve(); }, 1000); }); const { asFragment } = render( , ); expect(asFragment().firstChild).toMatchSnapshot(); }); it('switcherIcon in Tree could be render prop function', () => { const { container } = render( expanded ? : } > , ); expect(container.querySelectorAll('.open').length).toBe(1); }); // https://github.com/ant-design/ant-design/issues/23261 it('showLine is object type should render correctly', () => { const { asFragment } = render( , ); expect(asFragment().firstChild).toMatchSnapshot(); }); describe('draggable', () => { const dragTreeData = [ { title: 'bamboo', key: 'bamboo', }, ]; it('hide icon', () => { const { container } = render(); expect(container.querySelector('.anticon-holder')).toBeFalsy(); }); it('customize icon', () => { const { container } = render( }} />, ); expect(container.querySelector('.little')).toBeTruthy(); }); it('nodeDraggable', () => { const nodeDraggable = jest.fn(() => false); render(); expect(nodeDraggable).toHaveBeenCalledWith(dragTreeData[0]); }); it('nodeDraggable func', () => { const nodeDraggable = jest.fn(() => false); render(); expect(nodeDraggable).toHaveBeenCalledWith(dragTreeData[0]); }); }); });