@W-21933885: [MSDK Android] App Attestation Implementation (Fix TokenMigrationViewActivityTest Reliability For Firebase Test Lab)

Author: JohnsonEricAtSalesforce

libs/test/SalesforceSDKTest/src/com/salesforce/androidsdk/ui/TokenMigrationViewActivityTest.kt

@@ -43,6 +43,10 @@ import androidx.compose.ui.test.onNodeWithContentDescription
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.ViewModelProvider
 import androidx.lifecycle.viewmodel.CreationExtras
+import androidx.test.espresso.Espresso
+import androidx.test.espresso.IdlingPolicies
+import androidx.test.espresso.IdlingRegistry
+import androidx.test.espresso.IdlingResource
 import androidx.test.rule.GrantPermissionRule
 import com.salesforce.androidsdk.R
 import com.salesforce.androidsdk.app.SalesforceSDKManager
@@ -56,13 +60,93 @@ import org.junit.Before
 import org.junit.Ignore
 import org.junit.Rule
 import org.junit.Test
+import org.junit.rules.Timeout
+import java.util.concurrent.TimeUnit
+
+/**
+ * IdlingResource that waits for a condition to become true.
+ * Used to synchronize ViewModel state changes with test assertions.
+ */
+class ViewModelIdlingResource(
+    private val resourceName: String,
+    private val checkCondition: () -> Boolean
+) : IdlingResource {
+    @Volatile
+    private var callback: IdlingResource.ResourceCallback? = null
+
+    override fun getName(): String = resourceName
+
+    override fun isIdleNow(): Boolean {
+        val idle = checkCondition()
+        if (idle && callback != null) {
+            callback?.onTransitionToIdle()
+        }
+        return idle
+    }
+
+    override fun registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback?) {
+        this.callback = callback
+    }
+}
+
+/**
+ * Helper function to wait for a condition using IdlingResource.
+ * Automatically registers and unregisters the idling resource.
+ * Uses Espresso's built-in synchronization mechanism instead of manual polling.
+ *
+ * @param resourceName Name for the idling resource (for debugging)
+ * @param condition Lambda that returns true when the condition is met
+ * @param timeoutSeconds Maximum time to wait for the condition (default: 10 seconds)
+ * @param block Code to execute once the condition is met
+ */
+inline fun <T> waitForCondition(
+    resourceName: String,
+    noinline condition: () -> Boolean,
+    timeoutSeconds: Long = 10,
+    block: () -> T
+): T {
+    val idlingResource = ViewModelIdlingResource(resourceName, condition)
+
+    // Set custom timeout for this wait operation
+    val previousTimeout = IdlingPolicies.getMasterIdlingPolicy()
+    IdlingPolicies.setMasterPolicyTimeout(timeoutSeconds, TimeUnit.SECONDS)
+
+    IdlingRegistry.getInstance().register(idlingResource)
+    try {
+        // Let Espresso handle the synchronization - it will poll the IdlingResource
+        // and wait until isIdleNow() returns true, respecting the timeout policy
+        Espresso.onIdle()
+
+        return block()
+    } catch (e: Exception) {
+        // Provide better error message if timeout occurs
+        if (e.message?.contains("IdlingResource") == true ||
+            e.message?.contains("timeout") == true) {
+            throw AssertionError("Timeout waiting for condition '$resourceName' after ${timeoutSeconds}s", e)
+        }
+        throw e
+    } finally {
+        IdlingRegistry.getInstance().unregister(idlingResource)
+        // Restore previous timeout policy
+        IdlingPolicies.setMasterPolicyTimeout(
+            previousTimeout.idleTimeout,
+            previousTimeout.idleTimeoutUnit
+        )
+    }
+}
 
-@Ignore
 class TokenMigrationViewActivityTest {
 
     @get:Rule
     val androidComposeTestRule = createAndroidComposeRule<ComponentActivity>()
 
+    /**
+     * Global timeout rule for all tests in this class.
+     * Each test will timeout after 30 seconds to accommodate slower Firebase Test Lab devices.
+     */
+    @get:Rule
+    val globalTimeout: Timeout = Timeout(30, TimeUnit.SECONDS)
+
     // TODO: Remove if when min SDK version is 33
     @get:Rule
     val permissionRule: GrantPermissionRule = if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.TIRAMISU) {
@@ -172,11 +256,18 @@ class TokenMigrationViewActivityTest {
             }
         }
 
+        // Update background color
         androidComposeTestRule.runOnIdle {
             backgroundColor.value = White
         }
 
-        androidComposeTestRule.runOnIdle {
+        // Wait for ViewModel state to update
+        waitForCondition(
+            resourceName = "BackgroundColorUpdatedToWhite",
+            condition = {
+                backgroundColor.value == White
+            }
+        ) {
             assertEquals(
                 "Background color should update to White",
                 White,
@@ -201,11 +292,18 @@ class TokenMigrationViewActivityTest {
             }
         }
 
+        // Update background color
         androidComposeTestRule.runOnIdle {
             backgroundColor.value = Red
         }
 
-        androidComposeTestRule.runOnIdle {
+        // Wait for ViewModel state to update
+        waitForCondition(
+            resourceName = "BackgroundColorUpdatedToRed",
+            condition = {
+                backgroundColor.value == Red
+            }
+        ) {
             assertEquals(
                 "Background color should update to Red",
                 Red,