fix(table): table.columns.thClassName doesn't work (#3198) (#3238)

* fix(table): table.columns.thClassName doesn't work (#3198)

fix #3198

* test(table): add missing test: thClassName

src/table/__tests__/column.test.jsx

@@ -74,5 +74,27 @@ TABLES.forEach((TTable) => {
       expect(tdList.at(3).classes('tdesign-class')).toBeTruthy();
       expect(tdList.at(3).classes('tdesign-class1')).toBeFalsy();
     });
+
+    // 校验逻辑与上面columns.className一致
+    it('Props.columns.thClassName works fine', () => {
+      const columns = [
+        { title: 'Index', colKey: 'index', thClassName: () => ['th-class'] },
+        { title: 'Instance', colKey: 'instance', thClassName: 'th-class' },
+        { title: 'description', colKey: 'description', thClassName: [{ 'th-class': true }] },
+        { title: 'Owner', colKey: 'owner', thClassName: { 'th-class': true, 'th-class1': false } },
+      ];
+      const wrapper = mount({
+        render() {
+          return <TTable rowKey="index" data={data} columns={columns}></TTable>;
+        },
+      });
+      const thWrapper = wrapper.find('thead > tr');
+      const thList = thWrapper.findAll('th');
+      expect(thList.at(0).classes('th-class')).toBeTruthy();
+      expect(thList.at(1).classes('th-class')).toBeTruthy();
+      expect(thList.at(2).classes('th-class')).toBeTruthy();
+      expect(thList.at(3).classes('th-class')).toBeTruthy();
+      expect(thList.at(3).classes('th-class1')).toBeFalsy();
+    });
   });
 });

src/table/thead.tsx

@@ -139,12 +139,14 @@ export default defineComponent({
             rowIndex: -1,
           };
           const customClasses = formatClassNames(col.className, { ...colParams, type: 'th' });
+          const thCustomClasses = formatClassNames(col.thClassName, { ...colParams, type: 'th' });
           const isLeftFixedActive = this.showColumnShadow.left && col.fixed === 'left';
           const isRightFixedActive = this.showColumnShadow.right && col.fixed === 'right';
           const canDragSort = this.thDraggable && !(isLeftFixedActive || isRightFixedActive);
           const thClasses = [
             thStyles.classes,
             customClasses,
+            thCustomClasses,
             {
               // 受 rowspan 影响，部分 tr > th:first-child 需要补足左边框
               [this.tableHeaderClasses.thBordered]: thBorderMap.get(col),