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,66 @@ 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())
+
+        self.altdriver.stop()

sample/Tests/test/test_mac.py

@@ -273,3 +273,61 @@ def test_3_passport_functions(self):
 
     def test_5_zkevm_functions(self):
         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())

sample/Tests/test/test_windows.py

@@ -250,3 +250,52 @@ 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()
+        self.assertEqual(TestConfig.WALLET_ADDRESS, output.get_text())
+
+        print("COMPLETED TEST: test_7_reconnect_connect_imx")
+        print("=" * 60)