Try adding composeTestRule.waitForIdle()

Author: seladb

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

@@ -90,6 +90,8 @@ class StatsScreenTest {
                 }
             }
         }
+
+        composeTestRule.waitForIdle()
     }
 
     @Before
@@ -101,7 +103,6 @@ class StatsScreenTest {
     fun testVpnConnected() {
         renderScreen()
 
-        composeTestRule.waitForIdle()
         composeTestRule.onNodeWithText("Disconnect").assertIsDisplayed().assertIsEnabled()
     }
 
@@ -149,7 +150,6 @@ class StatsScreenTest {
                 Triple("UDPConn", "UDP", udpConnectionCount),
             )
 
-        composeTestRule.waitForIdle()
         composeTestRule.onNodeWithText("Total Packets").assertIsDisplayed().onSibling().assertTextEquals(totalPacketCount.toString())
 
         expectedValues.forEach { (testTag, label, count) ->
@@ -196,7 +196,6 @@ class StatsScreenTest {
         renderScreen()
 
         composeTestRule.onNodeWithText("Disconnect").performClick()
-        composeTestRule.waitForIdle()
 
         verify { mockViewModel.disconnectVpn() }
     }
@@ -205,7 +204,6 @@ class StatsScreenTest {
     fun testVpnDisconnecting() {
         renderScreen(vpnConnectionState = VpnConnectionState.DISCONNECTING)
 
-        composeTestRule.waitForIdle()
         composeTestRule.onNodeWithText("Disconnecting...").assertIsDisplayed().assertIsNotEnabled()
     }