test(mw/allow-chat): Add test from mw (#22)

src/middlewares/allowed-chat.test.js

@@ -0,0 +1,39 @@
+import test from 'ava'
+import { Context, next } from '../tests/telegraf'
+import { allowWhiteListChat } from './allowed-chat'
+
+/* eslint-disable no-param-reassign */
+
+test.beforeEach((t) => {
+  t.context.ctx = Context.create()
+})
+
+test('test public chat not in the white list', async (t) => {
+  t.context.ctx.chat.$type('public')
+  t.context.ctx.chat.$inWhiteList(false)
+
+  t.is(await allowWhiteListChat(t.context.ctx, next), null)
+})
+
+test('test public chat in the white list', async (t) => {
+  t.context.ctx.chat.$type('public')
+  t.context.ctx.chat.$inWhiteList(true)
+
+  try {
+    await allowWhiteListChat(t.context.ctx, next(() => t.pass()))
+  }
+  catch (error) {
+    t.fail()
+  }
+})
+
+test('test private chat', async (t) => {
+  t.context.ctx.chat.$type('private')
+
+  try {
+    await allowWhiteListChat(t.context.ctx, next(() => t.pass()))
+  }
+  catch (error) {
+    t.fail()
+  }
+})

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
@@ -24,6 +26,7 @@ class Chat {
   constructor() {
     this.id = ++ID_CHAT
     this.type = 'private'
+    this.inWhiteList = false
   }
 
   /**
@@ -35,6 +38,10 @@ class Chat {
     return this
   }
 
+  $inWhiteList(param) {
+    this.inWhiteList = param
+  }
+
   /**
    *
    * @param {string} title
@@ -75,10 +82,19 @@ class Context {
   get chat() {
     return this.message.chat
   }
+
+  isChatInWhiteList(chat) {
+    return chat.inWhiteList
+  }
+}
+
+function next(cb) {
+  return cb
 }
 
 module.exports = {
   Context,
   Message,
   Chat,
+  next,
 }