Add addOtherClient for GroupExtensions for tests, add other clients…

… before running tests for MessageSyncEvent
mamoe · Jun 13, 2022 · 0a8fc34 · 0a8fc34
1 parent 0332395
commit 0a8fc34
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/mirai-core/src/commonTest/kotlin/notice/processors/AbstractNoticeProcessorTest.kt b/mirai-core/src/commonTest/kotlin/notice/processors/AbstractNoticeProcessorTest.kt
@@ -16,10 +16,8 @@ import net.mamoe.mirai.Mirai
 import net.mamoe.mirai.contact.*
 import net.mamoe.mirai.internal.BotAccount
 import net.mamoe.mirai.internal.QQAndroidBot
-import net.mamoe.mirai.internal.contact.FriendImpl
-import net.mamoe.mirai.internal.contact.GroupImpl
-import net.mamoe.mirai.internal.contact.NormalMemberImpl
-import net.mamoe.mirai.internal.contact.StrangerImpl
+import net.mamoe.mirai.internal.asQQAndroidBot
+import net.mamoe.mirai.internal.contact.*
 import net.mamoe.mirai.internal.contact.info.FriendInfoImpl
 import net.mamoe.mirai.internal.contact.info.GroupInfoImpl
 import net.mamoe.mirai.internal.contact.info.MemberInfoImpl
@@ -211,4 +209,15 @@ internal interface GroupExtensions {
         members.delegate.add(NormalMemberImpl(this.cast(), this.coroutineContext, info))
         return this
     }
+
+    fun Bot.addOtherClient(
+        appId: Int,
+        platform: Platform = Platform.IOS,
+        deviceName: String = "my device $appId",
+        deviceKind: String = "my device kind $appId"
+    ): OtherClient {
+        return bot.asQQAndroidBot().createOtherClient(OtherClientInfo(appId, platform, deviceName, deviceKind)).also {
+            this.otherClients.delegate.add(it)
+        }
+    }
 }
diff --git a/mirai-core/src/commonTest/kotlin/notice/processors/MessageSyncTest.kt b/mirai-core/src/commonTest/kotlin/notice/processors/MessageSyncTest.kt
@@ -96,6 +96,7 @@ internal class MessageSyncTest : AbstractNoticeProcessorTest() {
                 addMember(1230001, permission = MemberPermission.OWNER)
                 addMember(1230002, permission = MemberPermission.MEMBER)
             }
+            addOtherClient(537067835)
         }
 
         runTest().toList().run {
@@ -162,6 +163,7 @@ internal class MessageSyncTest : AbstractNoticeProcessorTest() {
 
         setBot(1230002).apply {
             addFriend(1230001)
+            addOtherClient(537067835)
         }
 
         runTest().toList().run {