@@ -20,6 +20,9 @@ def tearDownClass(cls):
20
20
stop_sample_app ()
21
21
22
22
def test_1_device_code_login (self ):
23
+ # Select use device code auth
24
+ self .altdriver .find_object (By .NAME , "DeviceCodeAuth" ).tap ()
25
+
23
26
# Wait for unauthenticated screen
24
27
self .altdriver .wait_for_current_scene_to_be ("UnauthenticatedScene" )
25
28
@@ -48,9 +51,12 @@ def test_1_device_code_login(self):
48
51
if attempt == 0 :
49
52
# Reset app
50
53
51
- # Relogin
54
+ # Relogin (optional: only if the button is present)
52
55
print ("Try reset the app and log out once..." )
53
- self .altdriver .wait_for_object (By .NAME , "ReloginBtn" ).tap ()
56
+ try :
57
+ self .altdriver .wait_for_object (By .NAME , "ReloginBtn" ).tap ()
58
+ except Exception as e :
59
+ print ("ReloginBtn not found, skipping relogin step. User may already be in AuthenticatedScene." )
54
60
55
61
# Wait for authenticated screen
56
62
self .altdriver .wait_for_current_scene_to_be ("AuthenticatedScene" )
@@ -95,6 +101,9 @@ def test_6_relogin(self):
95
101
self .altdriver = AltDriver ()
96
102
time .sleep (5 )
97
103
104
+ # Select use device code auth
105
+ self .altdriver .find_object (By .NAME , "DeviceCodeAuth" ).tap ()
106
+
98
107
# Relogin
99
108
print ("Re-logging in..." )
100
109
self .altdriver .wait_for_object (By .NAME , "ReloginBtn" ).tap ()
@@ -124,6 +133,9 @@ def test_7_reconnect_device_code_connect_imx(self):
124
133
self .altdriver = AltDriver ()
125
134
time .sleep (5 )
126
135
136
+ # Select use device code auth
137
+ self .altdriver .find_object (By .NAME , "DeviceCodeAuth" ).tap ()
138
+
127
139
# Reconnect
128
140
print ("Reconnecting..." )
129
141
self .altdriver .wait_for_object (By .NAME , "ReconnectBtn" ).tap ()
0 commit comments