#!/usr/bin/env python # # A library that provides a Python interface to the Telegram Bot API # Copyright (C) 2015-2016 # Leandro Toledo de Souza # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see [http://www.gnu.org/licenses/]. """ This module contains a object that represents Tests for MessageHandler.Filters """ import sys import unittest from datetime import datetime sys.path.append('.') from telegram import Message, User, Chat from telegram.ext import Filters from tests.base import BaseTest class FiltersTest(BaseTest, unittest.TestCase): """This object represents Tests for MessageHandler.Filters""" def setUp(self): self.message = Message(0, User(0, "Testuser"), datetime.now(), Chat(0, 'private')) def test_filters_text(self): self.message.text = 'test' self.assertTrue(Filters.text(self.message)) self.message.text = '/test' self.assertFalse(Filters.text(self.message)) def test_filters_command(self): self.message.text = 'test' self.assertFalse(Filters.command(self.message)) self.message.text = '/test' self.assertTrue(Filters.command(self.message)) def test_filters_audio(self): self.message.audio = 'test' self.assertTrue(Filters.audio(self.message)) self.message.audio = None self.assertFalse(Filters.audio(self.message)) def test_filters_document(self): self.message.document = 'test' self.assertTrue(Filters.document(self.message)) self.message.document = None self.assertFalse(Filters.document(self.message)) def test_filters_photo(self): self.message.photo = 'test' self.assertTrue(Filters.photo(self.message)) self.message.photo = None self.assertFalse(Filters.photo(self.message)) def test_filters_sticker(self): self.message.sticker = 'test' self.assertTrue(Filters.sticker(self.message)) self.message.sticker = None self.assertFalse(Filters.sticker(self.message)) def test_filters_video(self): self.message.video = 'test' self.assertTrue(Filters.video(self.message)) self.message.video = None self.assertFalse(Filters.video(self.message)) def test_filters_voice(self): self.message.voice = 'test' self.assertTrue(Filters.voice(self.message)) self.message.voice = None self.assertFalse(Filters.voice(self.message)) def test_filters_contact(self): self.message.contact = 'test' self.assertTrue(Filters.contact(self.message)) self.message.contact = None self.assertFalse(Filters.contact(self.message)) def test_filters_location(self): self.message.location = 'test' self.assertTrue(Filters.location(self.message)) self.message.location = None self.assertFalse(Filters.location(self.message)) def test_filters_venue(self): self.message.venue = 'test' self.assertTrue(Filters.venue(self.message)) self.message.venue = None self.assertFalse(Filters.venue(self.message)) def test_filters_status_update(self): self.assertFalse(Filters.status_update(self.message)) self.message.new_chat_member = 'test' self.assertTrue(Filters.status_update(self.message)) self.message.new_chat_member = None self.message.left_chat_member = 'test' self.assertTrue(Filters.status_update(self.message)) self.message.left_chat_member = None self.message.new_chat_title = 'test' self.assertTrue(Filters.status_update(self.message)) self.message.new_chat_title = '' self.message.new_chat_photo = 'test' self.assertTrue(Filters.status_update(self.message)) self.message.new_chat_photo = None self.message.delete_chat_photo = True self.assertTrue(Filters.status_update(self.message)) self.message.delete_chat_photo = False self.message.group_chat_created = True self.assertTrue(Filters.status_update(self.message)) self.message.group_chat_created = False self.message.supergroup_chat_created = True self.assertTrue(Filters.status_update(self.message)) self.message.supergroup_chat_created = False self.message.migrate_to_chat_id = 100 self.assertTrue(Filters.status_update(self.message)) self.message.migrate_to_chat_id = 0 self.message.migrate_from_chat_id = 100 self.assertTrue(Filters.status_update(self.message)) self.message.migrate_from_chat_id = 0 self.message.channel_chat_created = True self.assertTrue(Filters.status_update(self.message)) self.message.channel_chat_created = False self.message.pinned_message = 'test' self.assertTrue(Filters.status_update(self.message)) self.message.pinned_message = None if __name__ == '__main__': unittest.main()