test(ui): restore PKCE relogin/reconnect tests, remove only IMX ConnectBtn tests

Author: rodrigo-fournier-immutable

sample/Tests/test/test_android.py

@@ -107,4 +107,76 @@ def test_3_passport_functions(self):
         self.test_1_passport_functions()
 
     def test_5_zkevm_functions(self):
-        self.test_3_zkevm_functions()
+        self.test_3_zkevm_functions()
+
+    def test_6_pkce_relogin(self):
+        driver = self.appium_driver
+
+        self.close_and_open_app()
+
+        # Restart AltTester
+        self.altdriver.stop()
+        self.altdriver = AltDriver()
+        time.sleep(5)
+
+        # # Select use PKCE auth
+        self.altdriver.find_object(By.NAME, "PKCE").tap()
+        # Wait for unauthenticated screen
+        self.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
+
+        # Relogin
+        print("Re-logging in...")
+        self.altdriver.wait_for_object(By.NAME, "ReloginBtn").tap()
+
+        # Wait for authenticated screen
+        self.altdriver.wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Re-logged in")
+
+        # Get access token
+        self.altdriver.find_object(By.NAME, "GetAccessTokenBtn").tap()
+        output = self.altdriver.find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+
+        self.altdriver.stop()
+
+    def test_7_pkce_reconnect(self):
+        self.close_and_open_app()
+
+        # Restart AltTester
+        self.altdriver.stop()
+        self.altdriver = AltDriver()
+        time.sleep(5)
+
+        # Select use PKCE auth
+        self.altdriver.find_object(By.NAME, "PKCE").tap()
+        # Wait for unauthenticated screen
+        self.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
+        
+        # Reconnect
+        print("Reconnecting...")
+        self.altdriver.wait_for_object(By.NAME, "ReconnectBtn").tap()
+
+        # Wait for authenticated screen
+        self.altdriver.wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Reconnected")
+
+        # Get access token
+        self.altdriver.find_object(By.NAME, "GetAccessTokenBtn").tap()
+        output = self.altdriver.find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+
+        # Get address without having to click Connect to IMX button
+        self.altdriver.find_object(By.NAME, "GetAddressBtn").tap()
+        self.assertEqual(TestConfig.WALLET_ADDRESS, output.get_text())
+
+        # Logout
+        print("Logging out...")
+        self.altdriver.find_object(By.NAME, "LogoutBtn").tap()
+        time.sleep(5)
+        
+        # Wait for authenticated screen
+        self.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
+        time.sleep(5)
+        print("Logged out")
+
+        self.altdriver.stop()

sample/Tests/test/test_mac.py

@@ -174,6 +174,7 @@ def logout(cls):
         bring_sample_app_to_foreground()
         cls.altdriver.find_object(By.NAME, "LogoutBtn").tap()
         time.sleep(5)
+        bring_sample_app_to_foreground()  # Bring app back to foreground after browser processes logout
         cls.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
         time.sleep(2)
         cls.stop_browser()
@@ -231,4 +232,65 @@ def test_3_passport_functions(self):
         self.test_1_passport_functions()
 
     def test_5_zkevm_functions(self):
