예제 #1
0
    def test_de_json(self, bot):
        json_dict = {
            'location': self.location.to_dict(),
            'address': self.address,
        }
        chat_location = ChatLocation.de_json(json_dict, bot)

        assert chat_location.location == self.location
        assert chat_location.address == self.address
예제 #2
0
    def test_equality(self, chat_location):
        a = chat_location
        b = ChatLocation(self.location, self.address)
        c = ChatLocation(self.location, 'Mordor')
        d = ChatLocation(Location(456, 132), self.address)
        e = User(456, '', False)

        assert a == b
        assert hash(a) == hash(b)
        assert a is not b

        assert a == c
        assert hash(a) == hash(c)

        assert a != d
        assert hash(a) != hash(d)

        assert a != e
        assert hash(a) != hash(e)
예제 #3
0
class TestChat:
    id_ = -28767330
    title = 'ToledosPalaceBot - Group'
    type_ = 'group'
    username = '******'
    all_members_are_administrators = False
    sticker_set_name = 'stickers'
    can_set_sticker_set = False
    permissions = ChatPermissions(
        can_send_messages=True,
        can_change_info=False,
        can_invite_users=True,
    )
    slow_mode_delay = 30
    message_auto_delete_time = 42
    bio = "I'm a Barbie Girl in a Barbie World"
    linked_chat_id = 11880
    location = ChatLocation(Location(123, 456), 'Barbie World')

    def test_de_json(self, bot):
        json_dict = {
            'id': self.id_,
            'title': self.title,
            'type': self.type_,
            'username': self.username,
            'all_members_are_administrators':
            self.all_members_are_administrators,
            'sticker_set_name': self.sticker_set_name,
            'can_set_sticker_set': self.can_set_sticker_set,
            'permissions': self.permissions.to_dict(),
            'slow_mode_delay': self.slow_mode_delay,
            'message_auto_delete_time': self.message_auto_delete_time,
            'bio': self.bio,
            'linked_chat_id': self.linked_chat_id,
            'location': self.location.to_dict(),
        }
        chat = Chat.de_json(json_dict, bot)

        assert chat.id == self.id_
        assert chat.title == self.title
        assert chat.type == self.type_
        assert chat.username == self.username
        assert chat.all_members_are_administrators == self.all_members_are_administrators
        assert chat.sticker_set_name == self.sticker_set_name
        assert chat.can_set_sticker_set == self.can_set_sticker_set
        assert chat.permissions == self.permissions
        assert chat.slow_mode_delay == self.slow_mode_delay
        assert chat.message_auto_delete_time == self.message_auto_delete_time
        assert chat.bio == self.bio
        assert chat.linked_chat_id == self.linked_chat_id
        assert chat.location.location == self.location.location
        assert chat.location.address == self.location.address

    def test_to_dict(self, chat):
        chat_dict = chat.to_dict()

        assert isinstance(chat_dict, dict)
        assert chat_dict['id'] == chat.id
        assert chat_dict['title'] == chat.title
        assert chat_dict['type'] == chat.type
        assert chat_dict['username'] == chat.username
        assert chat_dict[
            'all_members_are_administrators'] == chat.all_members_are_administrators
        assert chat_dict['permissions'] == chat.permissions.to_dict()
        assert chat_dict['slow_mode_delay'] == chat.slow_mode_delay
        assert chat_dict[
            'message_auto_delete_time'] == chat.message_auto_delete_time
        assert chat_dict['bio'] == chat.bio
        assert chat_dict['linked_chat_id'] == chat.linked_chat_id
        assert chat_dict['location'] == chat.location.to_dict()

    def test_link(self, chat):
        assert chat.link == f'https://t.me/{chat.username}'
        chat.username = None
        assert chat.link is None

    def test_full_name(self):
        chat = Chat(id=1,
                    type=Chat.PRIVATE,
                    first_name='first\u2022name',
                    last_name='last\u2022name')
        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'
        chat = Chat(
            id=1,
            type=Chat.PRIVATE,
        )
        assert chat.full_name is None

    def test_send_action(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            id_ = kwargs['chat_id'] == chat.id
            action = kwargs['action'] == ChatAction.TYPING
            return id_ and action

        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)

        monkeypatch.setattr(chat.bot, 'send_chat_action', make_assertion)
        assert chat.send_action(action=ChatAction.TYPING)
        assert chat.send_action(action=ChatAction.TYPING)

    def test_leave(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id

        assert check_shortcut_signature(Chat.leave, Bot.leave_chat,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.leave, chat.bot, 'leave_chat')
        assert check_defaults_handling(chat.leave, chat.bot)

        monkeypatch.setattr(chat.bot, 'leave_chat', make_assertion)
        assert chat.leave()

    def test_get_administrators(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id

        assert check_shortcut_signature(Chat.get_administrators,
                                        Bot.get_chat_administrators,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.get_administrators, chat.bot,
                                   'get_chat_administrators')
        assert check_defaults_handling(chat.get_administrators, chat.bot)

        monkeypatch.setattr(chat.bot, 'get_chat_administrators',
                            make_assertion)
        assert chat.get_administrators()

    def test_get_members_count(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id

        assert check_shortcut_signature(Chat.get_members_count,
                                        Bot.get_chat_members_count,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.get_members_count, chat.bot,
                                   'get_chat_members_count')
        assert check_defaults_handling(chat.get_members_count, chat.bot)

        monkeypatch.setattr(chat.bot, 'get_chat_members_count', make_assertion)
        assert chat.get_members_count()

    def test_get_member(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            chat_id = kwargs['chat_id'] == chat.id
            user_id = kwargs['user_id'] == 42
            return chat_id and user_id

        assert check_shortcut_signature(Chat.get_member, Bot.get_chat_member,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.get_member, chat.bot,
                                   'get_chat_member')
        assert check_defaults_handling(chat.get_member, chat.bot)

        monkeypatch.setattr(chat.bot, 'get_chat_member', make_assertion)
        assert chat.get_member(user_id=42)

    def test_kick_member(self, monkeypatch, chat):
        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

        assert check_shortcut_signature(Chat.kick_member, Bot.kick_chat_member,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.kick_member, chat.bot,
                                   'kick_chat_member')
        assert check_defaults_handling(chat.kick_member, chat.bot)

        monkeypatch.setattr(chat.bot, 'kick_chat_member', make_assertion)
        assert chat.kick_member(user_id=42, until_date=43)

    @pytest.mark.parametrize('only_if_banned', [True, False, None])
    def test_unban_member(self, monkeypatch, chat, only_if_banned):
        def make_assertion(*_, **kwargs):
            chat_id = kwargs['chat_id'] == chat.id
            user_id = kwargs['user_id'] == 42
            o_i_b = kwargs.get('only_if_banned', None) == only_if_banned
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.unban_member,
                                        Bot.unban_chat_member, ['chat_id'], [])
        assert check_shortcut_call(chat.unban_member, chat.bot,
                                   'unban_chat_member')
        assert check_defaults_handling(chat.unban_member, chat.bot)

        monkeypatch.setattr(chat.bot, 'unban_chat_member', make_assertion)
        assert chat.unban_member(user_id=42, only_if_banned=only_if_banned)

    @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
            o_i_b = kwargs.get('is_anonymous', None) == is_anonymous
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.promote_member,
                                        Bot.promote_chat_member, ['chat_id'],
                                        [])
        assert check_shortcut_call(chat.promote_member, chat.bot,
                                   'promote_chat_member')
        assert check_defaults_handling(chat.promote_member, chat.bot)

        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
            o_i_b = kwargs.get('permissions', None) == permissions
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.restrict_member,
                                        Bot.restrict_chat_member, ['chat_id'],
                                        [])
        assert check_shortcut_call(chat.restrict_member, chat.bot,
                                   'restrict_chat_member')
        assert check_defaults_handling(chat.restrict_member, chat.bot)

        monkeypatch.setattr(chat.bot, 'restrict_chat_member', make_assertion)
        assert chat.restrict_member(user_id=42, permissions=permissions)

    def test_set_permissions(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            chat_id = kwargs['chat_id'] == chat.id
            permissions = kwargs['permissions'] == self.permissions
            return chat_id and permissions

        assert check_shortcut_signature(Chat.set_permissions,
                                        Bot.set_chat_permissions, ['chat_id'],
                                        [])
        assert check_shortcut_call(chat.set_permissions, chat.bot,
                                   'set_chat_permissions')
        assert check_defaults_handling(chat.set_permissions, chat.bot)

        monkeypatch.setattr(chat.bot, 'set_chat_permissions', make_assertion)
        assert chat.set_permissions(permissions=self.permissions)

    def test_set_administrator_custom_title(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            chat_id = kwargs['chat_id'] == chat.id
            user_id = kwargs['user_id'] == 42
            custom_title = kwargs['custom_title'] == 'custom_title'
            return chat_id and user_id and custom_title

        monkeypatch.setattr('telegram.Bot.set_chat_administrator_custom_title',
                            make_assertion)
        assert chat.set_administrator_custom_title(user_id=42,
                                                   custom_title='custom_title')

    def test_pin_message(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs['message_id'] == 42

        assert check_shortcut_signature(Chat.pin_message, Bot.pin_chat_message,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.pin_message, chat.bot,
                                   'pin_chat_message')
        assert check_defaults_handling(chat.pin_message, chat.bot)

        monkeypatch.setattr(chat.bot, 'pin_chat_message', make_assertion)
        assert chat.pin_message(message_id=42)

    def test_unpin_message(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id

        assert check_shortcut_signature(Chat.unpin_message,
                                        Bot.unpin_chat_message, ['chat_id'],
                                        [])
        assert check_shortcut_call(chat.unpin_message, chat.bot,
                                   'unpin_chat_message')
        assert check_defaults_handling(chat.unpin_message, chat.bot)

        monkeypatch.setattr(chat.bot, 'unpin_chat_message', make_assertion)
        assert chat.unpin_message()

    def test_unpin_all_messages(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id

        assert check_shortcut_signature(Chat.unpin_all_messages,
                                        Bot.unpin_all_chat_messages,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.unpin_all_messages, chat.bot,
                                   'unpin_all_chat_messages')
        assert check_defaults_handling(chat.unpin_all_messages, chat.bot)

        monkeypatch.setattr(chat.bot, 'unpin_all_chat_messages',
                            make_assertion)
        assert chat.unpin_all_messages()

    def test_instance_method_send_message(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs['text'] == 'test'

        assert check_shortcut_signature(Chat.send_message, Bot.send_message,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_message, chat.bot, 'send_message')
        assert check_defaults_handling(chat.send_message, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_message', make_assertion)
        assert chat.send_message(text='test')

    def test_instance_method_send_media_group(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'media'] == 'test_media_group'

        assert check_shortcut_signature(Chat.send_media_group,
                                        Bot.send_media_group, ['chat_id'], [])
        assert check_shortcut_call(chat.send_media_group, chat.bot,
                                   'send_media_group')
        assert check_defaults_handling(chat.send_media_group, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_media_group', make_assertion)
        assert chat.send_media_group(media='test_media_group')

    def test_instance_method_send_photo(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'photo'] == 'test_photo'

        assert check_shortcut_signature(Chat.send_photo, Bot.send_photo,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_photo, chat.bot, 'send_photo')
        assert check_defaults_handling(chat.send_photo, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_photo', make_assertion)
        assert chat.send_photo(photo='test_photo')

    def test_instance_method_send_contact(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'phone_number'] == 'test_contact'

        assert check_shortcut_signature(Chat.send_contact, Bot.send_contact,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_contact, chat.bot, 'send_contact')
        assert check_defaults_handling(chat.send_contact, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_contact', make_assertion)
        assert chat.send_contact(phone_number='test_contact')

    def test_instance_method_send_audio(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'audio'] == 'test_audio'

        assert check_shortcut_signature(Chat.send_audio, Bot.send_audio,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_audio, chat.bot, 'send_audio')
        assert check_defaults_handling(chat.send_audio, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_audio', make_assertion)
        assert chat.send_audio(audio='test_audio')

    def test_instance_method_send_document(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'document'] == 'test_document'

        assert check_shortcut_signature(Chat.send_document, Bot.send_document,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_document, chat.bot,
                                   'send_document')
        assert check_defaults_handling(chat.send_document, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_document', make_assertion)
        assert chat.send_document(document='test_document')

    def test_instance_method_send_dice(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'emoji'] == 'test_dice'

        assert check_shortcut_signature(Chat.send_dice, Bot.send_dice,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_dice, chat.bot, 'send_dice')
        assert check_defaults_handling(chat.send_dice, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_dice', make_assertion)
        assert chat.send_dice(emoji='test_dice')

    def test_instance_method_send_game(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'game_short_name'] == 'test_game'

        assert check_shortcut_signature(Chat.send_game, Bot.send_game,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_game, chat.bot, 'send_game')
        assert check_defaults_handling(chat.send_game, chat.bot)

        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'
            start_parameter = kwargs['start_parameter'] == 'start_parameter'
            currency = kwargs['currency'] == 'currency'
            prices = kwargs['prices'] == 'prices'
            args = (title and description and payload and provider_token
                    and start_parameter and currency and prices)
            return kwargs['chat_id'] == chat.id and args

        assert check_shortcut_signature(Chat.send_invoice, Bot.send_invoice,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_invoice, chat.bot, 'send_invoice')
        assert check_defaults_handling(chat.send_invoice, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_invoice', make_assertion)
        assert chat.send_invoice(
            'title',
            'description',
            'payload',
            'provider_token',
            'start_parameter',
            'currency',
            'prices',
        )

    def test_instance_method_send_location(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'latitude'] == 'test_location'

        assert check_shortcut_signature(Chat.send_location, Bot.send_location,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_location, chat.bot,
                                   'send_location')
        assert check_defaults_handling(chat.send_location, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_location', make_assertion)
        assert chat.send_location(latitude='test_location')

    def test_instance_method_send_sticker(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'sticker'] == 'test_sticker'

        assert check_shortcut_signature(Chat.send_sticker, Bot.send_sticker,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_sticker, chat.bot, 'send_sticker')
        assert check_defaults_handling(chat.send_sticker, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_sticker', make_assertion)
        assert chat.send_sticker(sticker='test_sticker')

    def test_instance_method_send_venue(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'title'] == 'test_venue'

        assert check_shortcut_signature(Chat.send_venue, Bot.send_venue,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_venue, chat.bot, 'send_venue')
        assert check_defaults_handling(chat.send_venue, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_venue', make_assertion)
        assert chat.send_venue(title='test_venue')

    def test_instance_method_send_video(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'video'] == 'test_video'

        assert check_shortcut_signature(Chat.send_video, Bot.send_video,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_video, chat.bot, 'send_video')
        assert check_defaults_handling(chat.send_video, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_video', make_assertion)
        assert chat.send_video(video='test_video')

    def test_instance_method_send_video_note(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'video_note'] == 'test_video_note'

        assert check_shortcut_signature(Chat.send_video_note,
                                        Bot.send_video_note, ['chat_id'], [])
        assert check_shortcut_call(chat.send_video_note, chat.bot,
                                   'send_video_note')
        assert check_defaults_handling(chat.send_video_note, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_video_note', make_assertion)
        assert chat.send_video_note(video_note='test_video_note')

    def test_instance_method_send_voice(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'voice'] == 'test_voice'

        assert check_shortcut_signature(Chat.send_voice, Bot.send_voice,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_voice, chat.bot, 'send_voice')
        assert check_defaults_handling(chat.send_voice, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_voice', make_assertion)
        assert chat.send_voice(voice='test_voice')

    def test_instance_method_send_animation(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'animation'] == 'test_animation'

        assert check_shortcut_signature(Chat.send_animation,
                                        Bot.send_animation, ['chat_id'], [])
        assert check_shortcut_call(chat.send_animation, chat.bot,
                                   'send_animation')
        assert check_defaults_handling(chat.send_animation, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_animation', make_assertion)
        assert chat.send_animation(animation='test_animation')

    def test_instance_method_send_poll(self, monkeypatch, chat):
        def make_assertion(*_, **kwargs):
            return kwargs['chat_id'] == chat.id and kwargs[
                'question'] == 'test_poll'

        assert check_shortcut_signature(Chat.send_poll, Bot.send_poll,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.send_poll, chat.bot, 'send_poll')
        assert check_defaults_handling(chat.send_poll, chat.bot)

        monkeypatch.setattr(chat.bot, 'send_poll', make_assertion)
        assert chat.send_poll(question='test_poll', options=[1, 2])

    def test_instance_method_send_copy(self, monkeypatch, chat):
        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
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.send_copy, Bot.copy_message,
                                        ['chat_id'], [])
        assert check_shortcut_call(chat.copy_message, chat.bot, 'copy_message')
        assert check_defaults_handling(chat.copy_message, chat.bot)

        monkeypatch.setattr(chat.bot, 'copy_message', make_assertion)
        assert chat.send_copy(from_chat_id='test_copy', message_id=42)

    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'
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.copy_message, Bot.copy_message,
                                        ['from_chat_id'], [])
        assert check_shortcut_call(chat.copy_message, chat.bot, 'copy_message')
        assert check_defaults_handling(chat.copy_message, chat.bot)

        monkeypatch.setattr(chat.bot, 'copy_message', make_assertion)
        assert chat.copy_message(chat_id='test_copy', message_id=42)

    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)

    def test_equality(self):
        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)

        assert a == b
        assert hash(a) == hash(b)
        assert a is not b

        assert a == c
        assert hash(a) == hash(c)

        assert a != d
        assert hash(a) != hash(d)

        assert a != e
        assert hash(a) != hash(e)
예제 #4
0
class TestChat:
    id_ = -28767330
    title = 'ToledosPalaceBot - Group'
    type_ = 'group'
    username = '******'
    all_members_are_administrators = False
    sticker_set_name = 'stickers'
    can_set_sticker_set = False
    permissions = ChatPermissions(
        can_send_messages=True,
        can_change_info=False,
        can_invite_users=True,
    )
    slow_mode_delay = 30
    bio = "I'm a Barbie Girl in a Barbie World"
    linked_chat_id = 11880
    location = ChatLocation(Location(123, 456), 'Barbie World')

    def test_de_json(self, bot):
        json_dict = {
            'id': self.id_,
            'title': self.title,
            'type': self.type_,
            'username': self.username,
            'all_members_are_administrators':
            self.all_members_are_administrators,
            'sticker_set_name': self.sticker_set_name,
            'can_set_sticker_set': self.can_set_sticker_set,
            'permissions': self.permissions.to_dict(),
            'slow_mode_delay': self.slow_mode_delay,
            'bio': self.bio,
            'linked_chat_id': self.linked_chat_id,
            'location': self.location.to_dict(),
        }
        chat = Chat.de_json(json_dict, bot)

        assert chat.id == self.id_
        assert chat.title == self.title
        assert chat.type == self.type_
        assert chat.username == self.username
        assert chat.all_members_are_administrators == self.all_members_are_administrators
        assert chat.sticker_set_name == self.sticker_set_name
        assert chat.can_set_sticker_set == self.can_set_sticker_set
        assert chat.permissions == self.permissions
        assert chat.slow_mode_delay == self.slow_mode_delay
        assert chat.bio == self.bio
        assert chat.linked_chat_id == self.linked_chat_id
        assert chat.location.location == self.location.location
        assert chat.location.address == self.location.address

    def test_to_dict(self, chat):
        chat_dict = chat.to_dict()

        assert isinstance(chat_dict, dict)
        assert chat_dict['id'] == chat.id
        assert chat_dict['title'] == chat.title
        assert chat_dict['type'] == chat.type
        assert chat_dict['username'] == chat.username
        assert chat_dict[
            'all_members_are_administrators'] == chat.all_members_are_administrators
        assert chat_dict['permissions'] == chat.permissions.to_dict()
        assert chat_dict['slow_mode_delay'] == chat.slow_mode_delay
        assert chat_dict['bio'] == chat.bio
        assert chat_dict['linked_chat_id'] == chat.linked_chat_id
        assert chat_dict['location'] == chat.location.to_dict()

    def test_link(self, chat):
        assert chat.link == f'https://t.me/{chat.username}'
        chat.username = None
        assert chat.link is None

    def test_send_action(self, monkeypatch, chat):
        def test(*args, **kwargs):
            id_ = args[0] == chat.id
            action = kwargs['action'] == ChatAction.TYPING
            return id_ and action

        monkeypatch.setattr(chat.bot, 'send_chat_action', test)
        assert chat.send_action(action=ChatAction.TYPING)
        assert chat.send_chat_action(action=ChatAction.TYPING)

    def test_leave(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'leave_chat', test)
        assert chat.leave()

    def test_get_administrators(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'get_chat_administrators', test)
        assert chat.get_administrators()

    def test_get_members_count(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'get_chat_members_count', test)
        assert chat.get_members_count()

    def test_get_member(self, monkeypatch, chat):
        def test(*args, **kwargs):
            chat_id = args[0] == chat.id
            user_id = args[1] == 42
            return chat_id and user_id

        monkeypatch.setattr(chat.bot, 'get_chat_member', test)
        assert chat.get_member(42)

    def test_kick_member(self, monkeypatch, chat):
        def test(*args, **kwargs):
            chat_id = args[0] == chat.id
            user_id = args[1] == 42
            until = kwargs['until_date'] == 43
            return chat_id and user_id and until

        monkeypatch.setattr(chat.bot, 'kick_chat_member', test)
        assert chat.kick_member(42, until_date=43)

    @pytest.mark.parametrize('only_if_banned', [True, False, None])
    def test_unban_member(self, monkeypatch, chat, only_if_banned):
        def make_assertion(*args, **kwargs):
            chat_id = args[0] == chat.id
            user_id = args[1] == 42
            o_i_b = kwargs.get('only_if_banned', None) == only_if_banned
            return chat_id and user_id and o_i_b

        monkeypatch.setattr(chat.bot, 'unban_chat_member', make_assertion)
        assert chat.unban_member(42, only_if_banned=only_if_banned)

    def test_set_permissions(self, monkeypatch, chat):
        def test(*args, **kwargs):
            chat_id = args[0] == chat.id
            permissions = args[1] == self.permissions
            return chat_id and permissions

        monkeypatch.setattr(chat.bot, 'set_chat_permissions', test)
        assert chat.set_permissions(self.permissions)

    def test_set_administrator_custom_title(self, monkeypatch, chat):
        def test(*args, **kwargs):
            chat_id = args[1] == chat.id
            user_id = args[2] == 42
            custom_title = args[3] == 'custom_title'
            return chat_id and user_id and custom_title

        monkeypatch.setattr('telegram.Bot.set_chat_administrator_custom_title',
                            test)
        assert chat.set_administrator_custom_title(42, 'custom_title')

    def test_pin_message(self, monkeypatch, chat):
        def make_assertion(*args, **kwargs):
            try:
                return kwargs['chat_id'] == chat.id
            except KeyError:
                return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'pin_chat_message', make_assertion)
        assert chat.pin_message()

    def test_unpin_message(self, monkeypatch, chat):
        def make_assertion(*args, **kwargs):
            try:
                return kwargs['chat_id'] == chat.id
            except KeyError:
                return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'unpin_chat_message', make_assertion)
        assert chat.unpin_message()

    def test_unpin_all_messages(self, monkeypatch, chat):
        def make_assertion(*args, **kwargs):
            try:
                return kwargs['chat_id'] == chat.id
            except KeyError:
                return args[0] == chat.id

        monkeypatch.setattr(chat.bot, 'unpin_all_chat_messages',
                            make_assertion)
        assert chat.unpin_all_messages()

    def test_instance_method_send_message(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test'

        monkeypatch.setattr(chat.bot, 'send_message', test)
        assert chat.send_message('test')

    def test_instance_method_send_media_group(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_media_group'

        monkeypatch.setattr(chat.bot, 'send_media_group', test)
        assert chat.send_media_group('test_media_group')

    def test_instance_method_send_photo(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_photo'

        monkeypatch.setattr(chat.bot, 'send_photo', test)
        assert chat.send_photo('test_photo')

    def test_instance_method_send_contact(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_contact'

        monkeypatch.setattr(chat.bot, 'send_contact', test)
        assert chat.send_contact('test_contact')

    def test_instance_method_send_audio(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_audio'

        monkeypatch.setattr(chat.bot, 'send_audio', test)
        assert chat.send_audio('test_audio')

    def test_instance_method_send_document(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_document'

        monkeypatch.setattr(chat.bot, 'send_document', test)
        assert chat.send_document('test_document')

    def test_instance_method_send_dice(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_dice'

        monkeypatch.setattr(chat.bot, 'send_dice', test)
        assert chat.send_dice('test_dice')

    def test_instance_method_send_game(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_game'

        monkeypatch.setattr(chat.bot, 'send_game', test)
        assert chat.send_game('test_game')

    def test_instance_method_send_invoice(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_invoice'

        monkeypatch.setattr(chat.bot, 'send_invoice', test)
        assert chat.send_invoice('test_invoice')

    def test_instance_method_send_location(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_location'

        monkeypatch.setattr(chat.bot, 'send_location', test)
        assert chat.send_location('test_location')

    def test_instance_method_send_sticker(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_sticker'

        monkeypatch.setattr(chat.bot, 'send_sticker', test)
        assert chat.send_sticker('test_sticker')

    def test_instance_method_send_venue(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_venue'

        monkeypatch.setattr(chat.bot, 'send_venue', test)
        assert chat.send_venue('test_venue')

    def test_instance_method_send_video(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_video'

        monkeypatch.setattr(chat.bot, 'send_video', test)
        assert chat.send_video('test_video')

    def test_instance_method_send_video_note(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_video_note'

        monkeypatch.setattr(chat.bot, 'send_video_note', test)
        assert chat.send_video_note('test_video_note')

    def test_instance_method_send_voice(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_voice'

        monkeypatch.setattr(chat.bot, 'send_voice', test)
        assert chat.send_voice('test_voice')

    def test_instance_method_send_animation(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_animation'

        monkeypatch.setattr(chat.bot, 'send_animation', test)
        assert chat.send_animation('test_animation')

    def test_instance_method_send_poll(self, monkeypatch, chat):
        def test(*args, **kwargs):
            return args[0] == chat.id and args[1] == 'test_poll'

        monkeypatch.setattr(chat.bot, 'send_poll', test)
        assert chat.send_poll('test_poll')

    def test_instance_method_send_copy(self, monkeypatch, chat):
        def test(*args, **kwargs):
            assert args[0] == 'test_copy'
            assert kwargs['chat_id'] == chat.id
            return args

        monkeypatch.setattr(chat.bot, 'copy_message', test)
        assert chat.send_copy('test_copy')

    def test_instance_method_copy_message(self, monkeypatch, chat):
        def test(*args, **kwargs):
            assert args[0] == 'test_copy'
            assert kwargs['from_chat_id'] == chat.id
            return args

        monkeypatch.setattr(chat.bot, 'copy_message', test)
        assert chat.copy_message('test_copy')

    def test_equality(self):
        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)

        assert a == b
        assert hash(a) == hash(b)
        assert a is not b

        assert a == c
        assert hash(a) == hash(c)

        assert a != d
        assert hash(a) != hash(d)

        assert a != e
        assert hash(a) != hash(e)
예제 #5
0
class TestChat:
    id_ = -28767330
    title = "ToledosPalaceBot - Group"
    type_ = "group"
    username = "******"
    all_members_are_administrators = False
    sticker_set_name = "stickers"
    can_set_sticker_set = False
    permissions = ChatPermissions(
        can_send_messages=True,
        can_change_info=False,
        can_invite_users=True,
    )
    slow_mode_delay = 30
    bio = "I'm a Barbie Girl in a Barbie World"
    linked_chat_id = 11880
    location = ChatLocation(Location(123, 456), "Barbie World")
    has_protected_content = True
    has_private_forwards = True
    join_to_send_messages = True
    join_by_request = True

    def test_slot_behaviour(self, chat, mro_slots):
        for attr in chat.__slots__:
            assert getattr(chat, attr,
                           "err") != "err", f"got extra slot '{attr}'"
        assert len(mro_slots(chat)) == len(set(
            mro_slots(chat))), "duplicate slot"

    def test_de_json(self, bot):
        json_dict = {
            "id": self.id_,
            "title": self.title,
            "type": self.type_,
            "username": self.username,
            "all_members_are_administrators":
            self.all_members_are_administrators,
            "sticker_set_name": self.sticker_set_name,
            "can_set_sticker_set": self.can_set_sticker_set,
            "permissions": self.permissions.to_dict(),
            "slow_mode_delay": self.slow_mode_delay,
            "bio": self.bio,
            "has_protected_content": self.has_protected_content,
            "has_private_forwards": self.has_private_forwards,
            "linked_chat_id": self.linked_chat_id,
            "location": self.location.to_dict(),
            "join_to_send_messages": self.join_to_send_messages,
            "join_by_request": self.join_by_request,
        }
        chat = Chat.de_json(json_dict, bot)

        assert chat.id == self.id_
        assert chat.title == self.title
        assert chat.type == self.type_
        assert chat.username == self.username
        assert chat.all_members_are_administrators == self.all_members_are_administrators
        assert chat.sticker_set_name == self.sticker_set_name
        assert chat.can_set_sticker_set == self.can_set_sticker_set
        assert chat.permissions == self.permissions
        assert chat.slow_mode_delay == self.slow_mode_delay
        assert chat.bio == self.bio
        assert chat.has_protected_content == self.has_protected_content
        assert chat.has_private_forwards == self.has_private_forwards
        assert chat.linked_chat_id == self.linked_chat_id
        assert chat.location.location == self.location.location
        assert chat.location.address == self.location.address
        assert chat.join_to_send_messages == self.join_to_send_messages
        assert chat.join_by_request == self.join_by_request

    def test_to_dict(self, chat):
        chat_dict = chat.to_dict()

        assert isinstance(chat_dict, dict)
        assert chat_dict["id"] == chat.id
        assert chat_dict["title"] == chat.title
        assert chat_dict["type"] == chat.type
        assert chat_dict["username"] == chat.username
        assert chat_dict[
            "all_members_are_administrators"] == chat.all_members_are_administrators
        assert chat_dict["permissions"] == chat.permissions.to_dict()
        assert chat_dict["slow_mode_delay"] == chat.slow_mode_delay
        assert chat_dict["bio"] == chat.bio
        assert chat_dict["has_private_forwards"] == chat.has_private_forwards
        assert chat_dict["has_protected_content"] == chat.has_protected_content
        assert chat_dict["linked_chat_id"] == chat.linked_chat_id
        assert chat_dict["location"] == chat.location.to_dict()
        assert chat_dict["join_to_send_messages"] == chat.join_to_send_messages
        assert chat_dict["join_by_request"] == chat.join_by_request

    def test_enum_init(self):
        chat = Chat(id=1, type="foo")
        assert chat.type == "foo"
        chat = Chat(id=1, type="private")
        assert chat.type is ChatType.PRIVATE

    def test_link(self, chat):
        assert chat.link == f"https://t.me/{chat.username}"
        chat.username = None
        assert chat.link is None

    def test_full_name(self):
        chat = Chat(id=1,
                    type=Chat.PRIVATE,
                    first_name="first\u2022name",
                    last_name="last\u2022name")
        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"
        chat = Chat(
            id=1,
            type=Chat.PRIVATE,
        )
        assert chat.full_name is None

    async def test_send_action(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            id_ = kwargs["chat_id"] == chat.id
            action = kwargs["action"] == ChatAction.TYPING
            return id_ and action

        assert check_shortcut_signature(chat.send_action, Bot.send_chat_action,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_action, chat.get_bot(),
                                         "send_chat_action")
        assert await check_defaults_handling(chat.send_action, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_chat_action", make_assertion)
        assert await chat.send_action(action=ChatAction.TYPING)
        assert await chat.send_action(action=ChatAction.TYPING)

    async def test_leave(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.leave, Bot.leave_chat,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.leave, chat.get_bot(),
                                         "leave_chat")
        assert await check_defaults_handling(chat.leave, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "leave_chat", make_assertion)
        assert await chat.leave()

    async def test_get_administrators(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.get_administrators,
                                        Bot.get_chat_administrators,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.get_administrators,
                                         chat.get_bot(),
                                         "get_chat_administrators")
        assert await check_defaults_handling(chat.get_administrators,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "get_chat_administrators",
                            make_assertion)
        assert await chat.get_administrators()

    async def test_get_members_count(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.get_member_count,
                                        Bot.get_chat_member_count, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.get_member_count, chat.get_bot(),
                                         "get_chat_member_count")
        assert await check_defaults_handling(chat.get_member_count,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "get_chat_member_count",
                            make_assertion)
        assert await chat.get_member_count()

    async def test_get_member(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            user_id = kwargs["user_id"] == 42
            return chat_id and user_id

        assert check_shortcut_signature(Chat.get_member, Bot.get_chat_member,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.get_member, chat.get_bot(),
                                         "get_chat_member")
        assert await check_defaults_handling(chat.get_member, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "get_chat_member", make_assertion)
        assert await chat.get_member(user_id=42)

    async def test_ban_member(self, monkeypatch, chat):
        async 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

        assert check_shortcut_signature(Chat.ban_member, Bot.ban_chat_member,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.ban_member, chat.get_bot(),
                                         "ban_chat_member")
        assert await check_defaults_handling(chat.ban_member, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "ban_chat_member", make_assertion)
        assert await chat.ban_member(user_id=42, until_date=43)

    async def test_ban_sender_chat(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.ban_sender_chat, chat.get_bot(),
                                         "ban_chat_sender_chat")
        assert await check_defaults_handling(chat.ban_sender_chat,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "ban_chat_sender_chat",
                            make_assertion)
        assert await chat.ban_sender_chat(42)

    async def test_ban_chat(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.ban_chat, chat.get_bot(),
                                         "ban_chat_sender_chat")
        assert await check_defaults_handling(chat.ban_chat, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "ban_chat_sender_chat",
                            make_assertion)
        assert await chat.ban_chat(42)

    @pytest.mark.parametrize("only_if_banned", [True, False, None])
    async def test_unban_member(self, monkeypatch, chat, only_if_banned):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            user_id = kwargs["user_id"] == 42
            o_i_b = kwargs.get("only_if_banned", None) == only_if_banned
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.unban_member,
                                        Bot.unban_chat_member, ["chat_id"], [])
        assert await check_shortcut_call(chat.unban_member, chat.get_bot(),
                                         "unban_chat_member")
        assert await check_defaults_handling(chat.unban_member, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "unban_chat_member",
                            make_assertion)
        assert await chat.unban_member(user_id=42,
                                       only_if_banned=only_if_banned)

    async def test_unban_sender_chat(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.unban_sender_chat,
                                         chat.get_bot(),
                                         "unban_chat_sender_chat")
        assert await check_defaults_handling(chat.unban_sender_chat,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "unban_chat_sender_chat",
                            make_assertion)
        assert await chat.unban_sender_chat(42)

    async def test_unban_chat(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.unban_chat, chat.get_bot(),
                                         "unban_chat_sender_chat")
        assert await check_defaults_handling(chat.unban_chat, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "unban_chat_sender_chat",
                            make_assertion)
        assert await chat.unban_chat(42)

    @pytest.mark.parametrize("is_anonymous", [True, False, None])
    async def test_promote_member(self, monkeypatch, chat, is_anonymous):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            user_id = kwargs["user_id"] == 42
            o_i_b = kwargs.get("is_anonymous", None) == is_anonymous
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.promote_member,
                                        Bot.promote_chat_member, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.promote_member, chat.get_bot(),
                                         "promote_chat_member")
        assert await check_defaults_handling(chat.promote_member,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "promote_chat_member",
                            make_assertion)
        assert await chat.promote_member(user_id=42, is_anonymous=is_anonymous)

    async def test_restrict_member(self, monkeypatch, chat):
        permissions = ChatPermissions(True, False, True, False, True, False,
                                      True, False)

        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            user_id = kwargs["user_id"] == 42
            o_i_b = kwargs.get("permissions", None) == permissions
            return chat_id and user_id and o_i_b

        assert check_shortcut_signature(Chat.restrict_member,
                                        Bot.restrict_chat_member, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.restrict_member, chat.get_bot(),
                                         "restrict_chat_member")
        assert await check_defaults_handling(chat.restrict_member,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "restrict_chat_member",
                            make_assertion)
        assert await chat.restrict_member(user_id=42, permissions=permissions)

    async def test_set_permissions(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            permissions = kwargs["permissions"] == self.permissions
            return chat_id and permissions

        assert check_shortcut_signature(Chat.set_permissions,
                                        Bot.set_chat_permissions, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.set_permissions, chat.get_bot(),
                                         "set_chat_permissions")
        assert await check_defaults_handling(chat.set_permissions,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "set_chat_permissions",
                            make_assertion)
        assert await chat.set_permissions(permissions=self.permissions)

    async def test_set_administrator_custom_title(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            user_id = kwargs["user_id"] == 42
            custom_title = kwargs["custom_title"] == "custom_title"
            return chat_id and user_id and custom_title

        monkeypatch.setattr("telegram.Bot.set_chat_administrator_custom_title",
                            make_assertion)
        assert await chat.set_administrator_custom_title(
            user_id=42, custom_title="custom_title")

    async def test_set_photo(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            photo = kwargs["photo"] == "test_photo"
            return chat_id, photo

        assert check_shortcut_signature(Chat.set_photo, Bot.set_chat_photo,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.set_photo, chat.get_bot(),
                                         "set_chat_photo")
        assert await check_defaults_handling(chat.set_photo, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "set_chat_photo", make_assertion)
        assert await chat.set_photo(photo="test_photo")

    async def test_delete_photo(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            return chat_id

        assert check_shortcut_signature(Chat.delete_photo,
                                        Bot.delete_chat_photo, ["chat_id"], [])
        assert await check_shortcut_call(chat.delete_photo, chat.get_bot(),
                                         "delete_chat_photo")
        assert await check_defaults_handling(chat.delete_photo, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "delete_chat_photo",
                            make_assertion)
        assert await chat.delete_photo()

    async def test_set_title(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            title = kwargs["title"] == "test_title"
            return chat_id, title

        assert check_shortcut_signature(Chat.set_title, Bot.set_chat_title,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.set_title, chat.get_bot(),
                                         "set_chat_title")
        assert await check_defaults_handling(chat.set_title, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "set_chat_title", make_assertion)
        assert await chat.set_title(title="test_title")

    async def test_set_description(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            description = kwargs["description"] == "test_descripton"
            return chat_id, description

        assert check_shortcut_signature(Chat.set_description,
                                        Bot.set_chat_description, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.set_description, chat.get_bot(),
                                         "set_chat_description")
        assert await check_defaults_handling(chat.set_description,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "set_chat_description",
                            make_assertion)
        assert await chat.set_description(description="test_description")

    async def test_pin_message(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs["message_id"] == 42

        assert check_shortcut_signature(Chat.pin_message, Bot.pin_chat_message,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.pin_message, chat.get_bot(),
                                         "pin_chat_message")
        assert await check_defaults_handling(chat.pin_message, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "pin_chat_message", make_assertion)
        assert await chat.pin_message(message_id=42)

    async def test_unpin_message(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.unpin_message,
                                        Bot.unpin_chat_message, ["chat_id"],
                                        [])
        assert await check_shortcut_call(chat.unpin_message, chat.get_bot(),
                                         "unpin_chat_message")
        assert await check_defaults_handling(chat.unpin_message,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "unpin_chat_message",
                            make_assertion)
        assert await chat.unpin_message()

    async def test_unpin_all_messages(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.unpin_all_messages,
                                        Bot.unpin_all_chat_messages,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.unpin_all_messages,
                                         chat.get_bot(),
                                         "unpin_all_chat_messages")
        assert await check_defaults_handling(chat.unpin_all_messages,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "unpin_all_chat_messages",
                            make_assertion)
        assert await chat.unpin_all_messages()

    async def test_instance_method_send_message(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs["text"] == "test"

        assert check_shortcut_signature(Chat.send_message, Bot.send_message,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_message, chat.get_bot(),
                                         "send_message")
        assert await check_defaults_handling(chat.send_message, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_message", make_assertion)
        assert await chat.send_message(text="test")

    async def test_instance_method_send_media_group(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "media"] == "test_media_group"

        assert check_shortcut_signature(Chat.send_media_group,
                                        Bot.send_media_group, ["chat_id"], [])
        assert await check_shortcut_call(chat.send_media_group, chat.get_bot(),
                                         "send_media_group")
        assert await check_defaults_handling(chat.send_media_group,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_media_group", make_assertion)
        assert await chat.send_media_group(media="test_media_group")

    async def test_instance_method_send_photo(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "photo"] == "test_photo"

        assert check_shortcut_signature(Chat.send_photo, Bot.send_photo,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_photo, chat.get_bot(),
                                         "send_photo")
        assert await check_defaults_handling(chat.send_photo, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_photo", make_assertion)
        assert await chat.send_photo(photo="test_photo")

    async def test_instance_method_send_contact(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "phone_number"] == "test_contact"

        assert check_shortcut_signature(Chat.send_contact, Bot.send_contact,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_contact, chat.get_bot(),
                                         "send_contact")
        assert await check_defaults_handling(chat.send_contact, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_contact", make_assertion)
        assert await chat.send_contact(phone_number="test_contact")

    async def test_instance_method_send_audio(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "audio"] == "test_audio"

        assert check_shortcut_signature(Chat.send_audio, Bot.send_audio,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_audio, chat.get_bot(),
                                         "send_audio")
        assert await check_defaults_handling(chat.send_audio, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_audio", make_assertion)
        assert await chat.send_audio(audio="test_audio")

    async def test_instance_method_send_document(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "document"] == "test_document"

        assert check_shortcut_signature(Chat.send_document, Bot.send_document,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_document, chat.get_bot(),
                                         "send_document")
        assert await check_defaults_handling(chat.send_document,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_document", make_assertion)
        assert await chat.send_document(document="test_document")

    async def test_instance_method_send_dice(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "emoji"] == "test_dice"

        assert check_shortcut_signature(Chat.send_dice, Bot.send_dice,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_dice, chat.get_bot(),
                                         "send_dice")
        assert await check_defaults_handling(chat.send_dice, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_dice", make_assertion)
        assert await chat.send_dice(emoji="test_dice")

    async def test_instance_method_send_game(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "game_short_name"] == "test_game"

        assert check_shortcut_signature(Chat.send_game, Bot.send_game,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_game, chat.get_bot(),
                                         "send_game")
        assert await check_defaults_handling(chat.send_game, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_game", make_assertion)
        assert await chat.send_game(game_short_name="test_game")

    async def test_instance_method_send_invoice(self, monkeypatch, chat):
        async 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"
            args = title and description and payload and provider_token and currency and prices
            return kwargs["chat_id"] == chat.id and args

        assert check_shortcut_signature(Chat.send_invoice, Bot.send_invoice,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_invoice, chat.get_bot(),
                                         "send_invoice")
        assert await check_defaults_handling(chat.send_invoice, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_invoice", make_assertion)
        assert await chat.send_invoice(
            "title",
            "description",
            "payload",
            "provider_token",
            "currency",
            "prices",
        )

    async def test_instance_method_send_location(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "latitude"] == "test_location"

        assert check_shortcut_signature(Chat.send_location, Bot.send_location,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_location, chat.get_bot(),
                                         "send_location")
        assert await check_defaults_handling(chat.send_location,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_location", make_assertion)
        assert await chat.send_location(latitude="test_location")

    async def test_instance_method_send_sticker(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "sticker"] == "test_sticker"

        assert check_shortcut_signature(Chat.send_sticker, Bot.send_sticker,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_sticker, chat.get_bot(),
                                         "send_sticker")
        assert await check_defaults_handling(chat.send_sticker, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_sticker", make_assertion)
        assert await chat.send_sticker(sticker="test_sticker")

    async def test_instance_method_send_venue(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "title"] == "test_venue"

        assert check_shortcut_signature(Chat.send_venue, Bot.send_venue,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_venue, chat.get_bot(),
                                         "send_venue")
        assert await check_defaults_handling(chat.send_venue, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_venue", make_assertion)
        assert await chat.send_venue(title="test_venue")

    async def test_instance_method_send_video(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "video"] == "test_video"

        assert check_shortcut_signature(Chat.send_video, Bot.send_video,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_video, chat.get_bot(),
                                         "send_video")
        assert await check_defaults_handling(chat.send_video, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_video", make_assertion)
        assert await chat.send_video(video="test_video")

    async def test_instance_method_send_video_note(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "video_note"] == "test_video_note"

        assert check_shortcut_signature(Chat.send_video_note,
                                        Bot.send_video_note, ["chat_id"], [])
        assert await check_shortcut_call(chat.send_video_note, chat.get_bot(),
                                         "send_video_note")
        assert await check_defaults_handling(chat.send_video_note,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_video_note", make_assertion)
        assert await chat.send_video_note(video_note="test_video_note")

    async def test_instance_method_send_voice(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "voice"] == "test_voice"

        assert check_shortcut_signature(Chat.send_voice, Bot.send_voice,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_voice, chat.get_bot(),
                                         "send_voice")
        assert await check_defaults_handling(chat.send_voice, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_voice", make_assertion)
        assert await chat.send_voice(voice="test_voice")

    async def test_instance_method_send_animation(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "animation"] == "test_animation"

        assert check_shortcut_signature(Chat.send_animation,
                                        Bot.send_animation, ["chat_id"], [])
        assert await check_shortcut_call(chat.send_animation, chat.get_bot(),
                                         "send_animation")
        assert await check_defaults_handling(chat.send_animation,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_animation", make_assertion)
        assert await chat.send_animation(animation="test_animation")

    async def test_instance_method_send_poll(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "question"] == "test_poll"

        assert check_shortcut_signature(Chat.send_poll, Bot.send_poll,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.send_poll, chat.get_bot(),
                                         "send_poll")
        assert await check_defaults_handling(chat.send_poll, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "send_poll", make_assertion)
        assert await chat.send_poll(question="test_poll", options=[1, 2])

    async def test_instance_method_send_copy(self, monkeypatch, chat):
        async 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
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.send_copy, Bot.copy_message,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.copy_message, chat.get_bot(),
                                         "copy_message")
        assert await check_defaults_handling(chat.copy_message, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "copy_message", make_assertion)
        assert await chat.send_copy(from_chat_id="test_copy", message_id=42)

    async def test_instance_method_copy_message(self, monkeypatch, chat):
        async 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"
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.copy_message, Bot.copy_message,
                                        ["from_chat_id"], [])
        assert await check_shortcut_call(chat.copy_message, chat.get_bot(),
                                         "copy_message")
        assert await check_defaults_handling(chat.copy_message, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "copy_message", make_assertion)
        assert await chat.copy_message(chat_id="test_copy", message_id=42)

    async def test_instance_method_forward_from(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            chat_id = kwargs["chat_id"] == chat.id
            message_id = kwargs["message_id"] == 42
            from_chat_id = kwargs["from_chat_id"] == "test_forward"
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.forward_from, Bot.forward_message,
                                        ["chat_id"], [])
        assert await check_shortcut_call(chat.forward_from, chat.get_bot(),
                                         "forward_message")
        assert await check_defaults_handling(chat.forward_from, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "forward_message", make_assertion)
        assert await chat.forward_from(from_chat_id="test_forward",
                                       message_id=42)

    async def test_instance_method_forward_to(self, monkeypatch, chat):
        async 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_forward"
            return from_chat_id and message_id and chat_id

        assert check_shortcut_signature(Chat.forward_to, Bot.forward_message,
                                        ["from_chat_id"], [])
        assert await check_shortcut_call(chat.forward_to, chat.get_bot(),
                                         "forward_message")
        assert await check_defaults_handling(chat.forward_to, chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "forward_message", make_assertion)
        assert await chat.forward_to(chat_id="test_forward", message_id=42)

    async def test_export_invite_link(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.export_invite_link,
                                         chat.get_bot(),
                                         "export_chat_invite_link")
        assert await check_defaults_handling(chat.export_invite_link,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "export_chat_invite_link",
                            make_assertion)
        assert await chat.export_invite_link()

    async def test_create_invite_link(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.create_invite_link,
                                         chat.get_bot(),
                                         "create_chat_invite_link")
        assert await check_defaults_handling(chat.create_invite_link,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "create_chat_invite_link",
                            make_assertion)
        assert await chat.create_invite_link()

    async def test_edit_invite_link(self, monkeypatch, chat):
        link = "ThisIsALink"

        async 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 await check_shortcut_call(chat.edit_invite_link, chat.get_bot(),
                                         "edit_chat_invite_link")
        assert await check_defaults_handling(chat.edit_invite_link,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "edit_chat_invite_link",
                            make_assertion)
        assert await chat.edit_invite_link(invite_link=link)

    async def test_revoke_invite_link(self, monkeypatch, chat):
        link = "ThisIsALink"

        async 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 await check_shortcut_call(chat.revoke_invite_link,
                                         chat.get_bot(),
                                         "revoke_chat_invite_link")
        assert await check_defaults_handling(chat.revoke_invite_link,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "revoke_chat_invite_link",
                            make_assertion)
        assert await chat.revoke_invite_link(invite_link=link)

    async def test_instance_method_get_menu_button(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id

        assert check_shortcut_signature(Chat.get_menu_button,
                                        Bot.get_chat_menu_button, ["chat_id"],
                                        [])
        assert await check_shortcut_call(
            chat.get_menu_button,
            chat.get_bot(),
            "get_chat_menu_button",
            shortcut_kwargs=["chat_id"],
        )
        assert await check_defaults_handling(chat.get_menu_button,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "get_chat_menu_button",
                            make_assertion)
        assert await chat.get_menu_button()

    async def test_instance_method_set_menu_button(self, monkeypatch, chat):
        async def make_assertion(*_, **kwargs):
            return kwargs["chat_id"] == chat.id and kwargs[
                "menu_button"] == "menu_button"

        assert check_shortcut_signature(Chat.set_menu_button,
                                        Bot.set_chat_menu_button, ["chat_id"],
                                        [])
        assert await check_shortcut_call(
            chat.set_menu_button,
            chat.get_bot(),
            "set_chat_menu_button",
            shortcut_kwargs=["chat_id"],
        )
        assert await check_defaults_handling(chat.set_menu_button,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "set_chat_menu_button",
                            make_assertion)
        assert await chat.set_menu_button(menu_button="menu_button")

    async def test_approve_join_request(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.approve_join_request,
                                         chat.get_bot(),
                                         "approve_chat_join_request")
        assert await check_defaults_handling(chat.approve_join_request,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "approve_chat_join_request",
                            make_assertion)
        assert await chat.approve_join_request(user_id=42)

    async def test_decline_join_request(self, monkeypatch, chat):
        async 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 await check_shortcut_call(chat.decline_join_request,
                                         chat.get_bot(),
                                         "decline_chat_join_request")
        assert await check_defaults_handling(chat.decline_join_request,
                                             chat.get_bot())

        monkeypatch.setattr(chat.get_bot(), "decline_chat_join_request",
                            make_assertion)
        assert await chat.decline_join_request(user_id=42)

    def test_equality(self):
        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)

        assert a == b
        assert hash(a) == hash(b)
        assert a is not b

        assert a == c
        assert hash(a) == hash(c)

        assert a != d
        assert hash(a) != hash(d)

        assert a != e
        assert hash(a) != hash(e)
예제 #6
0
def chat_location(bot):
    return ChatLocation(TestChatLocation.location, TestChatLocation.address)