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

Make NettyUtilsTest faster

Browse Source
This commit is contained in:
Him188 2021-05-31 00:31:54 +08:00
parent ff6c63fafc
commit e0c0a189dc
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mirai-core/src/commonTest/kotlin/network/impl/netty/NettyUtilsTest.kt
View File

@ -11,7 +11,7 @@ package net.mamoe.mirai.internal.network.impl.netty
import io.netty.channel.DefaultChannelPromise import io.netty.channel.DefaultChannelPromise
import io.netty.channel.embedded.EmbeddedChannel import io.netty.channel.embedded.EmbeddedChannel
import kotlinx.coroutines.delay import kotlinx.coroutines.CoroutineStart
import kotlinx.coroutines.launch import kotlinx.coroutines.launch
import net.mamoe.mirai.internal.test.AbstractTest import net.mamoe.mirai.internal.test.AbstractTest
import net.mamoe.mirai.internal.test.runBlockingUnit import net.mamoe.mirai.internal.test.runBlockingUnit
@ -19,7 +19,7 @@ import org.junit.jupiter.api.AfterAll
import org.junit.jupiter.api.Test import org.junit.jupiter.api.Test
import kotlin.test.assertFailsWith import kotlin.test.assertFailsWith
import kotlin.test.assertTrue import kotlin.test.assertTrue
import kotlin.time.seconds import kotlin.time.Duration
/** /**
* @see awaitKt * @see awaitKt
@ -36,17 +36,16 @@ internal class NettyUtilsTest : AbstractTest() {
} }
@Test @Test
fun canAwait() = runBlockingUnit(timeout = 5.seconds) { fun canAwait() = runBlockingUnit(timeout = Duration.seconds(5)) {
val future = DefaultChannelPromise(channel) val future = DefaultChannelPromise(channel)
launch(start = CoroutineStart.UNDISPATCHED) { future.awaitKt() }
launch { launch {
delay(2000)
future.setSuccess() future.setSuccess()
} }
future.awaitKt()
} }
@Test @Test
fun returnsImmediatelyIfCompleted() = runBlockingUnit(timeout = 5.seconds) { fun returnsImmediatelyIfCompleted() = runBlockingUnit(timeout = Duration.seconds(5)) {
val future = DefaultChannelPromise(channel) val future = DefaultChannelPromise(channel)
future.setSuccess() future.setSuccess()
future.awaitKt() future.awaitKt()
@ -56,16 +55,17 @@ internal class NettyUtilsTest : AbstractTest() {
fun testAwait() { fun testAwait() {
class MyError : AssertionError("My") // coroutine debugger will modify the exception if inside coroutine class MyError : AssertionError("My") // coroutine debugger will modify the exception if inside coroutine
runBlockingUnit(timeout = 5.seconds) { runBlockingUnit(timeout = Duration.seconds(5)) {
val future = DefaultChannelPromise(channel) val future = DefaultChannelPromise(channel)
launch { launch(start = CoroutineStart.UNDISPATCHED) {
delay(2000) assertFailsWith<AssertionError> {
future.setFailure(MyError()) future.awaitKt()
}.let { actual ->
assertTrue { actual is MyError }
}
} }
assertFailsWith<AssertionError> { launch {
future.awaitKt() future.setFailure(MyError())
}.let { actual ->
assertTrue { actual is MyError }
} }
} }
} }