Try waitUntil in another test

Author: seladb

app/src/androidTest/java/com/pcapplusplus/toyvpn/StatsScreenTest.kt

@@ -2,7 +2,6 @@ package com.pcapplusplus.toyvpn
 
 import androidx.compose.material3.Text
 import androidx.compose.ui.test.assertIsDisplayed
-import androidx.compose.ui.test.assertIsEnabled
 import androidx.compose.ui.test.assertTextEquals
 import androidx.compose.ui.test.junit4.createComposeRule
 import androidx.compose.ui.test.onNodeWithTag
@@ -102,7 +101,14 @@ class StatsScreenTest {
     fun testVpnConnected() {
         renderScreen()
 
-        composeTestRule.onNodeWithText("Disconnect").assertIsDisplayed().assertIsEnabled()
+        composeTestRule.waitUntil(timeoutMillis = 10000) {
+            try {
+                composeTestRule.onNodeWithText("Disconnect").assertExists()
+                true
+            } catch (ex: AssertionError) {
+                false
+            }
+        }
     }
 
     @Test
@@ -205,7 +211,7 @@ class StatsScreenTest {
 
         composeTestRule.waitUntil(timeoutMillis = 10000) {
             try {
-                composeTestRule.onNodeWithText("Disconnecting...").fetchSemanticsNode()
+                composeTestRule.onNodeWithText("Disconnecting...").assertExists()
                 true
             } catch (ex: AssertionError) {
                 false