From e13b21f577d39f8f052d064a833fde1a7c6acca5 Mon Sep 17 00:00:00 2001
From: Petr Knetl <petr.knetl@satoshilabs.com>
Date: Fri, 17 Jan 2025 08:31:31 +0100
Subject: [PATCH] fix(suite-native): welcome flow test ids

---
 suite-native/atoms/src/Switch.tsx                            | 5 +++--
 .../module-onboarding/src/screens/AnalyticsConsentScreen.tsx | 5 +++--
 suite-native/module-onboarding/src/screens/WelcomeScreen.tsx | 5 ++++-
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/suite-native/atoms/src/Switch.tsx b/suite-native/atoms/src/Switch.tsx
index 0e1cd19dee..39ad60ed2d 100644
--- a/suite-native/atoms/src/Switch.tsx
+++ b/suite-native/atoms/src/Switch.tsx
@@ -17,6 +17,7 @@ type SwitchProps = {
     isChecked: boolean;
     onChange: (value: boolean) => void;
     isDisabled?: boolean; // Functionality of disabled works but styles are not implemented yet (waiting for design)
+    testID?: string;
 };
 
 const SWITCH_CONTAINER_BORDER_WIDTH = 1;
@@ -89,7 +90,7 @@ const useAnimationStyles = ({ isChecked }: Pick<SwitchProps, 'isChecked'>) => {
     };
 };
 
-export const Switch = ({ isChecked, onChange, isDisabled = false }: SwitchProps) => {
+export const Switch = ({ isChecked, onChange, isDisabled = false, testID }: SwitchProps) => {
     const { applyStyle } = useNativeStyles();
 
     const { animatedSwitchCircleStyle, animatedSwitchContainerStyle } = useAnimationStyles({
@@ -102,7 +103,7 @@ export const Switch = ({ isChecked, onChange, isDisabled = false }: SwitchProps)
     };
 
     return (
-        <Pressable onPress={handlePress} accessibilityRole="switch">
+        <Pressable onPress={handlePress} accessibilityRole="switch" testID={testID}>
             <Animated.View
                 style={[
                     animatedSwitchContainerStyle,
diff --git a/suite-native/module-onboarding/src/screens/AnalyticsConsentScreen.tsx b/suite-native/module-onboarding/src/screens/AnalyticsConsentScreen.tsx
index c3a6b53d71..f89741c80e 100644
--- a/suite-native/module-onboarding/src/screens/AnalyticsConsentScreen.tsx
+++ b/suite-native/module-onboarding/src/screens/AnalyticsConsentScreen.tsx
@@ -98,6 +98,7 @@ export const AnalyticsConsentScreen = () => {
                                         <Translation id="moduleOnboarding.analyticsConsentScreen.helpSwitchTitle" />
                                     </Text>
                                     <Switch
+                                        testID="@onboarding/AnalyticsConsent/consentSwitch"
                                         isChecked={isEnabled}
                                         onChange={enabled => {
                                             setIsEnabled(enabled);
@@ -110,14 +111,14 @@ export const AnalyticsConsentScreen = () => {
                 </VStack>
                 <VStack spacing="sp12">
                     <Button
-                        testID="@onboarding/UserDataConsent/allow"
+                        testID="@onboarding/AnalyticsConsent/nextBtn"
                         onPress={isEnabled ? handleAnalyticsConsent : handleRedirect}
                     >
                         <Translation id="generic.buttons.confirm" />
                     </Button>
                     <Button
                         colorScheme="tertiaryElevation0"
-                        testID="@onboarding/UserDataConsent/allow"
+                        testID="@onboarding/AnalyticsConsent/learMoreBtn"
                         onPress={handleClickOnLearMoreLink}
                     >
                         <Translation id="moduleOnboarding.analyticsConsentScreen.learnMoreButton" />
diff --git a/suite-native/module-onboarding/src/screens/WelcomeScreen.tsx b/suite-native/module-onboarding/src/screens/WelcomeScreen.tsx
index 494b2d8803..600875b5f2 100644
--- a/suite-native/module-onboarding/src/screens/WelcomeScreen.tsx
+++ b/suite-native/module-onboarding/src/screens/WelcomeScreen.tsx
@@ -77,7 +77,10 @@ export const WelcomeScreen = ({
                         </Box>
                     </VStack>
                     <Box style={applyStyle(buttonWrapperStyle)}>
-                        <Button onPress={navigateToAnalyticsConsent}>
+                        <Button
+                            onPress={navigateToAnalyticsConsent}
+                            testID="@onboarding/Welcome/nextBtn"
+                        >
                             <Translation id="moduleOnboarding.welcomeScreen.button" />
                         </Button>
                     </Box>