update render for table-cell test

mui · Jan 30, 2021 · 7f509aa · 7f509aa
1 parent e89f222
commit 7f509aa
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/material-ui/src/TableCell/TableCell.test.js b/packages/material-ui/src/TableCell/TableCell.test.js
@@ -31,8 +31,14 @@ describe('<TableCell />', () => {
       return wrapper.find('tr').childAt(0);
     },
     render: (node) => {
-      const { container, ...rest } = render(<table>{node}</table>);
-      return { container: container.firstChild, ...rest };
+      const { container, ...rest } = render(
+        <table>
+          <tbody>
+            <tr>{node}</tr>
+          </tbody>
+        </table>,
+      );
+      return { container: container.firstChild.firstChild.firstChild, ...rest };
     },
     muiName: 'MuiTableCell',
     testVariantProps: { variant: 'body' },