fix: set the label value to ElementType

rsuite · Jan 16, 2024 · 2605df2 · 2605df2
1 parent 24d67a1
commit 2605df2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/src/TagPicker/test/TagPickerSpec.tsx b/src/TagPicker/test/TagPickerSpec.tsx
@@ -13,7 +13,10 @@ import TagPicker from '../index';
 import Button from '../../Button';
 import { PickerHandle } from '../../Picker';
 
-const data = mockGroupData(['Eugenia', 'Kariane', 'Louisa'], { role: 'Master' });
+const data = mockGroupData(['Eugenia', 'Kariane', 'Louisa'], {
+  role: 'Master',
+  labelElementType: React.Fragment
+});
 
 describe('TagPicker', () => {
   testStandardProps(<TagPicker data={data} />, {
@@ -58,10 +61,10 @@ describe('TagPicker', () => {
   });
 
   it('Should not clean selected value', () => {
-    render(<TagPicker defaultOpen data={data} value={['Eugenia']} />);
+    render(<TagPicker data={data} value={['Eugenia']} />);
 
     fireEvent.click(screen.getByRole('button', { name: /clear/i }));
-    expect(screen.getByText('Eugenia', { selector: '.rs-tag-text' })).to.exist;
+    expect(screen.getByRole('option', { name: 'Eugenia' })).to.exist;
   });
 
   it('Should output a TagPicker', () => {
@@ -80,15 +83,15 @@ describe('TagPicker', () => {
     const value = 'Louisa';
     render(<TagPicker defaultOpen data={data} value={[value]} />);
 
-    expect(screen.getByText(value, { selector: '.rs-tag-text' })).to.exist;
+    expect(screen.queryAllByRole('option', { name: value })).to.have.lengthOf(2);
     expect(screen.getByRole('option', { name: value, selected: true })).to.exist;
   });
 
   it('Should active item by `defaultValue`', () => {
     const value = 'Louisa';
     render(<TagPicker defaultOpen data={data} defaultValue={[value]} />);
 
-    expect(screen.getByText(value, { selector: '.rs-tag-text' })).to.exist;
+    expect(screen.queryAllByRole('option', { name: value })).to.have.lengthOf(2);
     expect(screen.getByRole('option', { name: value, selected: true })).to.exist;
     expect(screen.getByRole('button', { name: /clear/i })).to.exist;
   });

diff --git a/test/mocks/data-mock.ts → test/mocks/data-mock.tsx b/test/mocks/data-mock.ts → test/mocks/data-mock.tsx
@@ -1,3 +1,4 @@
+import React from 'react';
 /**
  * Mock tree data
  * @example
@@ -37,13 +38,15 @@ export function mockTreeData(
   return items;
 }
 
-export function mockGroupData(data, appendInfo) {
+export function mockGroupData(data, options) {
   const items: any[] = [];
+  const { labelElementType: Label = 'span', ...rest } = options || {};
+
   data.forEach(item => {
     items.push({
-      label: item,
+      label: <Label>{item}</Label>,
       value: item,
-      ...appendInfo
+      ...rest
     });
   });
   return items;