github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-25 15:40:28 +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 org.junit.jupiter.api.Test
import java.util.concurrent.atomic.AtomicInteger
import kotlin.coroutines.resume
import kotlin.test.assertEquals
import kotlin.time.milliseconds
import kotlin.time.seconds
internal class TestCoroutineUtils {
@ -54,18 +56,20 @@ internal class TestCoroutineUtils {
fun `test launchTimedTask finishes multiple times`() = runBlocking {
val scope = CoroutineScope(SupervisorJob())
withTimeout(10000) {
suspendCancellableCoroutine<Unit> { cont ->
val task = scope.launchTimedTask(3.seconds.toLongMilliseconds()) {
cont.resume(Unit)
}
task.setChanged()
launch {
delay(4000)
task.setChanged()
}
}
val resumedTimes = AtomicInteger(0)
val task = scope.launchTimedTask(3000.milliseconds.toLongMilliseconds()) {
resumedTimes.incrementAndGet()
}
task.setChanged()
launch {
delay(4000)
task.setChanged()
}
delay(6000)
assertEquals(1, resumedTimes.get())
delay(15000)
assertEquals(2, resumedTimes.get())
scope.cancel()
}