-        self.test_3_zkevm_functions()
+        self.test_3_zkevm_functions()
+
+    def test_6_relogin(self):
+        # Close and reopen app
+        stop_sample_app()
+        open_sample_app()
+
+        # Restart AltTester
+        self.altdriver.stop()
+        self.__class__.altdriver = AltDriver()
+        time.sleep(5)
+
+        # Wait for unauthenticated screen
+        self.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
+
+        # Relogin
+        print("Re-logging in...")
+        self.altdriver.wait_for_object(By.NAME, "ReloginBtn").tap()
+
+        # Wait for authenticated screen
+        self.altdriver.wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Re-logged in")
+
+        # Get access token
+        self.altdriver.find_object(By.NAME, "GetAccessTokenBtn").tap()
+        output = self.altdriver.find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+
+        self.altdriver.stop()
+
+    def test_7_reconnect_connect_imx(self):
+        # Close and reopen app
+        stop_sample_app()
+        open_sample_app()
+
+        # Restart AltTester
+        self.altdriver.stop()
+        self.__class__.altdriver = AltDriver()
+        time.sleep(5)
+
+        # Wait for unauthenticated screen
+        self.altdriver.wait_for_current_scene_to_be("UnauthenticatedScene")
+        
+        # Reconnect
+        print("Reconnecting...")
+        self.altdriver.wait_for_object(By.NAME, "ReconnectBtn").tap()
+
+        # Wait for authenticated screen
+        self.altdriver.wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Reconnected")
+
+        # Get access token
+        self.altdriver.find_object(By.NAME, "GetAccessTokenBtn").tap()
+        output = self.altdriver.find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+
+        # Get address without having to click Connect to IMX button
+        self.altdriver.find_object(By.NAME, "GetAddressBtn").tap()
+        self.assertEqual(TestConfig.WALLET_ADDRESS, output.get_text())
+
+        # Logout
+        self.logout()

sample/Tests/test/test_windows.py

@@ -205,3 +205,75 @@ def test_5_zkevm_functions(self):
         self.test_3_zkevm_functions()
         print("COMPLETED TEST: test_5_zkevm_functions")
         print("=" * 60)
+
+    def test_6_relogin(self):
+        print("=" * 60)
+        print("STARTING TEST: test_6_relogin")
+        print("=" * 60)
+        self.restart_app_and_altdriver()
+
+        # Relogin
+        print("Re-logging in...")
+        self.get_altdriver().wait_for_object(By.NAME, "ReloginBtn").tap()
+
+        # Wait for authenticated screen
+        self.get_altdriver().wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Re-logged in")
+
+        # Get access token
+        self.get_altdriver().find_object(By.NAME, "GetAccessTokenBtn").tap()
+        output = self.get_altdriver().find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+        
+        print("COMPLETED TEST: test_6_relogin")
+        print("=" * 60)
+
+    def test_7_reconnect_connect_imx(self):
+        print("=" * 60)
+        print("STARTING TEST: test_7_reconnect_connect_imx")
+        print("=" * 60)
+        self.restart_app_and_altdriver()
+
+        # Reconnect
+        print("Reconnecting...")
+        self.get_altdriver().wait_for_object(By.NAME, "ReconnectBtn").tap()
+
+        # Wait for authenticated screen
+        self.get_altdriver().wait_for_current_scene_to_be("AuthenticatedScene")
+        print("Reconnected")
+
+        # Get access token
+        self.get_altdriver().find_object(By.NAME, "GetAccessTokenBtn").tap()
+        time.sleep(2)  # Give Unity time to fetch and display the access token
+        output = self.get_altdriver().find_object(By.NAME, "Output")
+        self.assertTrue(len(output.get_text()) > 50)
+
+        # Get address without having to click Connect to IMX button
+        self.get_altdriver().find_object(By.NAME, "GetAddressBtn").tap()
+        time.sleep(2)  # Give Unity time to fetch and display the address
+        output = self.get_altdriver().find_object(By.NAME, "Output")  # Re-fetch output after address is loaded
+        self.assertEqual(TestConfig.WALLET_ADDRESS, output.get_text())
+
+        # Logout
+        print("Logging out...")
+        launch_browser()
+        bring_sample_app_to_foreground()
+        self.get_altdriver().find_object(By.NAME, "LogoutBtn").tap()
+        
+        # Use controlled browser logout instead of waiting for scene change
+        logout_with_controlled_browser()
+        
+        # Bring Unity app to foreground immediately so it can receive the logout callback
+        bring_sample_app_to_foreground()
+        
+        # Give Unity time to process the logout callback
+        time.sleep(5)
+
+        # Wait for unauthenticated screen
+        self.get_altdriver().wait_for_current_scene_to_be("UnauthenticatedScene")
+        
+        stop_browser()
+        print("Logged out")
+
+        print("COMPLETED TEST: test_7_reconnect_connect_imx")
+        print("=" * 60)