#!/usr/bin/env python # # A library that provides a Python interface to the Telegram Bot API # Copyright (C) 2015-2024 # Leandro Toledo de Souza # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser Public License for more details. # # You should have received a copy of the GNU Lesser Public License # along with this program. If not, see [http://www.gnu.org/licenses/]. import pytest from telegram import Bot, InlineKeyboardButton, Update, User from telegram.helpers import escape_markdown from tests.auxil.bot_method_checks import ( check_defaults_handling, check_shortcut_call, check_shortcut_signature, ) from tests.auxil.slots import mro_slots @pytest.fixture(scope="module") def json_dict(): return { "id": UserTestBase.id_, "is_bot": UserTestBase.is_bot, "first_name": UserTestBase.first_name, "last_name": UserTestBase.last_name, "username": UserTestBase.username, "language_code": UserTestBase.language_code, "can_join_groups": UserTestBase.can_join_groups, "can_read_all_group_messages": UserTestBase.can_read_all_group_messages, "supports_inline_queries": UserTestBase.supports_inline_queries, "is_premium": UserTestBase.is_premium, "added_to_attachment_menu": UserTestBase.added_to_attachment_menu, "can_connect_to_business": UserTestBase.can_connect_to_business, "has_main_web_app": UserTestBase.has_main_web_app, } @pytest.fixture def user(bot): user = User( id=UserTestBase.id_, first_name=UserTestBase.first_name, is_bot=UserTestBase.is_bot, last_name=UserTestBase.last_name, username=UserTestBase.username, language_code=UserTestBase.language_code, can_join_groups=UserTestBase.can_join_groups, can_read_all_group_messages=UserTestBase.can_read_all_group_messages, supports_inline_queries=UserTestBase.supports_inline_queries, is_premium=UserTestBase.is_premium, added_to_attachment_menu=UserTestBase.added_to_attachment_menu, can_connect_to_business=UserTestBase.can_connect_to_business, has_main_web_app=UserTestBase.has_main_web_app, ) user.set_bot(bot) user._unfreeze() return user class UserTestBase: id_ = 1 is_bot = True first_name = "first\u2022name" last_name = "last\u2022name" username = "username" language_code = "en_us" can_join_groups = True can_read_all_group_messages = True supports_inline_queries = False is_premium = True added_to_attachment_menu = False can_connect_to_business = True has_main_web_app = False class TestUserWithoutRequest(UserTestBase): def test_slot_behaviour(self, user): for attr in user.__slots__: assert getattr(user, attr, "err") != "err", f"got extra slot '{attr}'" assert len(mro_slots(user)) == len(set(mro_slots(user))), "duplicate slot" def test_de_json(self, json_dict, bot): user = User.de_json(json_dict, bot) assert user.api_kwargs == {} assert user.id == self.id_ assert user.is_bot == self.is_bot assert user.first_name == self.first_name assert user.last_name == self.last_name assert user.username == self.username assert user.language_code == self.language_code assert user.can_join_groups == self.can_join_groups assert user.can_read_all_group_messages == self.can_read_all_group_messages assert user.supports_inline_queries == self.supports_inline_queries assert user.is_premium == self.is_premium assert user.added_to_attachment_menu == self.added_to_attachment_menu assert user.can_connect_to_business == self.can_connect_to_business assert user.has_main_web_app == self.has_main_web_app def test_to_dict(self, user): user_dict = user.to_dict() assert isinstance(user_dict, dict) assert user_dict["id"] == user.id assert user_dict["is_bot"] == user.is_bot assert user_dict["first_name"] == user.first_name assert user_dict["last_name"] == user.last_name assert user_dict["username"] == user.username assert user_dict["language_code"] == user.language_code assert user_dict["can_join_groups"] == user.can_join_groups assert user_dict["can_read_all_group_messages"] == user.can_read_all_group_messages assert user_dict["supports_inline_queries"] == user.supports_inline_queries assert user_dict["is_premium"] == user.is_premium assert user_dict["added_to_attachment_menu"] == user.added_to_attachment_menu assert user_dict["can_connect_to_business"] == user.can_connect_to_business assert user_dict["has_main_web_app"] == user.has_main_web_app def test_equality(self): a = User(self.id_, self.first_name, self.is_bot, self.last_name) b = User(self.id_, self.first_name, self.is_bot, self.last_name) c = User(self.id_, self.first_name, self.is_bot) d = User(0, self.first_name, self.is_bot, self.last_name) e = Update(self.id_) 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) def test_name(self, user): assert user.name == "@username" user.username = None assert user.name == "first\u2022name last\u2022name" user.last_name = None assert user.name == "first\u2022name" user.username = self.username assert user.name == "@username" def test_full_name(self, user): assert user.full_name == "first\u2022name last\u2022name" user.last_name = None assert user.full_name == "first\u2022name" def test_link(self, user): assert user.link == f"https://t.me/{user.username}" user.username = None assert user.link is None async def test_instance_method_get_profile_photos(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["user_id"] == user.id assert check_shortcut_signature( User.get_profile_photos, Bot.get_user_profile_photos, ["user_id"], [] ) assert await check_shortcut_call( user.get_profile_photos, user.get_bot(), "get_user_profile_photos" ) assert await check_defaults_handling(user.get_profile_photos, user.get_bot()) monkeypatch.setattr(user.get_bot(), "get_user_profile_photos", make_assertion) assert await user.get_profile_photos() async def test_instance_method_pin_message(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id assert check_shortcut_signature(User.pin_message, Bot.pin_chat_message, ["chat_id"], []) assert await check_shortcut_call(user.pin_message, user.get_bot(), "pin_chat_message") assert await check_defaults_handling(user.pin_message, user.get_bot()) monkeypatch.setattr(user.get_bot(), "pin_chat_message", make_assertion) assert await user.pin_message(1) async def test_instance_method_unpin_message(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id assert check_shortcut_signature( User.unpin_message, Bot.unpin_chat_message, ["chat_id"], [] ) assert await check_shortcut_call(user.unpin_message, user.get_bot(), "unpin_chat_message") assert await check_defaults_handling(user.unpin_message, user.get_bot()) monkeypatch.setattr(user.get_bot(), "unpin_chat_message", make_assertion) assert await user.unpin_message() async def test_instance_method_unpin_all_messages(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id assert check_shortcut_signature( User.unpin_all_messages, Bot.unpin_all_chat_messages, ["chat_id"], [] ) assert await check_shortcut_call( user.unpin_all_messages, user.get_bot(), "unpin_all_chat_messages" ) assert await check_defaults_handling(user.unpin_all_messages, user.get_bot()) monkeypatch.setattr(user.get_bot(), "unpin_all_chat_messages", make_assertion) assert await user.unpin_all_messages() async def test_instance_method_send_message(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["text"] == "test" assert check_shortcut_signature(User.send_message, Bot.send_message, ["chat_id"], []) assert await check_shortcut_call(user.send_message, user.get_bot(), "send_message") assert await check_defaults_handling(user.send_message, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_message", make_assertion) assert await user.send_message("test") async def test_instance_method_send_photo(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["photo"] == "test_photo" assert check_shortcut_signature(User.send_photo, Bot.send_photo, ["chat_id"], []) assert await check_shortcut_call(user.send_photo, user.get_bot(), "send_photo") assert await check_defaults_handling(user.send_photo, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_photo", make_assertion) assert await user.send_photo("test_photo") async def test_instance_method_send_media_group(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["media"] == "test_media_group" assert check_shortcut_signature( User.send_media_group, Bot.send_media_group, ["chat_id"], [] ) assert await check_shortcut_call(user.send_media_group, user.get_bot(), "send_media_group") assert await check_defaults_handling(user.send_media_group, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_media_group", make_assertion) assert await user.send_media_group("test_media_group") async def test_instance_method_send_audio(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["audio"] == "test_audio" assert check_shortcut_signature(User.send_audio, Bot.send_audio, ["chat_id"], []) assert await check_shortcut_call(user.send_audio, user.get_bot(), "send_audio") assert await check_defaults_handling(user.send_audio, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_audio", make_assertion) assert await user.send_audio("test_audio") async def test_instance_method_send_chat_action(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["action"] == "test_chat_action" assert check_shortcut_signature( User.send_chat_action, Bot.send_chat_action, ["chat_id"], [] ) assert await check_shortcut_call(user.send_chat_action, user.get_bot(), "send_chat_action") assert await check_defaults_handling(user.send_chat_action, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_chat_action", make_assertion) assert await user.send_chat_action("test_chat_action") async def test_instance_method_send_contact(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["phone_number"] == "test_contact" assert check_shortcut_signature(User.send_contact, Bot.send_contact, ["chat_id"], []) assert await check_shortcut_call(user.send_contact, user.get_bot(), "send_contact") assert await check_defaults_handling(user.send_contact, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_contact", make_assertion) assert await user.send_contact(phone_number="test_contact") async def test_instance_method_send_dice(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["emoji"] == "test_dice" assert check_shortcut_signature(User.send_dice, Bot.send_dice, ["chat_id"], []) assert await check_shortcut_call(user.send_dice, user.get_bot(), "send_dice") assert await check_defaults_handling(user.send_dice, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_dice", make_assertion) assert await user.send_dice(emoji="test_dice") async def test_instance_method_send_document(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["document"] == "test_document" assert check_shortcut_signature(User.send_document, Bot.send_document, ["chat_id"], []) assert await check_shortcut_call(user.send_document, user.get_bot(), "send_document") assert await check_defaults_handling(user.send_document, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_document", make_assertion) assert await user.send_document("test_document") async def test_instance_method_send_game(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["game_short_name"] == "test_game" assert check_shortcut_signature(User.send_game, Bot.send_game, ["chat_id"], []) assert await check_shortcut_call(user.send_game, user.get_bot(), "send_game") assert await check_defaults_handling(user.send_game, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_game", make_assertion) assert await user.send_game(game_short_name="test_game") async def test_instance_method_send_invoice(self, monkeypatch, user): 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"] == user.id and args assert check_shortcut_signature(User.send_invoice, Bot.send_invoice, ["chat_id"], []) assert await check_shortcut_call(user.send_invoice, user.get_bot(), "send_invoice") assert await check_defaults_handling(user.send_invoice, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_invoice", make_assertion) assert await user.send_invoice( "title", "description", "payload", "provider_token", "currency", "prices", ) async def test_instance_method_send_location(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["latitude"] == "test_location" assert check_shortcut_signature(User.send_location, Bot.send_location, ["chat_id"], []) assert await check_shortcut_call(user.send_location, user.get_bot(), "send_location") assert await check_defaults_handling(user.send_location, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_location", make_assertion) assert await user.send_location("test_location") async def test_instance_method_send_sticker(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["sticker"] == "test_sticker" assert check_shortcut_signature(User.send_sticker, Bot.send_sticker, ["chat_id"], []) assert await check_shortcut_call(user.send_sticker, user.get_bot(), "send_sticker") assert await check_defaults_handling(user.send_sticker, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_sticker", make_assertion) assert await user.send_sticker("test_sticker") async def test_instance_method_send_video(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["video"] == "test_video" assert check_shortcut_signature(User.send_video, Bot.send_video, ["chat_id"], []) assert await check_shortcut_call(user.send_video, user.get_bot(), "send_video") assert await check_defaults_handling(user.send_video, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_video", make_assertion) assert await user.send_video("test_video") async def test_instance_method_send_venue(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["title"] == "test_venue" assert check_shortcut_signature(User.send_venue, Bot.send_venue, ["chat_id"], []) assert await check_shortcut_call(user.send_venue, user.get_bot(), "send_venue") assert await check_defaults_handling(user.send_venue, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_venue", make_assertion) assert await user.send_venue(title="test_venue") async def test_instance_method_send_video_note(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["video_note"] == "test_video_note" assert check_shortcut_signature(User.send_video_note, Bot.send_video_note, ["chat_id"], []) assert await check_shortcut_call(user.send_video_note, user.get_bot(), "send_video_note") assert await check_defaults_handling(user.send_video_note, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_video_note", make_assertion) assert await user.send_video_note("test_video_note") async def test_instance_method_send_voice(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["voice"] == "test_voice" assert check_shortcut_signature(User.send_voice, Bot.send_voice, ["chat_id"], []) assert await check_shortcut_call(user.send_voice, user.get_bot(), "send_voice") assert await check_defaults_handling(user.send_voice, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_voice", make_assertion) assert await user.send_voice("test_voice") async def test_instance_method_send_animation(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["animation"] == "test_animation" assert check_shortcut_signature(User.send_animation, Bot.send_animation, ["chat_id"], []) assert await check_shortcut_call(user.send_animation, user.get_bot(), "send_animation") assert await check_defaults_handling(user.send_animation, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_animation", make_assertion) assert await user.send_animation("test_animation") async def test_instance_method_send_poll(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["question"] == "test_poll" assert check_shortcut_signature(User.send_poll, Bot.send_poll, ["chat_id"], []) assert await check_shortcut_call(user.send_poll, user.get_bot(), "send_poll") assert await check_defaults_handling(user.send_poll, user.get_bot()) monkeypatch.setattr(user.get_bot(), "send_poll", make_assertion) assert await user.send_poll(question="test_poll", options=[1, 2]) async def test_instance_method_send_copy(self, monkeypatch, user): async def make_assertion(*_, **kwargs): user_id = kwargs["chat_id"] == user.id message_id = kwargs["message_id"] == "message_id" from_chat_id = kwargs["from_chat_id"] == "from_chat_id" return from_chat_id and message_id and user_id assert check_shortcut_signature(User.send_copy, Bot.copy_message, ["chat_id"], []) assert await check_shortcut_call(user.send_copy, user.get_bot(), "copy_message") assert await check_defaults_handling(user.send_copy, user.get_bot()) monkeypatch.setattr(user.get_bot(), "copy_message", make_assertion) assert await user.send_copy(from_chat_id="from_chat_id", message_id="message_id") async def test_instance_method_copy_message(self, monkeypatch, user): async def make_assertion(*_, **kwargs): chat_id = kwargs["chat_id"] == "chat_id" message_id = kwargs["message_id"] == "message_id" user_id = kwargs["from_chat_id"] == user.id return chat_id and message_id and user_id assert check_shortcut_signature(User.copy_message, Bot.copy_message, ["from_chat_id"], []) assert await check_shortcut_call(user.copy_message, user.get_bot(), "copy_message") assert await check_defaults_handling(user.copy_message, user.get_bot()) monkeypatch.setattr(user.get_bot(), "copy_message", make_assertion) assert await user.copy_message(chat_id="chat_id", message_id="message_id") async def test_instance_method_get_user_chat_boosts(self, monkeypatch, user): async def make_assertion(*_, **kwargs): chat_id = kwargs["chat_id"] == "chat_id" user_id = kwargs["user_id"] == user.id return chat_id and user_id assert check_shortcut_signature( User.get_chat_boosts, Bot.get_user_chat_boosts, ["user_id"], [] ) assert await check_shortcut_call( user.get_chat_boosts, user.get_bot(), "get_user_chat_boosts" ) assert await check_defaults_handling(user.get_chat_boosts, user.get_bot()) monkeypatch.setattr(user.get_bot(), "get_user_chat_boosts", make_assertion) assert await user.get_chat_boosts(chat_id="chat_id") async def test_instance_method_get_menu_button(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id assert check_shortcut_signature( User.get_menu_button, Bot.get_chat_menu_button, ["chat_id"], [] ) assert await check_shortcut_call( user.get_menu_button, user.get_bot(), "get_chat_menu_button", shortcut_kwargs=["chat_id"], ) assert await check_defaults_handling(user.get_menu_button, user.get_bot()) monkeypatch.setattr(user.get_bot(), "get_chat_menu_button", make_assertion) assert await user.get_menu_button() async def test_instance_method_set_menu_button(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["menu_button"] == "menu_button" assert check_shortcut_signature( User.set_menu_button, Bot.set_chat_menu_button, ["chat_id"], [] ) assert await check_shortcut_call( user.set_menu_button, user.get_bot(), "set_chat_menu_button", shortcut_kwargs=["chat_id"], ) assert await check_defaults_handling(user.set_menu_button, user.get_bot()) monkeypatch.setattr(user.get_bot(), "set_chat_menu_button", make_assertion) assert await user.set_menu_button(menu_button="menu_button") async def test_instance_method_approve_join_request(self, monkeypatch, user): async def make_assertion(*_, **kwargs): chat_id = kwargs["chat_id"] == "chat_id" user_id = kwargs["user_id"] == user.id return chat_id and user_id assert check_shortcut_signature( User.approve_join_request, Bot.approve_chat_join_request, ["user_id"], [] ) assert await check_shortcut_call( user.approve_join_request, user.get_bot(), "approve_chat_join_request" ) assert await check_defaults_handling(user.approve_join_request, user.get_bot()) monkeypatch.setattr(user.get_bot(), "approve_chat_join_request", make_assertion) assert await user.approve_join_request(chat_id="chat_id") async def test_instance_method_decline_join_request(self, monkeypatch, user): async def make_assertion(*_, **kwargs): chat_id = kwargs["chat_id"] == "chat_id" user_id = kwargs["user_id"] == user.id return chat_id and user_id assert check_shortcut_signature( User.decline_join_request, Bot.decline_chat_join_request, ["user_id"], [] ) assert await check_shortcut_call( user.decline_join_request, user.get_bot(), "decline_chat_join_request" ) assert await check_defaults_handling(user.decline_join_request, user.get_bot()) monkeypatch.setattr(user.get_bot(), "decline_chat_join_request", make_assertion) assert await user.decline_join_request(chat_id="chat_id") async def test_mention_html(self, user): expected = '{}' assert user.mention_html() == expected.format(user.id, user.full_name) assert user.mention_html("thename\u2022") == expected.format( user.id, "the<b>name\u2022" ) assert user.mention_html(user.username) == expected.format(user.id, user.username) def test_mention_button(self, user): expected_name = InlineKeyboardButton(text="Bob", url=f"tg://user?id={user.id}") expected_full = InlineKeyboardButton(text=user.full_name, url=f"tg://user?id={user.id}") assert user.mention_button("Bob") == expected_name assert user.mention_button() == expected_full def test_mention_markdown(self, user): expected = "[{}](tg://user?id={})" assert user.mention_markdown() == expected.format(user.full_name, user.id) assert user.mention_markdown("the_name*\u2022") == expected.format( "the_name*\u2022", user.id ) assert user.mention_markdown(user.username) == expected.format(user.username, user.id) async def test_mention_markdown_v2(self, user): user.first_name = "first{name" user.last_name = "last_name" expected = "[{}](tg://user?id={})" assert user.mention_markdown_v2() == expected.format( escape_markdown(user.full_name, version=2), user.id ) assert user.mention_markdown_v2("the{name>\u2022") == expected.format( "the\\{name\\>\u2022", user.id ) assert user.mention_markdown_v2(user.username) == expected.format(user.username, user.id) async def test_delete_message(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["message_id"] == 42 assert check_shortcut_signature(user.delete_message, Bot.delete_message, ["chat_id"], []) assert await check_shortcut_call(user.delete_message, user.get_bot(), "delete_message") assert await check_defaults_handling(user.delete_message, user.get_bot()) monkeypatch.setattr(user.get_bot(), "delete_message", make_assertion) assert await user.delete_message(message_id=42) async def test_delete_messages(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["chat_id"] == user.id and kwargs["message_ids"] == (42, 43) assert check_shortcut_signature(user.delete_messages, Bot.delete_messages, ["chat_id"], []) assert await check_shortcut_call(user.delete_messages, user.get_bot(), "delete_messages") assert await check_defaults_handling(user.delete_messages, user.get_bot()) monkeypatch.setattr(user.get_bot(), "delete_messages", make_assertion) assert await user.delete_messages(message_ids=(42, 43)) async def test_instance_method_send_copies(self, monkeypatch, user): async def make_assertion(*_, **kwargs): from_chat_id = kwargs["from_chat_id"] == "test_copies" message_ids = kwargs["message_ids"] == (42, 43) user_id = kwargs["chat_id"] == user.id return from_chat_id and message_ids and user_id assert check_shortcut_signature(user.send_copies, Bot.copy_messages, ["chat_id"], []) assert await check_shortcut_call(user.send_copies, user.get_bot(), "copy_messages") assert await check_defaults_handling(user.send_copies, user.get_bot()) monkeypatch.setattr(user.get_bot(), "copy_messages", make_assertion) assert await user.send_copies(from_chat_id="test_copies", message_ids=(42, 43)) async def test_instance_method_copy_messages(self, monkeypatch, user): async def make_assertion(*_, **kwargs): from_chat_id = kwargs["from_chat_id"] == user.id message_ids = kwargs["message_ids"] == (42, 43) user_id = kwargs["chat_id"] == "test_copies" return from_chat_id and message_ids and user_id assert check_shortcut_signature( user.copy_messages, Bot.copy_messages, ["from_chat_id"], [] ) assert await check_shortcut_call(user.copy_messages, user.get_bot(), "copy_messages") assert await check_defaults_handling(user.copy_messages, user.get_bot()) monkeypatch.setattr(user.get_bot(), "copy_messages", make_assertion) assert await user.copy_messages(chat_id="test_copies", message_ids=(42, 43)) async def test_instance_method_forward_from(self, monkeypatch, user): async def make_assertion(*_, **kwargs): user_id = kwargs["chat_id"] == user.id message_id = kwargs["message_id"] == 42 from_chat_id = kwargs["from_chat_id"] == "test_forward" return from_chat_id and message_id and user_id assert check_shortcut_signature(user.forward_from, Bot.forward_message, ["chat_id"], []) assert await check_shortcut_call(user.forward_from, user.get_bot(), "forward_message") assert await check_defaults_handling(user.forward_from, user.get_bot()) monkeypatch.setattr(user.get_bot(), "forward_message", make_assertion) assert await user.forward_from(from_chat_id="test_forward", message_id=42) async def test_instance_method_forward_to(self, monkeypatch, user): async def make_assertion(*_, **kwargs): from_chat_id = kwargs["from_chat_id"] == user.id message_id = kwargs["message_id"] == 42 user_id = kwargs["chat_id"] == "test_forward" return from_chat_id and message_id and user_id assert check_shortcut_signature(user.forward_to, Bot.forward_message, ["from_chat_id"], []) assert await check_shortcut_call(user.forward_to, user.get_bot(), "forward_message") assert await check_defaults_handling(user.forward_to, user.get_bot()) monkeypatch.setattr(user.get_bot(), "forward_message", make_assertion) assert await user.forward_to(chat_id="test_forward", message_id=42) async def test_instance_method_forward_messages_from(self, monkeypatch, user): async def make_assertion(*_, **kwargs): user_id = kwargs["chat_id"] == user.id message_ids = kwargs["message_ids"] == (42, 43) from_chat_id = kwargs["from_chat_id"] == "test_forwards" return from_chat_id and message_ids and user_id assert check_shortcut_signature( user.forward_messages_from, Bot.forward_messages, ["chat_id"], [] ) assert await check_shortcut_call( user.forward_messages_from, user.get_bot(), "forward_messages" ) assert await check_defaults_handling(user.forward_messages_from, user.get_bot()) monkeypatch.setattr(user.get_bot(), "forward_messages", make_assertion) assert await user.forward_messages_from(from_chat_id="test_forwards", message_ids=(42, 43)) async def test_instance_method_forward_messages_to(self, monkeypatch, user): async def make_assertion(*_, **kwargs): from_chat_id = kwargs["from_chat_id"] == user.id message_ids = kwargs["message_ids"] == (42, 43) user_id = kwargs["chat_id"] == "test_forwards" return from_chat_id and message_ids and user_id assert check_shortcut_signature( user.forward_messages_to, Bot.forward_messages, ["from_chat_id"], [] ) assert await check_shortcut_call( user.forward_messages_to, user.get_bot(), "forward_messages" ) assert await check_defaults_handling(user.forward_messages_to, user.get_bot()) monkeypatch.setattr(user.get_bot(), "forward_messages", make_assertion) assert await user.forward_messages_to(chat_id="test_forwards", message_ids=(42, 43)) async def test_instance_method_refund_star_payment(self, monkeypatch, user): async def make_assertion(*_, **kwargs): return kwargs["user_id"] == user.id and kwargs["telegram_payment_charge_id"] == 42 assert check_shortcut_signature( user.refund_star_payment, Bot.refund_star_payment, ["user_id"], [] ) assert await check_shortcut_call( user.refund_star_payment, user.get_bot(), "refund_star_payment" ) assert await check_defaults_handling(user.refund_star_payment, user.get_bot()) monkeypatch.setattr(user.get_bot(), "refund_star_payment", make_assertion) assert await user.refund_star_payment(telegram_payment_charge_id=42)