fix: Fix column children undefined (#448)

* column should be render when children is undefined

* column should be render when children is undefined

* add test case

* update test case

* Update Table.spec.js

Co-authored-by: 二货机器人 <[email protected]>

Author: yoyo837

src/Header/Header.tsx

@@ -35,12 +35,12 @@ function parseHeaderRows<RecordType>(
 
       let colSpan: number = 1;
 
-      if ((column as ColumnGroupType<RecordType>).children) {
-        colSpan = fillRowCells(
-          (column as ColumnGroupType<RecordType>).children,
-          currentColIndex,
-          rowIndex + 1,
-        ).reduce((total, count) => total + count, 0);
+      const subColumns = (column as ColumnGroupType<RecordType>).children;
+      if (subColumns && subColumns.length > 0) {
+        colSpan = fillRowCells(subColumns, currentColIndex, rowIndex + 1).reduce(
+          (total, count) => total + count,
+          0,
+        );
         cell.hasSubColumns = true;
       }
 

src/hooks/useColumns.tsx

@@ -40,10 +40,11 @@ function flatColumns<RecordType>(columns: ColumnsType<RecordType>): ColumnType<R
     // Convert `fixed='true'` to `fixed='left'` instead
     const parsedFixed = fixed === true ? 'left' : fixed;
 
-    if ((column as ColumnGroupType<RecordType>).children) {
+    const subColumns = (column as ColumnGroupType<RecordType>).children;
+    if (subColumns && subColumns.length > 0) {
       return [
         ...list,
-        ...flatColumns((column as ColumnGroupType<RecordType>).children).map(subColum => ({
+        ...flatColumns(subColumns).map(subColum => ({
           fixed: parsedFixed,
           ...subColum,
         })),

tests/Table.spec.js

@@ -38,6 +38,40 @@ describe('Table.Basic', () => {
       const wrapper = mount(createTable({ columns: [] }));
       expect(wrapper.render()).toMatchSnapshot();
     });
+
+    it('column children undefined', () => {
+      const wrapper = mount(
+        createTable({
+          columns: [
+            {
+              title: '姓名',
+              dataIndex: 'name',
+              key: 'name',
+              children: [],
+            },
+            {
+              title: '年龄',
+              dataIndex: 'age',
+              key: 'age',
+              children: undefined,
+            },
+          ],
+        }),
+      );
+      expect(wrapper.render()).toMatchSnapshot();
+      expect(
+        wrapper
+          .find('th')
+          .at(0)
+          .text(),
+      ).toEqual('姓名');
+      expect(
+        wrapper
+          .find('th')
+          .at(1)
+          .text(),
+      ).toEqual('年龄');
+    });
   });
 
   describe('renders empty text correctly', () => {

tests/__snapshots__/Table.spec.js.snap

@@ -482,6 +482,72 @@ exports[`Table.Basic renders correctly basic 1`] = `
 </div>
 `;
 
+exports[`Table.Basic renders correctly column children undefined 1`] = `
+<div
+  class="rc-table"
+>
+  <div
+    class="rc-table-container"
+  >
+    <div
+      class="rc-table-content"
+    >
+      <table
+        style="table-layout: auto;"
+      >
+        <colgroup />
+        <thead
+          class="rc-table-thead"
+        >
+          <tr>
+            <th
+              class="rc-table-cell"
+            >
+              姓名
+            </th>
+            <th
+              class="rc-table-cell"
+            >
+              年龄
+            </th>
+          </tr>
+        </thead>
+        <tbody
+          class="rc-table-tbody"
+        >
+          <tr
+            class="rc-table-row rc-table-row-level-0"
+            data-row-key="key0"
+          >
+            <td
+              class="rc-table-cell"
+            >
+              Lucy
+            </td>
+            <td
+              class="rc-table-cell"
+            />
+          </tr>
+          <tr
+            class="rc-table-row rc-table-row-level-0"
+            data-row-key="key1"
+          >
+            <td
+              class="rc-table-cell"
+            >
+              Jack
+            </td>
+            <td
+              class="rc-table-cell"
+            />
+          </tr>
+        </tbody>
+      </table>
+    </div>
+  </div>
+</div>
+`;
+
 exports[`Table.Basic renders correctly no columns 1`] = `
 <div
   class="rc-table"