From bebb786b9ad322aed7b6c59722fbf4e72a809407 Mon Sep 17 00:00:00 2001
From: Him188 <Him188@mamoe.net>
Date: Sun, 13 Jun 2021 11:10:12 +0800
Subject: [PATCH] Fix build

---
 .../network/framework/AbstractRealNetworkHandlerTest.kt   | 8 ++++++++
 .../network/framework/components/TestSsoProcessor.kt      | 6 ++++++
 2 files changed, 14 insertions(+)

diff --git a/mirai-core/src/commonTest/kotlin/network/framework/AbstractRealNetworkHandlerTest.kt b/mirai-core/src/commonTest/kotlin/network/framework/AbstractRealNetworkHandlerTest.kt
index 78eea477e..5c9aa8296 100644
--- a/mirai-core/src/commonTest/kotlin/network/framework/AbstractRealNetworkHandlerTest.kt
+++ b/mirai-core/src/commonTest/kotlin/network/framework/AbstractRealNetworkHandlerTest.kt
@@ -24,6 +24,8 @@ import net.mamoe.mirai.internal.network.handler.NetworkHandler
 import net.mamoe.mirai.internal.network.handler.NetworkHandler.State
 import net.mamoe.mirai.internal.network.handler.NetworkHandlerContextImpl
 import net.mamoe.mirai.internal.network.handler.NetworkHandlerFactory
+import net.mamoe.mirai.internal.network.protocol.data.jce.SvcRespRegister
+import net.mamoe.mirai.internal.network.protocol.packet.login.StatSvc
 import net.mamoe.mirai.internal.test.AbstractTest
 import net.mamoe.mirai.internal.utils.subLogger
 import net.mamoe.mirai.utils.MiraiLogger
@@ -90,6 +92,12 @@ internal abstract class AbstractRealNetworkHandlerTest<H : NetworkHandler> : Abs
                 nhEvents.add(NHEvent.DoHeartbeatNow)
                 networkLogger.debug { "HeartbeatProcessor.doStatHeartbeatNow" }
             }
+
+            override suspend fun doRegisterNow(networkHandler: NetworkHandler): StatSvc.Register.Response {
+                nhEvents.add(NHEvent.DoHeartbeatNow)
+                networkLogger.debug { "HeartbeatProcessor.doRegisterNow" }
+                return StatSvc.Register.Response(SvcRespRegister())
+            }
         })
         set(KeyRefreshProcessor, object : KeyRefreshProcessor {
             override suspend fun keyRefreshLoop(handler: NetworkHandler) {}
diff --git a/mirai-core/src/commonTest/kotlin/network/framework/components/TestSsoProcessor.kt b/mirai-core/src/commonTest/kotlin/network/framework/components/TestSsoProcessor.kt
index 8879e89a0..393708a53 100644
--- a/mirai-core/src/commonTest/kotlin/network/framework/components/TestSsoProcessor.kt
+++ b/mirai-core/src/commonTest/kotlin/network/framework/components/TestSsoProcessor.kt
@@ -16,6 +16,7 @@ import net.mamoe.mirai.internal.network.context.SsoSession
 import net.mamoe.mirai.internal.network.handler.NetworkHandler
 import net.mamoe.mirai.internal.network.handler.logger
 import net.mamoe.mirai.internal.network.handler.state.StateObserver
+import net.mamoe.mirai.internal.network.protocol.data.jce.SvcRespRegister
 import net.mamoe.mirai.internal.network.protocol.packet.login.StatSvc
 import net.mamoe.mirai.utils.debug
 
@@ -33,4 +34,9 @@ internal open class TestSsoProcessor(private val bot: QQAndroidBot) : SsoProcess
     override suspend fun logout(handler: NetworkHandler) {
         bot.network.logger.debug { "SsoProcessor.logout" }
     }
+
+    override suspend fun sendRegister(handler: NetworkHandler): StatSvc.Register.Response {
+        bot.network.logger.debug { "SsoProcessor.sendRegister" }
+        return StatSvc.Register.Response(SvcRespRegister())
+    }
 }
\ No newline at end of file