Fix tests

mui · Mar 23, 2019 · 8de76e8 · 8de76e8
1 parent 7a95839
commit 8de76e8
Showing 1 changed file with 16 additions and 43 deletions.
diff --git a/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.test.js b/packages/material-ui-lab/src/ToggleButtonGroup/ToggleButtonGroup.test.js
@@ -18,8 +18,11 @@ describe('<ToggleButtonGroup />', () => {
     );
   });
 
-  function findButton(wrapper, value) {
-    return wrapper.find(`ToggleButton[value="${value}"]`).first();
+  function findToggleButton(wrapper, value) {
+    return wrapper
+      .find(ToggleButton)
+      .find(`[value="${value}"]`)
+      .first();
   }
 
   it('should render a <div> element', () => {
@@ -50,7 +53,7 @@ describe('<ToggleButtonGroup />', () => {
         </ToggleButtonGroup>,
       );
 
-      assert.strictEqual(findButton(wrapper, 'one').props().selected, true);
+      assert.strictEqual(findToggleButton(wrapper, 'one').props().selected, true);
     });
 
     it('should not render a selected ToggleButton when its value is not selected', () => {
@@ -61,7 +64,7 @@ describe('<ToggleButtonGroup />', () => {
         </ToggleButtonGroup>,
       );
 
-      assert.strictEqual(findButton(wrapper, 'two').props().selected, false);
+      assert.strictEqual(findToggleButton(wrapper, 'two').props().selected, false);
     });
   });
 
@@ -74,20 +77,8 @@ describe('<ToggleButtonGroup />', () => {
         </ToggleButtonGroup>,
       );
 
-      assert.strictEqual(
-        wrapper
-          .find('[value="one"]')
-          .first()
-          .props().selected,
-        true,
-      );
-      assert.strictEqual(
-        wrapper
-          .find('[value="two"]')
-          .first()
-          .props().selected,
-        false,
-      );
+      assert.strictEqual(findToggleButton(wrapper, 'one').props().selected, true);
+      assert.strictEqual(findToggleButton(wrapper, 'two').props().selected, false);
     });
   });
 
@@ -102,10 +93,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(0)
-          .simulate('click');
+        findToggleButton(wrapper, 'one').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.strictEqual(handleChange.args[0][1], null);
@@ -120,10 +108,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(0)
-          .simulate('click');
+        findToggleButton(wrapper, 'one').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.strictEqual(handleChange.args[0][1], 'one');
@@ -138,10 +123,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(1)
-          .simulate('click');
+        findToggleButton(wrapper, 'two').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.strictEqual(handleChange.args[0][1], 'two');
@@ -158,10 +140,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(0)
-          .simulate('click');
+        findToggleButton(wrapper, 'one').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.ok(Array.isArray(handleChange.args[0][1]));
@@ -177,10 +156,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(0)
-          .simulate('click');
+        findToggleButton(wrapper, 'one').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.deepEqual(handleChange.args[0][1], ['one']);
@@ -195,10 +171,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        wrapper
-          .find(ToggleButton)
-          .at(0)
-          .simulate('click');
+        findToggleButton(wrapper, 'one').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.deepEqual(handleChange.args[0][1], ['two']);
@@ -213,7 +186,7 @@ describe('<ToggleButtonGroup />', () => {
           </ToggleButtonGroup>,
         );
 
-        findButton(wrapper, 'two').simulate('click');
+        findToggleButton(wrapper, 'two').simulate('click');
 
         assert.strictEqual(handleChange.callCount, 1);
         assert.deepEqual(handleChange.args[0][1], ['one', 'two']);