fix ant-design/ant-design#11492. Pos isParent is meanless in the tree drag check

Author: zombieJ

package.json

@@ -61,7 +61,8 @@
     ],
     "transform": {
       "\\.jsx?$": "./node_modules/rc-tools/scripts/jestPreprocessor.js"
-    }
+    },
+    "testURL": "http://localhost"
   },
   "devDependencies": {
     "cross-env": "^5.2.0",

src/util.js

@@ -123,10 +123,8 @@ export function getDragNodesKeys(treeNodes, node) {
   const { eventKey, pos } = node.props;
   const dragNodesKeys = [];
 
-  traverseTreeNodes(treeNodes, ({ pos: nodePos, key }) => {
-    if (isParent(pos, nodePos)) {
-      dragNodesKeys.push(key);
-    }
+  traverseTreeNodes(treeNodes, ({ key }) => {
+    dragNodesKeys.push(key);
   });
   dragNodesKeys.push(eventKey || pos);
   return dragNodesKeys;

tests/util.spec.js

@@ -1,10 +1,12 @@
 /* eslint-disable no-undef, react/no-multi-comp, react/no-unused-state, react/prop-types, no-return-assign */
 import React from 'react';
+import { mount } from 'enzyme';
 import Tree from '../src/Tree';
 import TreeNode from '../src/TreeNode';
 import {
   convertDataToTree, convertTreeToEntities,
   conductCheck, conductExpandParent,
+  getDragNodesKeys,
 } from '../src/util';
 import { convertTreeToData } from './util';
 
@@ -193,4 +195,25 @@ describe('Util', () => {
     const keys = conductExpandParent(['good'], keyEntities);
     expect(keys.sort()).toEqual(['bamboo', 'is', 'good'].sort());
   });
+
+  it('getDragNodesKeys', () => {
+    const tree = mount(
+      <Tree defaultExpandAll>
+        <TreeNode key="000">
+          <TreeNode key="111">
+            <TreeNode key="222" />
+            <TreeNode key="333" />
+          </TreeNode>
+        </TreeNode>
+      </Tree>
+    );
+
+    const treeNode0 = tree.find(TreeNode).at(0).instance();
+    const keys0 = getDragNodesKeys(treeNode0.props.children, treeNode0);
+    expect(keys0.sort()).toEqual(['000', '111', '222', '333'].sort());
+
+    const treeNode1 = tree.find(TreeNode).at(1).instance();
+    const keys1 = getDragNodesKeys(treeNode1.props.children, treeNode1);
+    expect(keys1.sort()).toEqual(['111', '222', '333'].sort());
+  });
 });