test: add 'times' parameter to runTest

useful for diagnostics
RobozinhoD · Jan 2, 2024 · 1d1aa78 · 1d1aa78
1 parent 9d212a4
commit 1d1aa78
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/AnkiDroid/src/test/java/com/ichi2/testutils/TestClass.kt b/AnkiDroid/src/test/java/com/ichi2/testutils/TestClass.kt
@@ -31,6 +31,7 @@ import kotlinx.coroutines.test.TestScope
 import kotlinx.coroutines.test.UnconfinedTestDispatcher
 import kotlinx.coroutines.test.setMain
 import net.ankiweb.rsdroid.exceptions.BackendDeckIsFilteredException
+import timber.log.Timber
 import kotlin.coroutines.CoroutineContext
 import kotlin.coroutines.EmptyCoroutineContext
 import kotlin.time.Duration.Companion.milliseconds
@@ -175,12 +176,16 @@ interface TestClass {
     fun runTest(
         context: CoroutineContext = EmptyCoroutineContext,
         dispatchTimeoutMs: Long = 60_000L,
+        times: Int = 1,
         testBody: suspend TestScope.() -> Unit
     ) {
         Dispatchers.setMain(UnconfinedTestDispatcher())
-        kotlinx.coroutines.test.runTest(context, dispatchTimeoutMs.milliseconds) {
-            CollectionManager.setTestDispatcher(UnconfinedTestDispatcher(testScheduler))
-            testBody()
+        repeat(times) {
+            if (times != 1) Timber.d("------ Executing test $it/$times ------")
+            kotlinx.coroutines.test.runTest(context, dispatchTimeoutMs.milliseconds) {
+                CollectionManager.setTestDispatcher(UnconfinedTestDispatcher(testScheduler))
+                testBody()
+            }
         }
     }
 }