From 70bf038ec366b64436ff3f5613c907ac19ed1656 Mon Sep 17 00:00:00 2001
From: Supreeth <supreeth.mr1990@gmail.com>
Date: Tue, 17 Mar 2020 00:46:42 +0530
Subject: [PATCH] Fixed refCallBack not working for table rows (#2901)

Co-authored-by: Emily Rohrbough <emilyrohrbough@users.noreply.github.com>
---
 packages/terra-table/CHANGELOG.md             |  2 ++
 packages/terra-table/src/Table.jsx            |  1 +
 .../terra-table/tests/jest/Table.test.jsx     | 25 +++++++++++++
 .../jest/__snapshots__/Table.test.jsx.snap    | 36 +++++++++++++++++++
 4 files changed, 64 insertions(+)

diff --git a/packages/terra-table/CHANGELOG.md b/packages/terra-table/CHANGELOG.md
index cbb39ba80cb..4731b8e1196 100644
--- a/packages/terra-table/CHANGELOG.md
+++ b/packages/terra-table/CHANGELOG.md
@@ -3,6 +3,8 @@ ChangeLog
 
 Unreleased
 ----------
+### Fixed
+* Fixed refCallBack not working for table rows.
 
 4.4.0 - (March 10, 2020)
 ------------------
diff --git a/packages/terra-table/src/Table.jsx b/packages/terra-table/src/Table.jsx
index 7b0db85b4d7..e8eb188d4c0 100644
--- a/packages/terra-table/src/Table.jsx
+++ b/packages/terra-table/src/Table.jsx
@@ -263,6 +263,7 @@ const createRow = (tableData, rowData, rowIndex, sectionId) => {
       isDisabled={rowData.isDisabled}
       isStriped={rowData.isStriped}
       dividerStyle={tableData.dividerStyle}
+      refCallback={rowData.refCallback}
     >
       {createCheckCell(rowData, tableData.rowStyle, tableData.checkStyle)}
       {rowData.cells.map((cell, colIndex) => {
diff --git a/packages/terra-table/tests/jest/Table.test.jsx b/packages/terra-table/tests/jest/Table.test.jsx
index 6030a665745..bfc631225f0 100644
--- a/packages/terra-table/tests/jest/Table.test.jsx
+++ b/packages/terra-table/tests/jest/Table.test.jsx
@@ -59,6 +59,31 @@ describe('Table', () => {
     expect(shallowComponent).toMatchSnapshot();
   });
 
+  it('should render a Table with row and refCallback', () => {
+    const refCallback = jest.fn();
+    const shallowComponent = shallow(
+      <Table
+        summary="Test summary description"
+        summaryId="test-id"
+        numberOfColumns={0}
+        fill
+        scrollRefCallback={jest.fn()}
+        bodyData={[{
+          rows: [{
+            key: 'row-key',
+            refCallback,
+            cells: [{
+              key: 'child-cell-key',
+              id: 'child-cell-id',
+              children: 'content',
+            }],
+          }],
+        }]}
+      />,
+    );
+    expect(shallowComponent).toMatchSnapshot();
+  });
+
   it('should render a Table with padding standard', () => {
     const shallowComponent = shallow(
       <Table
diff --git a/packages/terra-table/tests/jest/__snapshots__/Table.test.jsx.snap b/packages/terra-table/tests/jest/__snapshots__/Table.test.jsx.snap
index 5a8b0789d58..81075c16219 100644
--- a/packages/terra-table/tests/jest/__snapshots__/Table.test.jsx.snap
+++ b/packages/terra-table/tests/jest/__snapshots__/Table.test.jsx.snap
@@ -524,6 +524,42 @@ exports[`Table should render a Table with padding standard 1`] = `
 </div>
 `;
 
+exports[`Table should render a Table with row and refCallback 1`] = `
+<div
+  aria-describedby="test-id"
+  aria-rowcount={1}
+  className="table fill outer"
+  role="grid"
+>
+  <VisuallyHiddenText
+    id="test-id"
+    text="Test summary description"
+  />
+  <div
+    className="body"
+    role="rowgroup"
+    tabIndex={0}
+  >
+    <Row
+      aria-rowindex={1}
+      isDisabled={false}
+      isSelectable={false}
+      isSelected={false}
+      isStriped={false}
+      key="row-key"
+      refCallback={[MockFunction]}
+    >
+      <Cell
+        key="child-cell-key"
+        removeInner={false}
+      >
+        content
+      </Cell>
+    </Row>
+  </div>
+</div>
+`;
+
 exports[`Table should render a Table with row style toggle 1`] = `
 <div
   aria-describedby="test-id"