github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-26 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

TestCoroutineUtils: add more cases

Browse Source
This commit is contained in:
Him188 2021-04-16 16:56:37 +08:00
parent 9cb75bd410
commit 76ab9f220e
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
backend/mirai-console/test/util/TestCoroutineUtils.kt
View File

@ -12,8 +12,10 @@ package net.mamoe.mirai.console.util
import kotlinx.coroutines.* import kotlinx.coroutines.*
import org.junit.jupiter.api.Test import org.junit.jupiter.api.Test
import java.util.concurrent.atomic.AtomicInteger
import kotlin.coroutines.resume import kotlin.coroutines.resume
import kotlin.test.assertEquals import kotlin.test.assertEquals
import kotlin.time.milliseconds
import kotlin.time.seconds import kotlin.time.seconds
internal class TestCoroutineUtils { internal class TestCoroutineUtils {
@ -54,18 +56,20 @@ internal class TestCoroutineUtils {
fun `test launchTimedTask finishes multiple times`() = runBlocking { fun `test launchTimedTask finishes multiple times`() = runBlocking {
val scope = CoroutineScope(SupervisorJob()) val scope = CoroutineScope(SupervisorJob())
withTimeout(10000) { val resumedTimes = AtomicInteger(0)
suspendCancellableCoroutine<Unit> { cont -> val task = scope.launchTimedTask(3000.milliseconds.toLongMilliseconds()) {
val task = scope.launchTimedTask(3.seconds.toLongMilliseconds()) { resumedTimes.incrementAndGet()
cont.resume(Unit)
}
task.setChanged()
launch {
delay(4000)
task.setChanged()
}
}
} }
task.setChanged()
launch {
delay(4000)
task.setChanged()
}
delay(6000)
assertEquals(1, resumedTimes.get())
delay(15000)
assertEquals(2, resumedTimes.get())
scope.cancel() scope.cancel()
} }