From 729133f428e74824419912bfcd09ea6fd11b238f Mon Sep 17 00:00:00 2001
From: Cagatay Civici <cagatay.civici@gmail.com>
Date: Sat, 15 May 2021 13:06:54 +0300
Subject: [PATCH] Fixed #1260 - TreeTable does not add column and style props

---
 src/components/treetable/BodyCell.vue   | 2 +-
 src/components/treetable/FooterCell.vue | 2 +-
 src/components/treetable/HeaderCell.vue | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/components/treetable/BodyCell.vue b/src/components/treetable/BodyCell.vue
index ad276c6e0..2993a8553 100644
--- a/src/components/treetable/BodyCell.vue
+++ b/src/components/treetable/BodyCell.vue
@@ -121,7 +121,7 @@ export default {
     },
     computed: {
         containerClass() {
-            return [this.columnProp('bodyClass'), {
+            return [this.columnProp('bodyClass'), this.columnProp('class'), {
                 'p-frozen-column': this.columnProp('frozen')
             }];
         },
diff --git a/src/components/treetable/FooterCell.vue b/src/components/treetable/FooterCell.vue
index e9477aa6e..84b31c01e 100644
--- a/src/components/treetable/FooterCell.vue
+++ b/src/components/treetable/FooterCell.vue
@@ -59,7 +59,7 @@ export default {
     },
     computed: {
         containerClass() {
-            return [this.columnProp('footerClass'), {
+            return [this.columnProp('footerClass'), this.columnProp('class'), {
                 'p-frozen-column': this.columnProp('frozen')
             }];
         },
diff --git a/src/components/treetable/HeaderCell.vue b/src/components/treetable/HeaderCell.vue
index 4a7cc1a45..e8e78d30d 100644
--- a/src/components/treetable/HeaderCell.vue
+++ b/src/components/treetable/HeaderCell.vue
@@ -121,7 +121,7 @@ export default {
     },
     computed: {
         containerClass() {
-            return [this.columnProp('headerClass'), {
+            return [this.columnProp('headerClass'), this.columnProp('class'), {
                 'p-sortable-column': this.columnProp('sortable'),
                 'p-resizable-column': this.resizableColumns,
                 'p-highlight': this.isColumnSorted(),