fix: defaultExpandAllRows should work with childrenColumnName (#424)

Author: zombieJ

.travis.yml

@@ -15,3 +15,4 @@ env:
   matrix:
     - TEST_TYPE=lint
     - TEST_TYPE=test
+    - TEST_TYPE=compile

src/Table.tsx

@@ -272,7 +272,7 @@ function Table<RecordType extends DefaultRecordType>(props: TableProps<RecordTyp
       return defaultExpandedRowKeys;
     }
     if (defaultExpandAllRows) {
-      return findAllChildrenKeys<RecordType>(mergedData, getRowKey);
+      return findAllChildrenKeys<RecordType>(mergedData, getRowKey, mergedChildrenColumnName);
     }
     return [];
   });

src/utils/expandUtil.tsx

@@ -34,14 +34,15 @@ export function renderExpandIcon<RecordType>({
 export function findAllChildrenKeys<RecordType>(
   data: RecordType[],
   getRowKey: GetRowKey<RecordType>,
+  childrenColumnName: string,
 ): Key[] {
   const keys: Key[] = [];
 
   function dig(list: RecordType[]) {
     (list || []).forEach((item, index) => {
       keys.push(getRowKey(item, index));
 
-      dig(((item as unknown) as { children: RecordType[] }).children);
+      dig((item as any)[childrenColumnName]);
     });
   }
 

tests/ExpandRow.spec.js

@@ -354,4 +354,18 @@ describe('Table.Expand', () => {
         .hasClass('rc-table-row-collapsed'),
     ).toBeTruthy();
   });
+
+  // https://github.com/ant-design/ant-design/issues/21788
+  it('`defaultExpandAllRows` with `childrenColumnName`', () => {
+    const data = [
+      {
+        key: 0,
+        sub: [{ key: 1, sub: [{ key: 2 }] }],
+      },
+    ];
+    const wrapper = mount(
+      createTable({ data, childrenColumnName: 'sub', expandable: { defaultExpandAllRows: true } }),
+    );
+    expect(wrapper.find('tbody tr')).toHaveLength(3);
+  });
 });