Add acceptance criteria that cancelation must happen within 20ms

freeletics · May 6, 2023 · 33aebae · 33aebae
1 parent 89126d0
commit 33aebae
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/flowredux/src/commonTest/kotlin/com/freeletics/flowredux/dsl/OnActionTest.kt b/flowredux/src/commonTest/kotlin/com/freeletics/flowredux/dsl/OnActionTest.kt
@@ -15,10 +15,14 @@ import kotlinx.coroutines.test.advanceUntilIdle
 import kotlinx.coroutines.test.runCurrent
 import kotlinx.coroutines.test.runTest
 import kotlinx.coroutines.withContext
+import kotlin.time.DurationUnit
+import kotlin.time.ExperimentalTime
+import kotlin.time.measureTime
 
 @OptIn(ExperimentalCoroutinesApi::class)
 internal class OnActionTest {
 
+    @OptIn(ExperimentalTime::class)
     @Test
     fun actionBlockStopsWhenMovedToAnotherState() = runTest {
         val signal = Channel<Unit>()
@@ -35,10 +39,13 @@ internal class OnActionTest {
                     // Therefore we add a tiny delay and use a "real" dispatcher where delay would wait
                     // for real (in contrast to TestDipsatcher used with runTest {...})
                     // to avoid flakiness.
-                    withContext(Dispatchers.Default){
-                        // 20 ms should be enough to make sure that the cancellation happened in the meantime
-                        // because of state transition to TestState.S2 in on<TestAction.A2>.
-                        delay(20)
+                    withContext(Dispatchers.Default) {
+                        val timeElapsed = measureTime {
+                            // 20 ms should be enough to make sure that the cancellation happened in the meantime
+                            // because of state transition to TestState.S2 in on<TestAction.A2>.
+                            delay(20)
+                        }
+                        assertTrue(timeElapsed.toDouble(DurationUnit.MILLISECONDS) < 20, "Time Elapsed: $timeElapsed but expected to be < 20")
                     }
                     // this should never be reached because state transition did happen in the meantime,
                     // therefore this whole block must be canceled