typings: fix mock plugin

koishijs · Jan 15, 2022 · 0368272 · 0368272
1 parent 3e932af
commit 0368272
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/plugins/mock/src/client.ts b/plugins/mock/src/client.ts
@@ -54,6 +54,7 @@ export class MessageClient {
         session.content = content
         this.app.emit(session as any, 'before-send', session)
         this.replies.push(content)
+        return []
       }
       const dispose = this.app.on('middleware', (session) => {
         if (session.id === uuid) process.nextTick(_resolve)

diff --git a/plugins/mock/src/webhook.ts b/plugins/mock/src/webhook.ts
@@ -64,11 +64,12 @@ export class Webhook {
         body += chunk
         return true
       }
-      res.end = (chunk: any) => {
-        res.write(chunk)
+      res.end = (callback: () => void) => {
         const code = res.statusCode
         const headers = res.getHeaders()
         resolve({ code, body, headers })
+        callback?.()
+        return res
       }
       this.app._httpServer.emit('request', req, res)
       req.emit('data', body)