chore: Add columnType to confirm is expand column

Author: zombieJ

src/ColGroup.tsx

@@ -21,7 +21,8 @@ function ColGroup<RecordType>({ colWidths, columns, columCount }: ColGroupProps<
     const additionalProps = column && column[INTERNAL_COL_DEFINE];
 
     if (width || additionalProps || mustInsert) {
-      cols.unshift(<col key={i} style={{ width }} {...additionalProps} />);
+      const { columnType, ...restAdditionalProps } = additionalProps || {};
+      cols.unshift(<col key={i} style={{ width }} {...restAdditionalProps} />);
       mustInsert = true;
     }
   }

src/hooks/useColumns.tsx

@@ -194,6 +194,7 @@ function useColumns<RecordType>(
       const expandColumn = {
         [INTERNAL_COL_DEFINE]: {
           className: `${prefixCls}-expand-icon-col`,
+          columnType: 'EXPAND_COLUMN',
         },
         title: '',
         fixed: fixedColumn,

tests/Table.spec.js

@@ -1,7 +1,7 @@
 import React from 'react';
 import { mount } from 'enzyme';
 import { resetWarned } from 'rc-util/lib/warning';
-import Table from '../src';
+import Table, { INTERNAL_COL_DEFINE } from '../src';
 import { INTERNAL_HOOKS } from '../src/Table';
 
 describe('Table.Basic', () => {
@@ -750,15 +750,41 @@ describe('Table.Basic', () => {
   });
 
   describe('internal api', () => {
-    it('transformColumns', () => {
-      const wrapper = mount(
-        createTable({
-          internalHooks: INTERNAL_HOOKS,
-          transformColumns: columns => [{ title: 'before' }, ...columns, { title: 'after' }],
-        }),
-      );
+    describe('transformColumns', () => {
+      it('basic', () => {
+        const wrapper = mount(
+          createTable({
+            internalHooks: INTERNAL_HOOKS,
+            transformColumns: columns => [{ title: 'before' }, ...columns, { title: 'after' }],
+          }),
+        );
 
-      expect(wrapper.render()).toMatchSnapshot();
+        expect(wrapper.render()).toMatchSnapshot();
+      });
+
+      // Used for antd to check if is expand column
+      // We'd better to move selection into rc-table also
+      it('internal columnType', () => {
+        let existExpandColumn = false;
+
+        mount(
+          createTable({
+            expandable: {
+              expandedRowRender: () => null,
+            },
+            internalHooks: INTERNAL_HOOKS,
+            transformColumns: columns => {
+              console.log(columns);
+              existExpandColumn = columns.some(
+                col => col[INTERNAL_COL_DEFINE]?.columnType === 'EXPAND_COLUMN',
+              );
+              return columns;
+            },
+          }),
+        );
+
+        expect(existExpandColumn).toBeTruthy();
+      });
     });
 
     it('internalRefs', () => {

tests/__snapshots__/Table.spec.js.snap

@@ -244,7 +244,7 @@ exports[`Table.Basic custom components scroll content with scroll 1`] = `
 </div>
 `;
 
-exports[`Table.Basic internal api transformColumns 1`] = `
+exports[`Table.Basic internal api transformColumns basic 1`] = `
 <div
   class="rc-table"
 >