fix tests

Author: Guccio163

tests/unit/ConfirmContentTest.tsx

@@ -2,26 +2,32 @@ import {render} from '@testing-library/react-native';
 
 import ConfirmContent from '@components/ConfirmContent';
 
+import CONST from '@src/CONST';
+
 import React from 'react';
 
 type ButtonProps = {
-    success?: boolean;
-    danger?: boolean;
-    text?: string;
+    variant?: string;
     onPress?: () => void;
+    children?: React.ReactNode;
     [key: string]: unknown;
 };
 
 const mockButtonSpy = jest.fn<void, [ButtonProps]>();
 
-jest.mock('@components/Button', () => {
+jest.mock('@components/ButtonComposed', () => {
     const ReactLib = jest.requireActual<typeof React>('react');
+    const MockButton = (props: ButtonProps) => {
+        mockButtonSpy(props);
+        return ReactLib.createElement('mock-button', props);
+    };
     return {
         __esModule: true,
-        default: (props: ButtonProps) => {
-            mockButtonSpy(props);
-            return ReactLib.createElement('mock-button', props);
-        },
+        default: Object.assign(MockButton, {
+            Text: () => null,
+            Icon: () => null,
+            KeyboardShortcut: () => null,
+        }),
     };
 });
 
@@ -65,12 +71,9 @@ describe('ConfirmContent', () => {
         mockButtonSpy.mockClear();
     });
 
-    function getConfirmButtonProps(shouldStackButtons: boolean): ButtonProps | undefined {
+    function getConfirmButtonProps(onConfirm: () => void): ButtonProps | undefined {
         const calls = mockButtonSpy.mock.calls;
-        if (shouldStackButtons) {
-            return calls.find((call) => call[0].pressOnEnter)?.[0];
-        }
-        return calls.find((call) => call[0].pressOnEnter)?.[0];
+        return calls.find((call) => call[0].onPress === onConfirm)?.[0];
     }
 
     const testCases = [
@@ -84,15 +87,26 @@ describe('ConfirmContent', () => {
         {shouldShowCancelButton: true, danger: true, success: true, expectedSuccess: false},
     ];
 
+    function expectedVariant(danger: boolean, expectedSuccess: boolean): string | undefined {
+        if (danger) {
+            return CONST.BUTTON_VARIANT.DANGER;
+        }
+        if (expectedSuccess) {
+            return CONST.BUTTON_VARIANT.SUCCESS;
+        }
+        return undefined;
+    }
+
     describe('stacked buttons (shouldStackButtons=true)', () => {
         it.each(testCases)(
-            'confirm button success=$expectedSuccess when shouldShowCancelButton=$shouldShowCancelButton, danger=$danger, success=$success',
+            'confirm button variant=$expectedSuccess when shouldShowCancelButton=$shouldShowCancelButton, danger=$danger, success=$success',
             ({shouldShowCancelButton, danger, success, expectedSuccess}) => {
                 mockButtonSpy.mockClear();
+                const onConfirm = jest.fn();
                 render(
                     <ConfirmContent
                         title="Test"
-                        onConfirm={jest.fn()}
+                        onConfirm={onConfirm}
                         isVisible
                         shouldStackButtons
                         shouldShowCancelButton={shouldShowCancelButton}
@@ -101,22 +115,22 @@ describe('ConfirmContent', () => {
                     />,
                 );
 
-                const confirmProps = getConfirmButtonProps(true);
-                expect(confirmProps?.success).toBe(expectedSuccess);
-                expect(confirmProps?.danger).toBe(danger);
+                const confirmProps = getConfirmButtonProps(onConfirm);
+                expect(confirmProps?.variant).toBe(expectedVariant(danger, expectedSuccess));
             },
         );
     });
 
     describe('side-by-side buttons (shouldStackButtons=false)', () => {
         it.each(testCases)(
-            'confirm button success=$expectedSuccess when shouldShowCancelButton=$shouldShowCancelButton, danger=$danger, success=$success',
+            'confirm button variant=$expectedSuccess when shouldShowCancelButton=$shouldShowCancelButton, danger=$danger, success=$success',
             ({shouldShowCancelButton, danger, success, expectedSuccess}) => {
                 mockButtonSpy.mockClear();
+                const onConfirm = jest.fn();
                 render(
                     <ConfirmContent
                         title="Test"
-                        onConfirm={jest.fn()}
+                        onConfirm={onConfirm}
                         isVisible
                         shouldStackButtons={false}
                         shouldShowCancelButton={shouldShowCancelButton}
@@ -125,9 +139,8 @@ describe('ConfirmContent', () => {
                     />,
                 );
 
-                const confirmProps = getConfirmButtonProps(false);
-                expect(confirmProps?.success).toBe(expectedSuccess);
-                expect(confirmProps?.danger).toBe(danger);
+                const confirmProps = getConfirmButtonProps(onConfirm);
+                expect(confirmProps?.variant).toBe(expectedVariant(danger, expectedSuccess));
             },
         );
     });