mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2025-03-16 12:25:45 +01:00
Inprove coverage
This commit is contained in:
parent
c626044a30
commit
f99b2f8f3b
2 changed files with 22 additions and 5 deletions
|
@ -28,7 +28,7 @@ import functools
|
|||
sys.path.append('.')
|
||||
|
||||
from telegram import Message, User, Chat, MessageEntity
|
||||
from telegram.ext import Filters
|
||||
from telegram.ext import Filters, BaseFilter
|
||||
from tests.base import BaseTest
|
||||
|
||||
|
||||
|
@ -206,6 +206,24 @@ class FiltersTest(BaseTest, unittest.TestCase):
|
|||
self.assertTrue((Filters.text & (Filters.forwarded | Filters.entity(MessageEntity.MENTION))
|
||||
)(self.message))
|
||||
|
||||
self.assertRegex(
|
||||
str((Filters.text & (Filters.forwarded | Filters.entity(MessageEntity.MENTION)))),
|
||||
r"<telegram.ext.filters.MergedFilter consisting of "
|
||||
r"<telegram.ext.filters.Filters._Text object at .*?> and "
|
||||
r"<telegram.ext.filters.MergedFilter consisting of "
|
||||
r"<telegram.ext.filters.Filters._Forwarded object at .*?> or "
|
||||
r"<telegram.ext.filters.Filters.entity object at .*?>>>")
|
||||
|
||||
def test_faulty_custom_filter(self):
|
||||
|
||||
class _CustomFilter(BaseFilter):
|
||||
pass
|
||||
|
||||
custom = _CustomFilter()
|
||||
|
||||
with self.assertRaises(NotImplementedError):
|
||||
(custom & Filters.text)(self.message)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
|
|
@ -182,7 +182,7 @@ class UpdaterTest(BaseTest, unittest.TestCase):
|
|||
self._setup_updater('Test', edited=True)
|
||||
d = self.updater.dispatcher
|
||||
from telegram.ext import Filters
|
||||
handler = MessageHandler([Filters.text], self.telegramHandlerEditedTest, allow_edited=True)
|
||||
handler = MessageHandler(Filters.text, self.telegramHandlerEditedTest, allow_edited=True)
|
||||
d.addHandler(handler)
|
||||
self.updater.start_polling(0.01)
|
||||
sleep(.1)
|
||||
|
@ -190,8 +190,7 @@ class UpdaterTest(BaseTest, unittest.TestCase):
|
|||
|
||||
# Remove handler
|
||||
d.removeHandler(handler)
|
||||
handler = MessageHandler(
|
||||
[Filters.text], self.telegramHandlerEditedTest, allow_edited=False)
|
||||
handler = MessageHandler(Filters.text, self.telegramHandlerEditedTest, allow_edited=False)
|
||||
d.addHandler(handler)
|
||||
self.reset()
|
||||
|
||||
|
@ -201,7 +200,7 @@ class UpdaterTest(BaseTest, unittest.TestCase):
|
|||
|
||||
def test_addTelegramMessageHandlerMultipleMessages(self):
|
||||
self._setup_updater('Multiple', 100)
|
||||
self.updater.dispatcher.add_handler(MessageHandler([], self.telegramHandlerTest))
|
||||
self.updater.dispatcher.add_handler(MessageHandler(Filters.all, self.telegramHandlerTest))
|
||||
self.updater.start_polling(0.0)
|
||||
sleep(2)
|
||||
self.assertEqual(self.received_message, 'Multiple')
|
||||
|
|
Loading…
Add table
Reference in a new issue