Unregister idling resource at constructir time

badoo · May 8, 2024 · f6e7911 · f6e7911
1 parent 8d86462
commit f6e7911
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...esso-server/app/src/androidTest/java/io/appium/espressoserver/EspressoServerRunnerTest.kt b/...esso-server/app/src/androidTest/java/io/appium/espressoserver/EspressoServerRunnerTest.kt
@@ -79,6 +79,15 @@ class EspressoServerRunnerTest {
 
     init {
         context.driverContextChangeListener = listener
+        if (context.currentStrategyType == DriverContext.StrategyType.COMPOSE) {
+            composeIdlingResource?.let {
+                composeTestRule.registerIdlingResource(composeIdlingResource)
+            }
+        } else {
+            composeIdlingResource?.let {
+                composeTestRule.unregisterIdlingResource(composeIdlingResource)
+            }
+        }
     }