kscience/dataforge-core
8
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases 12 Wiki Activity

Fix flaky coroutines tests

Browse Source
This commit is contained in:
Alexander Nozik 2024-07-22 11:45:50 +03:00
parent b4ebdfe089
commit 523db20e4a
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
dataforge-data/src/jvmTest/kotlin/space/kscience/dataforge/data/ActionsTest.kt
View File

@ -1,6 +1,11 @@
package space.kscience.dataforge.data package space.kscience.dataforge.data
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.delay
import kotlinx.coroutines.test.advanceUntilIdle
import kotlinx.coroutines.test.runTest import kotlinx.coroutines.test.runTest
import kotlinx.coroutines.withContext
import space.kscience.dataforge.actions.Action import space.kscience.dataforge.actions.Action
import space.kscience.dataforge.actions.invoke import space.kscience.dataforge.actions.invoke
import space.kscience.dataforge.actions.mapping import space.kscience.dataforge.actions.mapping
@ -9,7 +14,7 @@ import kotlin.test.Test
import kotlin.test.assertEquals import kotlin.test.assertEquals
import kotlin.time.Duration.Companion.milliseconds import kotlin.time.Duration.Companion.milliseconds
@OptIn(DFExperimental::class) @OptIn(DFExperimental::class, ExperimentalCoroutinesApi::class)
internal class ActionsTest { internal class ActionsTest {
@Test @Test
fun testStaticMapAction() = runTest(timeout = 500.milliseconds) { fun testStaticMapAction() = runTest(timeout = 500.milliseconds) {
@ -24,6 +29,8 @@ internal class ActionsTest {
} }
val result = plusOne(data) val result = plusOne(data)
advanceUntilIdle()
assertEquals(2, result["1"]?.await()) assertEquals(2, result["1"]?.await())
} }
@ -38,10 +45,14 @@ internal class ActionsTest {
val result = plusOne(source) val result = plusOne(source)
withContext(Dispatchers.Default) {
repeat(10) { repeat(10) {
source.updateValue(it.toString(), it) source.updateValue(it.toString(), it)
} }
delay(50)
}
// result.updates.take(10).onEach { println(it.name) }.collect() // result.updates.take(10).onEach { println(it.name) }.collect()
assertEquals(2, result["1"]?.await()) assertEquals(2, result["1"]?.await())