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

[core] Tried to fix AuthorizationReasonTest.force offline

Browse Source
This commit is contained in:
Karlatemp 2023-07-25 15:29:32 +08:00
parent 061bd28798
commit 8a8689eb82
No known key found for this signature in database
GPG Key ID: BA173CA2B9956C59
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mirai-core/src/commonTest/kotlin/network/auth/AuthorizationReasonTest.kt
View File

@ -10,6 +10,7 @@
package net.mamoe.mirai.internal.network.auth
import kotlinx.coroutines.test.runTest
import kotlinx.coroutines.yield
import net.mamoe.mirai.auth.AuthReason
import net.mamoe.mirai.auth.BotAuthResult
import net.mamoe.mirai.internal.MockAccount
@ -22,6 +23,8 @@ import net.mamoe.mirai.internal.network.protocol.packet.IncomingPacket
import net.mamoe.mirai.internal.network.protocol.packet.chat.receive.MessageSvcPushForceOffline
import net.mamoe.mirai.internal.network.protocol.packet.login.StatSvc
import net.mamoe.mirai.internal.network.protocol.packet.login.WtLogin
import org.junit.jupiter.api.RepeatedTest
import java.util.concurrent.atomic.AtomicReference
import kotlin.test.Test
import kotlin.test.assertFalse
import kotlin.test.assertIs
@ -106,14 +109,17 @@ internal class AuthorizationReasonTest : AbstractBotAuthTest() {
assertIs<AuthReason.FastLoginError>(authReason)
}
@Test
// @Test
@RepeatedTest(20)
fun `force offline`() = runTest {
var isFirstLogin: Boolean = true
var authReason: AuthReason? = null
// volatile
val authReason = AtomicReference<AuthReason?>(null)
setAuthorization { auth, info ->
isFirstLogin = info.isFirstLogin
authReason = info.reason
authReason.set(info.reason)
auth.authByPassword("")
return@setAuthorization object : BotAuthResult {}
@ -142,8 +148,9 @@ internal class AuthorizationReasonTest : AbstractBotAuthTest() {
)
eventDispatcher.joinBroadcast() // why test finished before code reaches end??
yield()
assertFalse(isFirstLogin)
assertIs<AuthReason.ForceOffline>(authReason)
assertIs<AuthReason.ForceOffline>(authReason.get(), message = authReason.toString())
}
}