Add Filters.entities test.

jsmnbom · jsmnbom · commit 7ab007d8d48f · 2016-09-13T20:47:43.000+02:00
diff --git a/tests/test_filters.py b/tests/test_filters.py
@@ -23,10 +23,11 @@
 import sys
 import unittest
 from datetime import datetime
+import functools
 
 sys.path.append('.')
 
-from telegram import Message, User, Chat
+from telegram import Message, User, Chat, MessageEntity
 from telegram.ext import Filters
 from tests.base import BaseTest
 
@@ -150,6 +151,27 @@ def test_filters_status_update(self):
         self.assertTrue(Filters.status_update(self.message))
         self.message.pinned_message = None
 
+    def test_entities_filter(self):
+        e = functools.partial(MessageEntity, offset=0, length=0)
+
+        self.message.entities = [e(MessageEntity.MENTION)]
+        self.assertTrue(Filters.entities([MessageEntity.MENTION])(self.message))
+
+        self.message.entities = []
+        self.assertFalse(Filters.entities([MessageEntity.MENTION])(self.message))
+
+        self.message.entities = [e(MessageEntity.BOLD)]
+        self.assertFalse(Filters.entities([MessageEntity.MENTION])(self.message))
+
+        self.message.entities = [e(MessageEntity.MENTION)]
+        self.assertTrue(
+            Filters.entities([MessageEntity.MENTION, MessageEntity.BOLD])(self.message))
+        self.message.entities = [e(MessageEntity.BOLD)]
+        self.assertTrue(
+            Filters.entities([MessageEntity.MENTION, MessageEntity.BOLD])(self.message))
+        self.assertFalse(
+            Filters.entities([MessageEntity.MENTION, MessageEntity.TEXT_MENTION])(self.message))
+
 
 if __name__ == '__main__':
     unittest.main()
