Use MainScope for tracking window layout info in BaseSingleFoldDeviceAction

PiperOrigin-RevId: 548821296

Author: Paige McAuliffe

espresso/device/java/androidx/test/espresso/device/action/BUILD

@@ -24,5 +24,6 @@ kt_android_library(
         "//runner/monitor",
         "@maven//:androidx_window_window",
         "@maven//:androidx_window_window_java",
+        "@maven//:org_jetbrains_kotlinx_kotlinx_coroutines_android",
     ],
 )

espresso/device/java/androidx/test/espresso/device/action/BaseSingleFoldDeviceAction.kt

@@ -30,8 +30,7 @@ import androidx.window.layout.WindowLayoutInfo
 import java.util.concurrent.CountDownLatch
 import java.util.concurrent.Executor
 import java.util.concurrent.TimeUnit
-import kotlinx.coroutines.CoroutineScope
-import kotlinx.coroutines.asCoroutineDispatcher
+import kotlinx.coroutines.MainScope
 import kotlinx.coroutines.flow.collect
 import kotlinx.coroutines.flow.distinctUntilChanged
 import kotlinx.coroutines.launch
@@ -62,7 +61,7 @@ internal open class BaseSingleFoldDeviceAction(
     val latch: CountDownLatch = CountDownLatch(1)
     val windowInfoTracker = WindowInfoTracker.getOrCreate(activity)
 
-    CoroutineScope(mainExecutor.asCoroutineDispatcher()).launch {
+    MainScope().launch {
       windowInfoTracker.windowLayoutInfo(activity).distinctUntilChanged().collect {
         windowLayoutInfo: WindowLayoutInfo ->
         val foldingFeatures = windowLayoutInfo.displayFeatures.filterIsInstance<FoldingFeature>()