test(middlewares/allowed-chat): Add test for allowWhiteListChat (#22)

src/middlewares/allowed-chat.js

@@ -3,13 +3,12 @@ const debug = require('debug')('rubot:middlewares:allowed-chat')
 
 const allowWhiteListChat = async ({ chat, isChatInWhiteList }, next) => {
   debug('allowWhiteListChat', chat.id, chat.type)
-  if (chat.type !== 'private') {
-    if (!isChatInWhiteList(chat)) {
-      return null
-    }
+
+  if (isChatInWhiteList(chat)) {
+    return next()
   }
 
-  return next()
+  return null
 }
 
 module.exports = {

src/middlewares/allowed-chat.test.js

@@ -0,0 +1,37 @@
+import test from 'ava'
+import sinon from 'sinon'
+import { Context } from '../tests/telegraf'
+import { allowWhiteListChat } from './allowed-chat'
+
+/* eslint-disable no-param-reassign */
+
+test.beforeEach((t) => {
+  t.context.create = ({ inWhitelist }) => {
+    const ctx = Context.create()
+    const res = Math.random()
+    const next = sinon.stub().returns(res)
+
+    ctx.isChatInWhiteList = sinon.stub().returns(inWhitelist)
+
+    return { ctx, res, next }
+  }
+})
+
+test('should pass in whitelist', async (t) => {
+  const { ctx, res, next } = t.context.create({ inWhitelist: true })
+  const result = await allowWhiteListChat(ctx, next)
+
+  t.true(next.called)
+  t.is(result, res)
+  t.true(ctx.isChatInWhiteList.calledWith(ctx.chat))
+})
+
+test('should stop not in whitelist', async (t) => {
+  const { ctx, next } = t.context.create({ inWhitelist: false })
+  const result = await allowWhiteListChat(ctx, next)
+
+  t.false(next.called)
+  t.is(result, null)
+  t.true(ctx.isChatInWhiteList.calledWith(ctx.chat))
+})
+

src/tests/telegraf.js

@@ -10,6 +10,8 @@ let ID_USER = ID_USER_START
 let ID_MESSAGE = ID_MESSAGE_START
 let ID_CHAT = ID_CHAT_START
 
+/* eslint-disable class-methods-use-this */
+
 class User {
   constructor() {
     this.id = ++ID_USER
@@ -77,6 +79,7 @@ class Context {
   }
 }
 
+
 module.exports = {
   Context,
   Message,