feat: Table expanded icon pass internal status (#413)

* feat: Pass parent status for render icon

* add test case

Author: zombieJ

src/Table.tsx

@@ -246,9 +246,24 @@ function Table<RecordType extends DefaultRecordType>(props: TableProps<RecordTyp
     if (expandedRowRender) {
       return 'row';
     }
-    if (mergedData.some(record => mergedChildrenColumnName in record)) {
+    /* eslint-disable no-underscore-dangle */
+    /**
+     * Fix https://github.com/ant-design/ant-design/issues/21154
+     * This is a workaround to not to break current behavior.
+     * We can remove follow code after final release.
+     *
+     * To other developer:
+     *  Do not use `__PARENT_RENDER_ICON__` in prod since we will remove this when refactor
+     */
+    if (
+      (props.expandable &&
+        internalHooks === INTERNAL_HOOKS &&
+        (props.expandable as any).__PARENT_RENDER_ICON__) ||
+      mergedData.some(record => mergedChildrenColumnName in record)
+    ) {
       return 'nest';
     }
+    /* eslint-enable */
     return false;
   }, [!!expandedRowRender, mergedData]);
 

tests/Internal.spec.js

@@ -0,0 +1,23 @@
+import React from 'react';
+import { mount } from 'enzyme';
+import Table from '../src';
+import { INTERNAL_HOOKS } from '../src/Table';
+
+// All follow test is only for internal usage which should be removed when refactor
+describe('Table.Internal', () => {
+  it('internal should pass `__PARENT_RENDER_ICON__` for expandable', () => {
+    const wrapper = mount(
+      <Table
+        internalHooks={INTERNAL_HOOKS}
+        columns={[{ dataIndex: 'key' }]}
+        data={[{ key: 233 }]}
+        expandable={{
+          __PARENT_RENDER_ICON__: true,
+          expandIcon: () => <div className="expand-icon" />,
+        }}
+      />,
+    );
+
+    expect(wrapper.find('.expand-icon')).toHaveLength(1);
+  });
+});