fix: TreeNode props should not miss by children node

Author: zombieJ

src/hooks/useTreeData.ts

@@ -63,14 +63,13 @@ function formatTreeData(
 
   function dig(dataNodes: DataNode[]) {
     return (dataNodes || []).map(node => {
-      const { key, disableCheckbox, disabled } = node;
+      const { key, children, ...restProps } = node;
 
       const value = node[fieldValue];
       const mergedValue = fieldValue in node ? value : key;
 
       const dataNode: InternalDataEntity = {
-        disableCheckbox,
-        disabled,
+        ...restProps,
         key: key !== null && key !== undefined ? key : mergedValue,
         value: mergedValue,
         title: getLabelProp(node),

tests/Select.tree.spec.js

@@ -83,7 +83,10 @@ describe('TreeSelect.tree', () => {
     const spy = jest.spyOn(console, 'error').mockImplementation(() => {});
     mount(
       <TreeSelect
-        treeData={[{ title: 'little', value: 'ttt' }, { title: 'bamboo', value: 'ttt' }]}
+        treeData={[
+          { title: 'little', value: 'ttt' },
+          { title: 'bamboo', value: 'ttt' },
+        ]}
       />,
     );
     expect(spy).toHaveBeenCalledWith('Warning: Same `value` exist in the tree: ttt');
@@ -125,4 +128,14 @@ describe('TreeSelect.tree', () => {
       });
     });
   });
+
+  it('Node icon', () => {
+    const wrapper = mount(
+      <TreeSelect open>
+        <SelectNode value="little" title="Little" icon={<span className="bamboo-light" />} />
+      </TreeSelect>,
+    );
+
+    expect(wrapper.exists('.bamboo-light')).toBeTruthy();
+  });
 });