implement some method in MockPuppet,including start() stop() logout() ding()

Author: redmaple1

wechaty-puppet-mock/src/main/kotlin/io/github/wechaty/MockData.kt

@@ -34,6 +34,13 @@ class MockData {
             return FileBox.fromUrl(faker.avatar().image(), null)
         }
 
+        fun getMessagePayload(fromId: String, toId: String): MessagePayload {
+            val messagePayload = getFakeMessagePayload()
+            messagePayload.fromId = fromId
+            messagePayload.toId = toId
+            return messagePayload
+        }
+
         fun getFakeMessagePayload(): MessagePayload {
             val messagePayload = MessagePayload(UUID.randomUUID().toString())
             messagePayload.fromId = UUID.randomUUID().toString()

wechaty-puppet-mock/src/main/kotlin/io/github/wechaty/MockPuppet.kt

@@ -1,12 +1,21 @@
 package io.github.wechaty
 
-import io.github.wechaty.Puppet
 import io.github.wechaty.filebox.FileBox
 import io.github.wechaty.schemas.*
 import org.slf4j.LoggerFactory
+import java.util.*
+import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
+import kotlin.concurrent.scheduleAtFixedRate
 
 class MockPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
+
+    private val timer: Timer
+
+    init {
+        timer = Timer()
+    }
+
     override fun start(): Future<Void> {
 
         log.info("MockPuppet start()")
@@ -29,20 +38,55 @@ class MockPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
 
 
         val userPayload = MockData.getFakeContactPayload()
-        TODO("Not yet implemented")
+        cacheContactPayload.put(userPayload.id, userPayload)
 
+        setId(userPayload.id)
+
+        emit("login", EventLoginPayload(userPayload.id))
+
+        timer.scheduleAtFixedRate(0, 5000) {
+            val fromContactPayload = MockData.getFakeContactPayload()
+            cacheContactPayload.put(fromContactPayload.id, fromContactPayload)
+            val messagePayload = MockData.getMessagePayload(fromContactPayload.id, userPayload.id)
+
+            cacheMessagePayload.put(messagePayload.id, messagePayload)
+            log.info("MockPuppet start() schedule pretending received a new message:${messagePayload.id}")
+            emit("message", EventMessagePayload(messagePayload.id))
+        }
+        return CompletableFuture.completedFuture(null)
     }
 
     override fun stop(): Future<Void> {
-        TODO("Not yet implemented")
+        log.info("MockPuppet stop()")
+        //TODO StateSwitch
+        /*
+        if (this.state.off()) {
+          log.warn('PuppetMock', 'stop() is called on a OFF puppet. await ready(off) and return.')
+          await this.state.ready('off')
+          return
+        }
+
+        this.state.off('pending')
+         */
+        timer.cancel()
+
+        //this.state.off(true)
+        return CompletableFuture.completedFuture(null)
     }
 
     override fun logout(): Future<Void> {
-        TODO("Not yet implemented")
+        log.info("MockPuppet logout()")
+        val id = getId() ?: throw Exception("logout before login?")
+
+        emit("logout", EventLogoutPayload(id, "test"))
+        setId(null)
+
+        return CompletableFuture.completedFuture(null)
     }
 
     override fun ding(data: String?) {
-        TODO("Not yet implemented")
+        log.info("MockPuppet ding($data?:'')")
+        emit("dong", EventDongPayload(data ?: ""))
     }
 
     override fun contactSelfName(name: String): Future<Void> {

wechaty-puppet/src/main/kotlin/Puppet.kt

@@ -44,12 +44,12 @@ abstract class Puppet : EventEmitter {
 
     private val executorService = Executors.newSingleThreadScheduledExecutor()
 
-    private lateinit var cacheContactPayload: Cache<String, ContactPayload>
-    private lateinit var cacheFriendshipPayload: Cache<String, FriendshipPayload>
-    private lateinit var cacheMessagePayload: Cache<String, MessagePayload>
-    private lateinit var cacheRoomPayload: Cache<String, RoomPayload>
-    private lateinit var cacheRoomMemberPayload: Cache<String, RoomMemberPayload>
-    private lateinit var cacheRoomInvitationPayload: Cache<String, RoomInvitationPayload>
+    protected lateinit var cacheContactPayload: Cache<String, ContactPayload>
+    protected lateinit var cacheFriendshipPayload: Cache<String, FriendshipPayload>
+    protected lateinit var cacheMessagePayload: Cache<String, MessagePayload>
+    protected lateinit var cacheRoomPayload: Cache<String, RoomPayload>
+    protected lateinit var cacheRoomMemberPayload: Cache<String, RoomMemberPayload>
+    protected lateinit var cacheRoomInvitationPayload: Cache<String, RoomInvitationPayload>
     private val count = AtomicLong()
     private var id: String? = null
     protected var puppetOptions: PuppetOptions? = null