fixed tests

patorjk · Sep 21, 2019 · 58251c7 · 58251c7
1 parent f7953de
commit 58251c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/MUIDataTableBody.test.js b/test/MUIDataTableBody.test.js
@@ -560,7 +560,7 @@ describe('<TableBody />', function() {
     );
 
     mountWrapper
-      .find('TableSelectCell')
+      .find('Memo(TableSelectCell)')
       .first()
       .find('input')
       .simulate('click');
@@ -591,8 +591,9 @@ describe('<TableBody />', function() {
     );
 
     mountWrapper
-      .find('TableSelectCell')
+      .find('Memo(TableSelectCell)')
       .first()
+      .find('td')
       .simulate('click');
 
     assert.strictEqual(options.onRowClick.callCount, 0);