fix: Fix mute dialog + test in test-middlewares

Botfuel · Jan 24, 2018 · d1d88d9 · d1d88d9
1 parent d0b7834
commit d1d88d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/test-middlewares/src/dialogs/mute-dialog.js b/packages/test-middlewares/src/dialogs/mute-dialog.js
@@ -1,8 +1,8 @@
 const { TextDialog } = require('botfuel-dialog');
 
 class Mute extends TextDialog {
-  dialogWillComplete(userId) {
-    this.brain.userSet(userId, '_isMuted', true);
+  dialogWillComplete(userMessage) {
+    this.brain.userSet(userMessage.user, '_isMuted', true);
   }
 }
 

diff --git a/packages/test-middlewares/tests/mute.test.js b/packages/test-middlewares/tests/mute.test.js
@@ -15,5 +15,7 @@ describe('Mute', function () {
       new BotTextMessage('Muted!'),
       new UserTextMessage('What\'s up'),
     ].map(msg => msg.toJson(userId)));
+    const isMuted = await bot.brain.userGet(userId, '_isMuted');
+    expect(isMuted).to.be(true);
   });
 });