test(ui): fix test for dIconRight of button

DevCloudFE · Mar 17, 2023 · d2dc4f1 · d2dc4f1
1 parent 9e4339e
commit d2dc4f1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/ui/src/components/button/Button.test.tsx b/packages/ui/src/components/button/Button.test.tsx
@@ -13,7 +13,7 @@ const getTypeClass = (dType: DButtonProps['dType'], prefix = dPrefix) => {
 const getThemeClass = (dTheme: DButtonProps['dTheme'], prefix = 't') => {
   return `${prefix}-${dTheme || 'primary'}`;
 };
-const getOtherClass = (ots?: 'icon' | DSize | 'block' | DButtonProps['dVariant'] | 'icon-right', separator = '--', prefix = dPrefix) => {
+const getOtherClass = (ots?: 'icon' | DSize | 'block' | DButtonProps['dVariant'], separator = '--', prefix = dPrefix) => {
   return !ots ? `${prefix}button` : `${prefix}button${separator}${ots || ''}`;
 };
 
@@ -191,7 +191,8 @@ describe('DButton', () => {
         </DButton>
       );
       const button = getByRole('button');
-      expect(button.className.includes(getOtherClass('icon-right'))).toBe(true);
+      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+      expect(button.children.item(0)!.innerHTML.includes('Hello World')).toBe(true);
     });
   });