Enable Robolectric paused mode (#35)

vinted · Sep 13, 2023 · 78a446d · 78a446d
1 parent 9d76e73
commit 78a446d
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/library/src/test/java/com/vinted/coper/CoperImplTest.kt b/library/src/test/java/com/vinted/coper/CoperImplTest.kt
@@ -23,13 +23,15 @@ import org.robolectric.Robolectric
 import org.robolectric.RobolectricTestRunner
 import org.robolectric.Shadows.shadowOf
 import org.robolectric.annotation.Config
+import org.robolectric.annotation.LooperMode
 import kotlin.test.assertEquals
 import kotlin.test.assertFalse
 import kotlin.test.assertNotEquals
 import kotlin.test.assertTrue
 
 @ExperimentalCoroutinesApi
 @RunWith(RobolectricTestRunner::class)
+@LooperMode(LooperMode.Mode.PAUSED)
 @Config(manifest = Config.NONE, sdk = [23, 27])
 class CoperImplTest {
 
@@ -39,7 +41,7 @@ class CoperImplTest {
 
     @Before
     fun setup() = runTest {
-        fixture.mockGetFragmentWithStub()
+        runIdlingMainThread { fixture.mockGetFragmentWithStub() }
         mockCheckPermissions("test", PackageManager.PERMISSION_GRANTED)
     }
 
@@ -415,7 +417,7 @@ class CoperImplTest {
         )
 
         val fragment = withTimeout(1000) {
-            fixture.getFragmentSafely()
+            runIdlingMainThread { fixture.getFragmentSafely() }
         }
 
         assertEquals(fragment, activity.supportFragmentManager.findFragmentByTag(FRAGMENT_TAG))
@@ -467,7 +469,7 @@ class CoperImplTest {
             lifecycle = activity.lifecycle,
         )
 
-        val fragment = fixture.getFragmentSafely()
+        val fragment = runIdlingMainThread { fixture.getFragmentSafely() }
 
         val job = async {
             fixture.request(permission)
@@ -486,7 +488,7 @@ class CoperImplTest {
             lifecycle = activity.lifecycle,
             fragmentManager = fragmentManager
         )
-        val fragment = fixture.getFragmentSafely()
+        val fragment = runIdlingMainThread { fixture.getFragmentSafely() }
         val permission = "onConfigurationChange"
         activity.setPermissionResult(permission, PackageManager.PERMISSION_DENIED)
 
@@ -854,4 +856,12 @@ class CoperImplTest {
             .add(spyCoperFragment, FRAGMENT_TAG)
             .commitNow()
     }
+
+    private suspend fun <T> runIdlingMainThread(block: suspend () -> T): T {
+        return coroutineScope {
+            val asyncBlock = async { block() }
+            val asyncMainThreadIdle = async { shadowOf(getMainLooper()).idle() }
+            awaitAll(asyncBlock, asyncMainThreadIdle).first() as T
+        }
+    }
 }