mirror of
https://github.com/python-telegram-bot/python-telegram-bot.git
synced 2025-03-13 11:18:20 +01:00
This commit is contained in:
parent
25e5449e97
commit
5c45e469d5
2 changed files with 22 additions and 12 deletions
|
@ -209,10 +209,11 @@ class Message(TelegramObject):
|
|||
ATTACHMENT_TYPES = ['audio', 'game', 'animation', 'document', 'photo', 'sticker', 'video',
|
||||
'voice', 'video_note', 'contact', 'location', 'venue', 'invoice',
|
||||
'successful_payment']
|
||||
MESSAGE_TYPES = ['text', 'new_chat_members', 'new_chat_title', 'new_chat_photo',
|
||||
'delete_chat_photo', 'group_chat_created', 'supergroup_chat_created',
|
||||
'channel_chat_created', 'migrate_to_chat_id', 'migrate_from_chat_id',
|
||||
'pinned_message', 'passport_data'] + ATTACHMENT_TYPES
|
||||
MESSAGE_TYPES = ['text', 'new_chat_members', 'left_chat_member', 'new_chat_title',
|
||||
'new_chat_photo', 'delete_chat_photo', 'group_chat_created',
|
||||
'supergroup_chat_created', 'channel_chat_created', 'migrate_to_chat_id',
|
||||
'migrate_from_chat_id', 'pinned_message',
|
||||
'passport_data'] + ATTACHMENT_TYPES
|
||||
|
||||
def __init__(self,
|
||||
message_id,
|
||||
|
|
|
@ -32,24 +32,33 @@ class TestHelpers(object):
|
|||
assert expected_str == helpers.escape_markdown(test_str)
|
||||
|
||||
def test_effective_message_type(self):
|
||||
test_message = Message(message_id=1,
|
||||
from_user=None,
|
||||
date=None,
|
||||
chat=None)
|
||||
|
||||
test_message.text = 'Test'
|
||||
def build_test_message(**kwargs):
|
||||
config = dict(
|
||||
message_id=1,
|
||||
from_user=None,
|
||||
date=None,
|
||||
chat=None,
|
||||
)
|
||||
config.update(**kwargs)
|
||||
return Message(**config)
|
||||
|
||||
test_message = build_test_message(text='Test')
|
||||
assert helpers.effective_message_type(test_message) == 'text'
|
||||
test_message.text = None
|
||||
|
||||
test_message.sticker = Sticker('sticker_id', 50, 50)
|
||||
test_message = build_test_message(sticker=Sticker('sticker_id', 50, 50))
|
||||
assert helpers.effective_message_type(test_message) == 'sticker'
|
||||
test_message.sticker = None
|
||||
|
||||
test_message.new_chat_members = [User(55, 'new_user', False)]
|
||||
test_message = build_test_message(new_chat_members=[User(55, 'new_user', False)])
|
||||
assert helpers.effective_message_type(test_message) == 'new_chat_members'
|
||||
|
||||
test_message = build_test_message(left_chat_member=[User(55, 'new_user', False)])
|
||||
assert helpers.effective_message_type(test_message) == 'left_chat_member'
|
||||
|
||||
test_update = Update(1)
|
||||
test_message.text = 'Test'
|
||||
test_message = build_test_message(text='Test')
|
||||
test_update.message = test_message
|
||||
assert helpers.effective_message_type(test_update) == 'text'
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue