2016-05-15 02:39:11 +02:00
|
|
|
#!/usr/bin/env python
|
2015-09-07 19:10:57 +02:00
|
|
|
#
|
|
|
|
# A library that provides a Python interface to the Telegram Bot API
|
2021-01-03 06:10:24 +01:00
|
|
|
# Copyright (C) 2015-2021
|
2016-01-05 14:12:03 +01:00
|
|
|
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
|
2015-09-07 19:10:57 +02:00
|
|
|
#
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
2017-08-11 23:58:41 +02:00
|
|
|
# it under the terms of the GNU Lesser Public License as published by
|
2015-09-07 19:10:57 +02:00
|
|
|
# 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
|
2017-08-11 23:58:41 +02:00
|
|
|
# GNU Lesser Public License for more details.
|
2015-09-07 19:10:57 +02:00
|
|
|
#
|
2017-08-11 23:58:41 +02:00
|
|
|
# You should have received a copy of the GNU Lesser Public License
|
2015-09-07 19:10:57 +02:00
|
|
|
# along with this program. If not, see [http://www.gnu.org/licenses/].
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
import pytest
|
2016-09-20 06:36:55 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
from telegram import Chat, ChatAction, ChatPermissions, ChatLocation, Location, Bot
|
2020-07-19 17:47:26 +02:00
|
|
|
from telegram import User
|
2021-02-19 19:07:48 +01:00
|
|
|
from tests.conftest import check_shortcut_signature, check_shortcut_call, check_defaults_handling
|
2016-09-20 06:36:55 +02:00
|
|
|
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
@pytest.fixture(scope='class')
|
|
|
|
def chat(bot):
|
2020-10-09 17:22:07 +02:00
|
|
|
return Chat(
|
|
|
|
TestChat.id_,
|
|
|
|
TestChat.title,
|
|
|
|
TestChat.type_,
|
|
|
|
username=TestChat.username,
|
|
|
|
all_members_are_administrators=TestChat.all_members_are_administrators,
|
|
|
|
bot=bot,
|
|
|
|
sticker_set_name=TestChat.sticker_set_name,
|
|
|
|
can_set_sticker_set=TestChat.can_set_sticker_set,
|
|
|
|
permissions=TestChat.permissions,
|
|
|
|
slow_mode_delay=TestChat.slow_mode_delay,
|
2021-03-14 16:41:35 +01:00
|
|
|
message_auto_delete_time=TestChat.message_auto_delete_time,
|
2020-11-29 16:20:46 +01:00
|
|
|
bio=TestChat.bio,
|
|
|
|
linked_chat_id=TestChat.linked_chat_id,
|
|
|
|
location=TestChat.location,
|
2021-12-11 15:21:56 +01:00
|
|
|
has_private_forwards=True,
|
|
|
|
has_protected_content=True,
|
2020-10-09 17:22:07 +02:00
|
|
|
)
|
2015-09-07 19:10:57 +02:00
|
|
|
|
|
|
|
|
2020-06-15 18:20:51 +02:00
|
|
|
class TestChat:
|
2020-02-23 22:03:58 +01:00
|
|
|
id_ = -28767330
|
2017-08-11 23:58:41 +02:00
|
|
|
title = 'ToledosPalaceBot - Group'
|
2020-02-23 22:03:58 +01:00
|
|
|
type_ = 'group'
|
2018-05-09 11:42:12 +02:00
|
|
|
username = 'username'
|
2017-08-11 23:58:41 +02:00
|
|
|
all_members_are_administrators = False
|
2017-10-14 20:03:02 +02:00
|
|
|
sticker_set_name = 'stickers'
|
|
|
|
can_set_sticker_set = False
|
2019-09-06 21:41:43 +02:00
|
|
|
permissions = ChatPermissions(
|
|
|
|
can_send_messages=True,
|
|
|
|
can_change_info=False,
|
|
|
|
can_invite_users=True,
|
|
|
|
)
|
2020-03-28 16:37:26 +01:00
|
|
|
slow_mode_delay = 30
|
2021-03-14 16:41:35 +01:00
|
|
|
message_auto_delete_time = 42
|
2020-11-29 16:20:46 +01:00
|
|
|
bio = "I'm a Barbie Girl in a Barbie World"
|
|
|
|
linked_chat_id = 11880
|
|
|
|
location = ChatLocation(Location(123, 456), 'Barbie World')
|
2021-12-11 15:21:56 +01:00
|
|
|
has_protected_content = True
|
|
|
|
has_private_forwards = True
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2021-05-29 16:18:16 +02:00
|
|
|
def test_slot_behaviour(self, chat, recwarn, mro_slots):
|
|
|
|
for attr in chat.__slots__:
|
|
|
|
assert getattr(chat, attr, 'err') != 'err', f"got extra slot '{attr}'"
|
|
|
|
assert not chat.__dict__, f"got missing slot(s): {chat.__dict__}"
|
|
|
|
assert len(mro_slots(chat)) == len(set(mro_slots(chat))), "duplicate slot"
|
|
|
|
chat.custom, chat.id = 'should give warning', self.id_
|
|
|
|
assert len(recwarn) == 1 and 'custom' in str(recwarn[0].message), recwarn.list
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_de_json(self, bot):
|
|
|
|
json_dict = {
|
2020-02-23 22:03:58 +01:00
|
|
|
'id': self.id_,
|
2017-10-14 20:03:02 +02:00
|
|
|
'title': self.title,
|
2020-02-23 22:03:58 +01:00
|
|
|
'type': self.type_,
|
2018-05-09 11:42:12 +02:00
|
|
|
'username': self.username,
|
2017-10-14 20:03:02 +02:00
|
|
|
'all_members_are_administrators': self.all_members_are_administrators,
|
|
|
|
'sticker_set_name': self.sticker_set_name,
|
2019-09-06 21:41:43 +02:00
|
|
|
'can_set_sticker_set': self.can_set_sticker_set,
|
2020-03-28 16:37:26 +01:00
|
|
|
'permissions': self.permissions.to_dict(),
|
2020-10-09 17:22:07 +02:00
|
|
|
'slow_mode_delay': self.slow_mode_delay,
|
2021-03-14 16:41:35 +01:00
|
|
|
'message_auto_delete_time': self.message_auto_delete_time,
|
2020-11-29 16:20:46 +01:00
|
|
|
'bio': self.bio,
|
2021-12-11 15:21:56 +01:00
|
|
|
'has_protected_content': self.has_protected_content,
|
|
|
|
'has_private_forwards': self.has_private_forwards,
|
2020-11-29 16:20:46 +01:00
|
|
|
'linked_chat_id': self.linked_chat_id,
|
|
|
|
'location': self.location.to_dict(),
|
2016-10-12 22:56:57 +02:00
|
|
|
}
|
2017-08-11 23:58:41 +02:00
|
|
|
chat = Chat.de_json(json_dict, bot)
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2020-02-23 22:03:58 +01:00
|
|
|
assert chat.id == self.id_
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.title == self.title
|
2020-02-23 22:03:58 +01:00
|
|
|
assert chat.type == self.type_
|
2018-05-09 11:42:12 +02:00
|
|
|
assert chat.username == self.username
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.all_members_are_administrators == self.all_members_are_administrators
|
2017-10-14 20:03:02 +02:00
|
|
|
assert chat.sticker_set_name == self.sticker_set_name
|
|
|
|
assert chat.can_set_sticker_set == self.can_set_sticker_set
|
2019-09-06 21:41:43 +02:00
|
|
|
assert chat.permissions == self.permissions
|
2020-03-28 16:37:26 +01:00
|
|
|
assert chat.slow_mode_delay == self.slow_mode_delay
|
2021-03-14 16:41:35 +01:00
|
|
|
assert chat.message_auto_delete_time == self.message_auto_delete_time
|
2020-11-29 16:20:46 +01:00
|
|
|
assert chat.bio == self.bio
|
2021-12-11 15:21:56 +01:00
|
|
|
assert chat.has_protected_content == self.has_protected_content
|
|
|
|
assert chat.has_private_forwards == self.has_private_forwards
|
2020-11-29 16:20:46 +01:00
|
|
|
assert chat.linked_chat_id == self.linked_chat_id
|
|
|
|
assert chat.location.location == self.location.location
|
|
|
|
assert chat.location.address == self.location.address
|
2015-09-08 01:11:02 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_to_dict(self, chat):
|
|
|
|
chat_dict = chat.to_dict()
|
2015-09-08 01:11:02 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert isinstance(chat_dict, dict)
|
|
|
|
assert chat_dict['id'] == chat.id
|
|
|
|
assert chat_dict['title'] == chat.title
|
|
|
|
assert chat_dict['type'] == chat.type
|
2018-05-09 11:42:12 +02:00
|
|
|
assert chat_dict['username'] == chat.username
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat_dict['all_members_are_administrators'] == chat.all_members_are_administrators
|
2019-09-06 21:41:43 +02:00
|
|
|
assert chat_dict['permissions'] == chat.permissions.to_dict()
|
2020-03-28 16:37:26 +01:00
|
|
|
assert chat_dict['slow_mode_delay'] == chat.slow_mode_delay
|
2021-03-14 16:41:35 +01:00
|
|
|
assert chat_dict['message_auto_delete_time'] == chat.message_auto_delete_time
|
2020-11-29 16:20:46 +01:00
|
|
|
assert chat_dict['bio'] == chat.bio
|
2021-12-11 15:21:56 +01:00
|
|
|
assert chat_dict['has_private_forwards'] == chat.has_private_forwards
|
|
|
|
assert chat_dict['has_protected_content'] == chat.has_protected_content
|
2020-11-29 16:20:46 +01:00
|
|
|
assert chat_dict['linked_chat_id'] == chat.linked_chat_id
|
|
|
|
assert chat_dict['location'] == chat.location.to_dict()
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2018-05-09 11:42:12 +02:00
|
|
|
def test_link(self, chat):
|
2020-11-23 22:09:29 +01:00
|
|
|
assert chat.link == f'https://t.me/{chat.username}'
|
2018-05-09 11:42:12 +02:00
|
|
|
chat.username = None
|
|
|
|
assert chat.link is None
|
|
|
|
|
2021-01-09 17:48:56 +01:00
|
|
|
def test_full_name(self):
|
|
|
|
chat = Chat(
|
2021-01-17 23:24:20 +01:00
|
|
|
id=1, type=Chat.PRIVATE, first_name='first\u2022name', last_name='last\u2022name'
|
2021-01-09 17:48:56 +01:00
|
|
|
)
|
2021-01-17 23:24:20 +01:00
|
|
|
assert chat.full_name == 'first\u2022name last\u2022name'
|
|
|
|
chat = Chat(id=1, type=Chat.PRIVATE, first_name='first\u2022name')
|
|
|
|
assert chat.full_name == 'first\u2022name'
|
2021-01-09 17:48:56 +01:00
|
|
|
chat = Chat(
|
|
|
|
id=1,
|
|
|
|
type=Chat.PRIVATE,
|
|
|
|
)
|
|
|
|
assert chat.full_name is None
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_send_action(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
id_ = kwargs['chat_id'] == chat.id
|
2017-08-11 23:58:41 +02:00
|
|
|
action = kwargs['action'] == ChatAction.TYPING
|
2021-02-19 19:07:48 +01:00
|
|
|
return id_ and action
|
2020-12-30 13:41:07 +01:00
|
|
|
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_signature(chat.send_action, Bot.send_chat_action, ['chat_id'], [])
|
|
|
|
assert check_shortcut_call(chat.send_action, chat.bot, 'send_chat_action')
|
|
|
|
assert check_defaults_handling(chat.send_action, chat.bot)
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_chat_action', make_assertion)
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.send_action(action=ChatAction.TYPING)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert chat.send_action(action=ChatAction.TYPING)
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_leave(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.leave, Bot.leave_chat, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.leave, chat.bot, 'leave_chat')
|
|
|
|
assert check_defaults_handling(chat.leave, chat.bot)
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'leave_chat', make_assertion)
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.leave()
|
2015-09-07 19:10:57 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_get_administrators(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.get_administrators, Bot.get_chat_administrators, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.get_administrators, chat.bot, 'get_chat_administrators')
|
|
|
|
assert check_defaults_handling(chat.get_administrators, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'get_chat_administrators', make_assertion)
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.get_administrators()
|
2016-09-20 06:36:55 +02:00
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
def test_get_member_count(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
2021-07-01 17:45:19 +02:00
|
|
|
Chat.get_member_count, Bot.get_chat_member_count, ['chat_id'], []
|
2020-12-30 13:41:07 +01:00
|
|
|
)
|
2021-07-01 17:45:19 +02:00
|
|
|
assert check_shortcut_call(chat.get_member_count, chat.bot, 'get_chat_member_count')
|
|
|
|
assert check_defaults_handling(chat.get_member_count, chat.bot)
|
2016-09-20 06:36:55 +02:00
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
monkeypatch.setattr(chat.bot, 'get_chat_member_count', make_assertion)
|
|
|
|
assert chat.get_member_count()
|
|
|
|
|
|
|
|
def test_get_members_count_warning(self, chat, monkeypatch, recwarn):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'get_chat_member_count', make_assertion)
|
2017-08-11 23:58:41 +02:00
|
|
|
assert chat.get_members_count()
|
2021-07-01 17:45:19 +02:00
|
|
|
assert len(recwarn) == 1
|
|
|
|
assert '`Chat.get_members_count` is deprecated' in str(recwarn[0].message)
|
2016-09-20 06:36:55 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_get_member(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and user_id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.get_member, Bot.get_chat_member, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.get_member, chat.bot, 'get_chat_member')
|
|
|
|
assert check_defaults_handling(chat.get_member, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'get_chat_member', make_assertion)
|
|
|
|
assert chat.get_member(user_id=42)
|
2017-08-11 23:58:41 +02:00
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
def test_ban_member(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
2017-08-11 23:58:41 +02:00
|
|
|
until = kwargs['until_date'] == 43
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and user_id and until
|
2020-12-30 13:41:07 +01:00
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
assert check_shortcut_signature(Chat.ban_member, Bot.ban_chat_member, ['chat_id'], [])
|
|
|
|
assert check_shortcut_call(chat.ban_member, chat.bot, 'ban_chat_member')
|
|
|
|
assert check_defaults_handling(chat.ban_member, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'ban_chat_member', make_assertion)
|
|
|
|
assert chat.ban_member(user_id=42, until_date=43)
|
|
|
|
|
2021-12-11 15:21:56 +01:00
|
|
|
def test_ban_sender_chat(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
sender_chat_id = kwargs['sender_chat_id'] == 42
|
|
|
|
return chat_id and sender_chat_id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.ban_sender_chat, Bot.ban_chat_sender_chat, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.ban_sender_chat, chat.bot, 'ban_chat_sender_chat')
|
|
|
|
assert check_defaults_handling(chat.ban_sender_chat, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'ban_chat_sender_chat', make_assertion)
|
|
|
|
assert chat.ban_sender_chat(42)
|
|
|
|
|
|
|
|
def test_ban_chat(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == 42
|
|
|
|
sender_chat_id = kwargs['sender_chat_id'] == chat.id
|
|
|
|
return chat_id and sender_chat_id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.ban_chat, Bot.ban_chat_sender_chat, ['sender_chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.ban_chat, chat.bot, 'ban_chat_sender_chat')
|
|
|
|
assert check_defaults_handling(chat.ban_chat, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'ban_chat_sender_chat', make_assertion)
|
|
|
|
assert chat.ban_chat(42)
|
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
def test_kick_member_warning(self, chat, monkeypatch, recwarn):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
|
|
|
until = kwargs['until_date'] == 43
|
|
|
|
return chat_id and user_id and until
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2021-07-01 17:45:19 +02:00
|
|
|
monkeypatch.setattr(chat.bot, 'ban_chat_member', make_assertion)
|
2020-12-30 13:41:07 +01:00
|
|
|
assert chat.kick_member(user_id=42, until_date=43)
|
2021-07-01 17:45:19 +02:00
|
|
|
assert len(recwarn) == 1
|
|
|
|
assert '`Chat.kick_member` is deprecated' in str(recwarn[0].message)
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2020-11-29 16:20:46 +01:00
|
|
|
@pytest.mark.parametrize('only_if_banned', [True, False, None])
|
|
|
|
def test_unban_member(self, monkeypatch, chat, only_if_banned):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
2021-04-05 13:25:27 +02:00
|
|
|
o_i_b = kwargs.get('only_if_banned') == only_if_banned
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and user_id and o_i_b
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.unban_member, Bot.unban_chat_member, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.unban_member, chat.bot, 'unban_chat_member')
|
|
|
|
assert check_defaults_handling(chat.unban_member, chat.bot)
|
2017-08-11 23:58:41 +02:00
|
|
|
|
2020-11-29 16:20:46 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'unban_chat_member', make_assertion)
|
2020-12-30 13:41:07 +01:00
|
|
|
assert chat.unban_member(user_id=42, only_if_banned=only_if_banned)
|
2017-08-11 23:58:41 +02:00
|
|
|
|
2021-12-11 15:21:56 +01:00
|
|
|
def test_unban_sender_chat(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
sender_chat_id = kwargs['sender_chat_id'] == 42
|
|
|
|
return chat_id and sender_chat_id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.unban_sender_chat, Bot.unban_chat_sender_chat, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.unban_sender_chat, chat.bot, 'unban_chat_sender_chat')
|
|
|
|
assert check_defaults_handling(chat.unban_sender_chat, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'unban_chat_sender_chat', make_assertion)
|
|
|
|
assert chat.unban_sender_chat(42)
|
|
|
|
|
|
|
|
def test_unban_chat(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == 42
|
|
|
|
sender_chat_id = kwargs['sender_chat_id'] == chat.id
|
|
|
|
return chat_id and sender_chat_id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.unban_chat, Bot.ban_chat_sender_chat, ['sender_chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.unban_chat, chat.bot, 'unban_chat_sender_chat')
|
|
|
|
assert check_defaults_handling(chat.unban_chat, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'unban_chat_sender_chat', make_assertion)
|
|
|
|
assert chat.unban_chat(42)
|
|
|
|
|
2021-01-09 17:48:56 +01:00
|
|
|
@pytest.mark.parametrize('is_anonymous', [True, False, None])
|
|
|
|
def test_promote_member(self, monkeypatch, chat, is_anonymous):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
2021-04-05 13:25:27 +02:00
|
|
|
o_i_b = kwargs.get('is_anonymous') == is_anonymous
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and user_id and o_i_b
|
2021-01-09 17:48:56 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.promote_member, Bot.promote_chat_member, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.promote_member, chat.bot, 'promote_chat_member')
|
|
|
|
assert check_defaults_handling(chat.promote_member, chat.bot)
|
2021-01-09 17:48:56 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'promote_chat_member', make_assertion)
|
|
|
|
assert chat.promote_member(user_id=42, is_anonymous=is_anonymous)
|
|
|
|
|
|
|
|
def test_restrict_member(self, monkeypatch, chat):
|
|
|
|
permissions = ChatPermissions(True, False, True, False, True, False, True, False)
|
|
|
|
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
2021-04-05 13:25:27 +02:00
|
|
|
o_i_b = kwargs.get('permissions') == permissions
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and user_id and o_i_b
|
2021-01-09 17:48:56 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.restrict_member, Bot.restrict_chat_member, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.restrict_member, chat.bot, 'restrict_chat_member')
|
|
|
|
assert check_defaults_handling(chat.restrict_member, chat.bot)
|
2021-01-09 17:48:56 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'restrict_chat_member', make_assertion)
|
|
|
|
assert chat.restrict_member(user_id=42, permissions=permissions)
|
|
|
|
|
2019-09-06 21:41:43 +02:00
|
|
|
def test_set_permissions(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
permissions = kwargs['permissions'] == self.permissions
|
2021-02-19 19:07:48 +01:00
|
|
|
return chat_id and permissions
|
2019-09-06 21:41:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.set_permissions, Bot.set_chat_permissions, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.set_permissions, chat.bot, 'set_chat_permissions')
|
|
|
|
assert check_defaults_handling(chat.set_permissions, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'set_chat_permissions', make_assertion)
|
|
|
|
assert chat.set_permissions(permissions=self.permissions)
|
2019-09-06 21:41:43 +02:00
|
|
|
|
2020-03-28 16:37:26 +01:00
|
|
|
def test_set_administrator_custom_title(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
|
|
|
user_id = kwargs['user_id'] == 42
|
|
|
|
custom_title = kwargs['custom_title'] == 'custom_title'
|
2020-03-28 16:37:26 +01:00
|
|
|
return chat_id and user_id and custom_title
|
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr('telegram.Bot.set_chat_administrator_custom_title', make_assertion)
|
|
|
|
assert chat.set_administrator_custom_title(user_id=42, custom_title='custom_title')
|
2020-03-28 16:37:26 +01:00
|
|
|
|
2020-11-29 16:20:46 +01:00
|
|
|
def test_pin_message(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['message_id'] == 42
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.pin_message, Bot.pin_chat_message, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.pin_message, chat.bot, 'pin_chat_message')
|
|
|
|
assert check_defaults_handling(chat.pin_message, chat.bot)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'pin_chat_message', make_assertion)
|
2020-12-30 13:41:07 +01:00
|
|
|
assert chat.pin_message(message_id=42)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
|
|
|
def test_unpin_message(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.unpin_message, Bot.unpin_chat_message, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.unpin_message, chat.bot, 'unpin_chat_message')
|
|
|
|
assert check_defaults_handling(chat.unpin_message, chat.bot)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'unpin_chat_message', make_assertion)
|
|
|
|
assert chat.unpin_message()
|
|
|
|
|
|
|
|
def test_unpin_all_messages(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.unpin_all_messages, Bot.unpin_all_chat_messages, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.unpin_all_messages, chat.bot, 'unpin_all_chat_messages')
|
|
|
|
assert check_defaults_handling(chat.unpin_all_messages, chat.bot)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'unpin_all_chat_messages', make_assertion)
|
|
|
|
assert chat.unpin_all_messages()
|
|
|
|
|
2018-02-18 16:49:52 +01:00
|
|
|
def test_instance_method_send_message(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['text'] == 'test'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_message, Bot.send_message, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_message, chat.bot, 'send_message')
|
|
|
|
assert check_defaults_handling(chat.send_message, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_message', make_assertion)
|
|
|
|
assert chat.send_message(text='test')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-08-13 13:39:43 +02:00
|
|
|
def test_instance_method_send_media_group(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['media'] == 'test_media_group'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.send_media_group, Bot.send_media_group, ['chat_id'], []
|
|
|
|
)
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_media_group, chat.bot, 'send_media_group')
|
|
|
|
assert check_defaults_handling(chat.send_media_group, chat.bot)
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_media_group', make_assertion)
|
|
|
|
assert chat.send_media_group(media='test_media_group')
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2018-03-17 00:10:11 +01:00
|
|
|
def test_instance_method_send_photo(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['photo'] == 'test_photo'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_photo, Bot.send_photo, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_photo, chat.bot, 'send_photo')
|
|
|
|
assert check_defaults_handling(chat.send_photo, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_photo', make_assertion)
|
|
|
|
assert chat.send_photo(photo='test_photo')
|
2018-03-17 00:10:11 +01:00
|
|
|
|
2020-08-13 13:39:43 +02:00
|
|
|
def test_instance_method_send_contact(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['phone_number'] == 'test_contact'
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_contact, Bot.send_contact, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_contact, chat.bot, 'send_contact')
|
|
|
|
assert check_defaults_handling(chat.send_contact, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_contact', make_assertion)
|
|
|
|
assert chat.send_contact(phone_number='test_contact')
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2018-02-18 16:49:52 +01:00
|
|
|
def test_instance_method_send_audio(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['audio'] == 'test_audio'
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_audio, Bot.send_audio, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_audio, chat.bot, 'send_audio')
|
|
|
|
assert check_defaults_handling(chat.send_audio, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_audio', make_assertion)
|
|
|
|
assert chat.send_audio(audio='test_audio')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
|
|
|
def test_instance_method_send_document(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['document'] == 'test_document'
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_document, Bot.send_document, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_document, chat.bot, 'send_document')
|
|
|
|
assert check_defaults_handling(chat.send_document, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_document', make_assertion)
|
|
|
|
assert chat.send_document(document='test_document')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-08-13 13:39:43 +02:00
|
|
|
def test_instance_method_send_dice(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['emoji'] == 'test_dice'
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_dice, Bot.send_dice, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_dice, chat.bot, 'send_dice')
|
|
|
|
assert check_defaults_handling(chat.send_dice, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_dice', make_assertion)
|
|
|
|
assert chat.send_dice(emoji='test_dice')
|
2020-08-13 13:39:43 +02:00
|
|
|
|
|
|
|
def test_instance_method_send_game(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['game_short_name'] == 'test_game'
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_game, Bot.send_game, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_game, chat.bot, 'send_game')
|
|
|
|
assert check_defaults_handling(chat.send_game, chat.bot)
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_game', make_assertion)
|
|
|
|
assert chat.send_game(game_short_name='test_game')
|
|
|
|
|
|
|
|
def test_instance_method_send_invoice(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
title = kwargs['title'] == 'title'
|
|
|
|
description = kwargs['description'] == 'description'
|
|
|
|
payload = kwargs['payload'] == 'payload'
|
|
|
|
provider_token = kwargs['provider_token'] == 'provider_token'
|
|
|
|
currency = kwargs['currency'] == 'currency'
|
|
|
|
prices = kwargs['prices'] == 'prices'
|
2021-04-30 10:09:21 +02:00
|
|
|
args = title and description and payload and provider_token and currency and prices
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and args
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_invoice, Bot.send_invoice, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_invoice, chat.bot, 'send_invoice')
|
|
|
|
assert check_defaults_handling(chat.send_invoice, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_invoice', make_assertion)
|
|
|
|
assert chat.send_invoice(
|
|
|
|
'title',
|
|
|
|
'description',
|
|
|
|
'payload',
|
|
|
|
'provider_token',
|
|
|
|
'currency',
|
|
|
|
'prices',
|
|
|
|
)
|
2020-08-13 13:39:43 +02:00
|
|
|
|
|
|
|
def test_instance_method_send_location(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['latitude'] == 'test_location'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_location, Bot.send_location, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_location, chat.bot, 'send_location')
|
|
|
|
assert check_defaults_handling(chat.send_location, chat.bot)
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_location', make_assertion)
|
|
|
|
assert chat.send_location(latitude='test_location')
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2018-02-18 16:49:52 +01:00
|
|
|
def test_instance_method_send_sticker(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['sticker'] == 'test_sticker'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_sticker, Bot.send_sticker, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_sticker, chat.bot, 'send_sticker')
|
|
|
|
assert check_defaults_handling(chat.send_sticker, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_sticker', make_assertion)
|
|
|
|
assert chat.send_sticker(sticker='test_sticker')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-08-13 13:39:43 +02:00
|
|
|
def test_instance_method_send_venue(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['title'] == 'test_venue'
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_venue, Bot.send_venue, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_venue, chat.bot, 'send_venue')
|
|
|
|
assert check_defaults_handling(chat.send_venue, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_venue', make_assertion)
|
|
|
|
assert chat.send_venue(title='test_venue')
|
2020-08-13 13:39:43 +02:00
|
|
|
|
2018-02-18 16:49:52 +01:00
|
|
|
def test_instance_method_send_video(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['video'] == 'test_video'
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_video, Bot.send_video, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_video, chat.bot, 'send_video')
|
|
|
|
assert check_defaults_handling(chat.send_video, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_video', make_assertion)
|
|
|
|
assert chat.send_video(video='test_video')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
|
|
|
def test_instance_method_send_video_note(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['video_note'] == 'test_video_note'
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_video_note, Bot.send_video_note, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_video_note, chat.bot, 'send_video_note')
|
|
|
|
assert check_defaults_handling(chat.send_video_note, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_video_note', make_assertion)
|
|
|
|
assert chat.send_video_note(video_note='test_video_note')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
|
|
|
def test_instance_method_send_voice(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['voice'] == 'test_voice'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_voice, Bot.send_voice, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_voice, chat.bot, 'send_voice')
|
|
|
|
assert check_defaults_handling(chat.send_voice, chat.bot)
|
2018-02-18 16:49:52 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_voice', make_assertion)
|
|
|
|
assert chat.send_voice(voice='test_voice')
|
2018-02-18 16:49:52 +01:00
|
|
|
|
Bot API 4.0 (#1168)
Telegram Passport (#1174):
- Add full support for telegram passport.
- New types: PassportData, PassportFile, EncryptedPassportElement, EncryptedCredentials, PassportElementError, PassportElementErrorDataField, PassportElementErrorFrontSide, PassportElementErrorReverseSide, PassportElementErrorSelfie, PassportElementErrorFile and PassportElementErrorFiles.
- New bot method: set_passport_data_errors
- New filter: Filters.passport_data
- Field passport_data field on Message
- PassportData is automagically decrypted when you specify your private key when creating Updater or Bot.
- PassportFiles is also automagically decrypted as you download/retrieve them.
- See new passportbot.py example for details on how to use, or go to our telegram passport wiki page for more info
- NOTE: Passport decryption requires new dependency `cryptography`.
Inputfile rework (#1184):
- Change how Inputfile is handled internally
- This allows support for specifying the thumbnails of photos and videos using the thumb= argument in the different send_ methods.
- Also allows Bot.send_media_group to actually finally send more than one media.
- Add thumb to Audio, Video and Videonote
- Add Bot.edit_message_media together with InputMediaAnimation, InputMediaAudio, and inputMediaDocument.
Other Bot API 4.0 changes:
- Add forusquare_type to Venue, InlineQueryResultVenue, InputVenueMessageContent, and Bot.send_venue. (#1170)
- Add vCard support by adding vcard field to Contact, InlineQueryResultContact, InputContactMessageContent, and Bot.send_contact. (#1166)
- Support new message entities: CASHTAG and PHONE_NUMBER. (#1179)
- Cashtag seems to be things like $USD and $GBP, but it seems telegram doesn't currently send them to bots.
- Phone number also seems to have limited support for now
- Add Bot.send_animation, add width, height, and duration to Animation, and add Filters.animation. (#1172)
Co-authored-by: Jasmin Bom <jsmnbom@gmail.com>
Co-authored-by: code1mountain <32801117+code1mountain@users.noreply.github.com>
Co-authored-by: Eldinnie <pieter.schutz+github@gmail.com>
Co-authored-by: mathefreak1 <mathefreak@hi2.in>
2018-08-29 14:18:58 +02:00
|
|
|
def test_instance_method_send_animation(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['animation'] == 'test_animation'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_animation, Bot.send_animation, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_animation, chat.bot, 'send_animation')
|
|
|
|
assert check_defaults_handling(chat.send_animation, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'send_animation', make_assertion)
|
|
|
|
assert chat.send_animation(animation='test_animation')
|
Bot API 4.0 (#1168)
Telegram Passport (#1174):
- Add full support for telegram passport.
- New types: PassportData, PassportFile, EncryptedPassportElement, EncryptedCredentials, PassportElementError, PassportElementErrorDataField, PassportElementErrorFrontSide, PassportElementErrorReverseSide, PassportElementErrorSelfie, PassportElementErrorFile and PassportElementErrorFiles.
- New bot method: set_passport_data_errors
- New filter: Filters.passport_data
- Field passport_data field on Message
- PassportData is automagically decrypted when you specify your private key when creating Updater or Bot.
- PassportFiles is also automagically decrypted as you download/retrieve them.
- See new passportbot.py example for details on how to use, or go to our telegram passport wiki page for more info
- NOTE: Passport decryption requires new dependency `cryptography`.
Inputfile rework (#1184):
- Change how Inputfile is handled internally
- This allows support for specifying the thumbnails of photos and videos using the thumb= argument in the different send_ methods.
- Also allows Bot.send_media_group to actually finally send more than one media.
- Add thumb to Audio, Video and Videonote
- Add Bot.edit_message_media together with InputMediaAnimation, InputMediaAudio, and inputMediaDocument.
Other Bot API 4.0 changes:
- Add forusquare_type to Venue, InlineQueryResultVenue, InputVenueMessageContent, and Bot.send_venue. (#1170)
- Add vCard support by adding vcard field to Contact, InlineQueryResultContact, InputContactMessageContent, and Bot.send_contact. (#1166)
- Support new message entities: CASHTAG and PHONE_NUMBER. (#1179)
- Cashtag seems to be things like $USD and $GBP, but it seems telegram doesn't currently send them to bots.
- Phone number also seems to have limited support for now
- Add Bot.send_animation, add width, height, and duration to Animation, and add Filters.animation. (#1172)
Co-authored-by: Jasmin Bom <jsmnbom@gmail.com>
Co-authored-by: code1mountain <32801117+code1mountain@users.noreply.github.com>
Co-authored-by: Eldinnie <pieter.schutz+github@gmail.com>
Co-authored-by: mathefreak1 <mathefreak@hi2.in>
2018-08-29 14:18:58 +02:00
|
|
|
|
2019-08-23 21:20:41 +02:00
|
|
|
def test_instance_method_send_poll(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
2021-02-19 19:07:48 +01:00
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['question'] == 'test_poll'
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.send_poll, Bot.send_poll, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.send_poll, chat.bot, 'send_poll')
|
|
|
|
assert check_defaults_handling(chat.send_poll, chat.bot)
|
2019-08-23 21:20:41 +02:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
monkeypatch.setattr(chat.bot, 'send_poll', make_assertion)
|
|
|
|
assert chat.send_poll(question='test_poll', options=[1, 2])
|
2019-08-23 21:20:41 +02:00
|
|
|
|
2020-11-29 16:20:46 +01:00
|
|
|
def test_instance_method_send_copy(self, monkeypatch, chat):
|
2020-12-30 13:41:07 +01:00
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
from_chat_id = kwargs['from_chat_id'] == 'test_copy'
|
|
|
|
message_id = kwargs['message_id'] == 42
|
|
|
|
chat_id = kwargs['chat_id'] == chat.id
|
2021-02-19 19:07:48 +01:00
|
|
|
return from_chat_id and message_id and chat_id
|
2020-11-29 16:20:46 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
assert check_shortcut_signature(Chat.send_copy, Bot.copy_message, ['chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.copy_message, chat.bot, 'copy_message')
|
|
|
|
assert check_defaults_handling(chat.copy_message, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'copy_message', make_assertion)
|
|
|
|
assert chat.send_copy(from_chat_id='test_copy', message_id=42)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
2020-12-30 13:41:07 +01:00
|
|
|
def test_instance_method_copy_message(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
from_chat_id = kwargs['from_chat_id'] == chat.id
|
|
|
|
message_id = kwargs['message_id'] == 42
|
|
|
|
chat_id = kwargs['chat_id'] == 'test_copy'
|
2021-02-19 19:07:48 +01:00
|
|
|
return from_chat_id and message_id and chat_id
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
assert check_shortcut_signature(Chat.copy_message, Bot.copy_message, ['from_chat_id'], [])
|
2021-02-19 19:07:48 +01:00
|
|
|
assert check_shortcut_call(chat.copy_message, chat.bot, 'copy_message')
|
|
|
|
assert check_defaults_handling(chat.copy_message, chat.bot)
|
2020-12-30 13:41:07 +01:00
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'copy_message', make_assertion)
|
|
|
|
assert chat.copy_message(chat_id='test_copy', message_id=42)
|
2020-11-29 16:20:46 +01:00
|
|
|
|
2021-03-14 16:41:35 +01:00
|
|
|
def test_export_invite_link(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.export_invite_link, Bot.export_chat_invite_link, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.export_invite_link, chat.bot, 'export_chat_invite_link')
|
|
|
|
assert check_defaults_handling(chat.export_invite_link, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'export_chat_invite_link', make_assertion)
|
|
|
|
assert chat.export_invite_link()
|
|
|
|
|
|
|
|
def test_create_invite_link(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.create_invite_link, Bot.create_chat_invite_link, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.create_invite_link, chat.bot, 'create_chat_invite_link')
|
|
|
|
assert check_defaults_handling(chat.create_invite_link, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'create_chat_invite_link', make_assertion)
|
|
|
|
assert chat.create_invite_link()
|
|
|
|
|
|
|
|
def test_edit_invite_link(self, monkeypatch, chat):
|
|
|
|
link = "ThisIsALink"
|
|
|
|
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['invite_link'] == link
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.edit_invite_link, Bot.edit_chat_invite_link, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.edit_invite_link, chat.bot, 'edit_chat_invite_link')
|
|
|
|
assert check_defaults_handling(chat.edit_invite_link, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'edit_chat_invite_link', make_assertion)
|
|
|
|
assert chat.edit_invite_link(invite_link=link)
|
|
|
|
|
|
|
|
def test_revoke_invite_link(self, monkeypatch, chat):
|
|
|
|
link = "ThisIsALink"
|
|
|
|
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['invite_link'] == link
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.revoke_invite_link, Bot.revoke_chat_invite_link, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(chat.revoke_invite_link, chat.bot, 'revoke_chat_invite_link')
|
|
|
|
assert check_defaults_handling(chat.revoke_invite_link, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'revoke_chat_invite_link', make_assertion)
|
|
|
|
assert chat.revoke_invite_link(invite_link=link)
|
|
|
|
|
2021-11-08 19:02:20 +01:00
|
|
|
def test_approve_join_request(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['user_id'] == 42
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.approve_join_request, Bot.approve_chat_join_request, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(
|
|
|
|
chat.approve_join_request, chat.bot, 'approve_chat_join_request'
|
|
|
|
)
|
|
|
|
assert check_defaults_handling(chat.approve_join_request, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'approve_chat_join_request', make_assertion)
|
|
|
|
assert chat.approve_join_request(user_id=42)
|
|
|
|
|
|
|
|
def test_decline_join_request(self, monkeypatch, chat):
|
|
|
|
def make_assertion(*_, **kwargs):
|
|
|
|
return kwargs['chat_id'] == chat.id and kwargs['user_id'] == 42
|
|
|
|
|
|
|
|
assert check_shortcut_signature(
|
|
|
|
Chat.decline_join_request, Bot.decline_chat_join_request, ['chat_id'], []
|
|
|
|
)
|
|
|
|
assert check_shortcut_call(
|
|
|
|
chat.decline_join_request, chat.bot, 'decline_chat_join_request'
|
|
|
|
)
|
|
|
|
assert check_defaults_handling(chat.decline_join_request, chat.bot)
|
|
|
|
|
|
|
|
monkeypatch.setattr(chat.bot, 'decline_chat_join_request', make_assertion)
|
|
|
|
assert chat.decline_join_request(user_id=42)
|
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
def test_equality(self):
|
2020-02-23 22:03:58 +01:00
|
|
|
a = Chat(self.id_, self.title, self.type_)
|
|
|
|
b = Chat(self.id_, self.title, self.type_)
|
|
|
|
c = Chat(self.id_, '', '')
|
|
|
|
d = Chat(0, self.title, self.type_)
|
|
|
|
e = User(self.id_, '', False)
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a == b
|
|
|
|
assert hash(a) == hash(b)
|
|
|
|
assert a is not b
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a == c
|
|
|
|
assert hash(a) == hash(c)
|
2017-05-14 23:29:31 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a != d
|
|
|
|
assert hash(a) != hash(d)
|
2016-05-15 02:39:11 +02:00
|
|
|
|
2017-08-11 23:58:41 +02:00
|
|
|
assert a != e
|
|
|
|
assert hash(a) != hash(e)
|