test: fix test in segmented (#35183)

ant-design · Apr 22, 2022 · e2cddd6 · e2cddd6
1 parent eb70223
commit e2cddd6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/components/segmented/__tests__/index.test.tsx b/components/segmented/__tests__/index.test.tsx
@@ -99,7 +99,7 @@ describe('Segmented', () => {
         .map(el => (el.getDOMNode() as HTMLInputElement).checked),
     ).toEqual([true, false, false]);
     expect(
-      wrapper.find(`.${prefixCls}-item`).at(0).hasClass(`${prefixCls}-item-selected`),
+      wrapper.find(`label.${prefixCls}-item`).at(0).hasClass(`${prefixCls}-item-selected`),
     ).toBeTruthy();
 
     wrapper.find(`.${prefixCls}-item-input`).at(2).simulate('change');
@@ -164,7 +164,7 @@ describe('Segmented', () => {
     );
     expect(wrapper.render()).toMatchSnapshot();
     expect(
-      wrapper.find(`.${prefixCls}-item`).at(1).hasClass(`${prefixCls}-item-disabled`),
+      wrapper.find(`label.${prefixCls}-item`).at(1).hasClass(`${prefixCls}-item-disabled`),
     ).toBeTruthy();
     expect(wrapper.find(`.${prefixCls}-item-input`).at(1).prop('disabled')).toBeTruthy();
 
@@ -304,7 +304,7 @@ describe('Segmented', () => {
         .map(el => (el.getDOMNode() as HTMLInputElement).checked),
     ).toEqual([true, false, false]);
     expect(
-      wrapper.find(`.${prefixCls}-item`).at(0).hasClass(`${prefixCls}-item-selected`),
+      wrapper.find(`label.${prefixCls}-item`).at(0).hasClass(`${prefixCls}-item-selected`),
     ).toBeTruthy();
 
     wrapper.find(`.${prefixCls}-item-input`).at(2).simulate('change');