Add game filter

This commit is contained in:
Jacob Bom 2016-10-07 22:37:29 +02:00
parent ade89772f4
commit b7c7612b3f
2 changed files with 10 additions and 0 deletions

View file

@ -73,6 +73,10 @@ class Filters(object):
def venue(message):
return bool(message.venue)
@staticmethod
def game(message):
return bool(message.game)
@staticmethod
def status_update(message):
return bool(message.new_chat_member or message.left_chat_member or message.new_chat_title

View file

@ -104,6 +104,12 @@ class FiltersTest(BaseTest, unittest.TestCase):
self.message.venue = None
self.assertFalse(Filters.venue(self.message))
def test_filters_game(self):
self.message.game = 'test'
self.assertTrue(Filters.game(self.message))
self.message.game = None
self.assertFalse(Filters.game(self.message))
def test_filters_status_update(self):
self.assertFalse(Filters.status_update(self.message))