[tests] Replace query selectors with getByTestId

mui · Sep 18, 2020 · dfa3d8c · dfa3d8c
1 parent f948fa2
commit dfa3d8c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 12 deletions.
diff --git a/packages/material-ui/src/Checkbox/Checkbox.test.js b/packages/material-ui/src/Checkbox/Checkbox.test.js
@@ -64,10 +64,8 @@ describe('<Checkbox />', () => {
 
   describe('prop: indeterminate', () => {
     it('should render an indeterminate icon', () => {
-      const { container } = render(<Checkbox indeterminate />);
-      expect(container.querySelector('svg[data-testid="IndeterminateCheckBoxIcon"]')).not.to.equal(
-        null,
-      );
+      const { getByTestId } = render(<Checkbox indeterminate />);
+      expect(getByTestId('IndeterminateCheckBoxIcon')).not.to.equal(null);
     });
   });
 

diff --git a/packages/material-ui/src/Chip/Chip.test.js b/packages/material-ui/src/Chip/Chip.test.js
@@ -532,9 +532,9 @@ describe('<Chip />', () => {
     });
 
     it('should render the delete icon with the deleteIcon and deleteIconSmall classes', () => {
-      const { container } = render(<Chip size="small" onDelete={() => {}} />);
+      const { getByTestId } = render(<Chip size="small" onDelete={() => {}} />);
 
-      const icon = container.querySelector('svg[data-testid="CancelIcon"]');
+      const icon = getByTestId('CancelIcon');
       expect(icon).to.have.class(classes.deleteIcon);
       expect(icon).to.have.class(classes.deleteIconSmall);
     });

diff --git a/packages/material-ui/src/Radio/Radio.test.js b/packages/material-ui/src/Radio/Radio.test.js
@@ -33,18 +33,14 @@ describe('<Radio />', () => {
   describe('prop: unchecked', () => {
     it('should render an unchecked icon', () => {
       const { getAllByTestId } = render(<Radio />);
-      expect(
-        getAllByTestId("RadioButtonUncheckedIcon").length
-      ).to.equal(1);
+      expect(getAllByTestId('RadioButtonUncheckedIcon').length).to.equal(1);
     });
   });
 
   describe('prop: checked', () => {
     it('should render a checked icon', () => {
       const { getAllByTestId } = render(<Radio checked />);
-      expect(
-        getAllByTestId("RadioButtonCheckedIcon").length
-      ).to.equal(1);
+      expect(getAllByTestId('RadioButtonCheckedIcon').length).to.equal(1);
     });
   });