python-telegram-bot/tests/test_user.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

745 lines
35 KiB
Python
Raw Normal View History

2016-05-15 02:39:11 +02:00
#!/usr/bin/env python
2015-09-07 19:03:20 +02:00
#
# A library that provides a Python interface to the Telegram Bot API
2024-02-19 20:06:25 +01:00
# Copyright (C) 2015-2024
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
2015-09-07 19:03:20 +02:00
#
# 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
2015-09-07 19:03:20 +02:00
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser Public License for more details.
2015-09-07 19:03:20 +02:00
#
# You should have received a copy of the GNU Lesser Public License
2015-09-07 19:03:20 +02:00
# along with this program. If not, see [http://www.gnu.org/licenses/].
import pytest
2015-09-07 19:03:20 +02:00
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,
)
2023-02-22 20:19:46 +01:00
from tests.auxil.slots import mro_slots
@pytest.fixture(scope="module")
def json_dict():
return {
2024-09-03 05:24:25 +02:00
"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,
}
2015-09-07 19:03:20 +02:00
@pytest.fixture
def user(bot):
user = User(
2024-09-03 05:24:25 +02:00
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
2015-09-07 19:03:20 +02:00
2024-09-03 05:24:25 +02:00
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
2024-08-02 13:43:27 +02:00
has_main_web_app = False
2015-09-07 19:03:20 +02:00
2024-09-03 05:24:25 +02:00
class TestUserWithoutRequest(UserTestBase):
2023-02-22 20:19:46 +01:00
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, offline_bot):
user = User.de_json(json_dict, offline_bot)
assert user.api_kwargs == {}
2015-09-07 19:03:20 +02:00
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
2024-08-02 13:43:27 +02:00
assert user.has_main_web_app == self.has_main_web_app
2015-09-07 19:03:20 +02:00
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
2024-08-02 13:43:27 +02:00
assert user_dict["has_main_web_app"] == user.has_main_web_app
2015-09-07 19:03:20 +02:00
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_)
2015-09-07 19:03:20 +02:00
assert a == b
assert hash(a) == hash(b)
assert a is not b
2015-09-07 19:03:20 +02:00
assert a == c
assert hash(a) == hash(c)
2015-09-07 19:03:20 +02:00
assert a != d
assert hash(a) != hash(d)
2015-09-07 19:03:20 +02:00
assert a != e
assert hash(a) != hash(e)
2015-09-07 19:03:20 +02:00
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"
2015-09-08 01:11:02 +02:00
def test_link(self, user):
2020-11-23 22:09:29 +01:00
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())
Bot API 4.0 (#1168) Telegram Passport (#1174): - Add full support for telegram passport. - New types: PassportData, PassportFile, EncryptedPassportElement, EncryptedCredentials, PassportElementError, PassportElementErrorDataField, PassportElementErrorFrontSide, PassportElementErrorReverseSide, PassportElementErrorSelfie, PassportElementErrorFile and PassportElementErrorFiles. - New bot method: set_passport_data_errors - New filter: Filters.passport_data - Field passport_data field on Message - PassportData is automagically decrypted when you specify your private key when creating Updater or Bot. - PassportFiles is also automagically decrypted as you download/retrieve them. - See new passportbot.py example for details on how to use, or go to our telegram passport wiki page for more info - NOTE: Passport decryption requires new dependency `cryptography`. Inputfile rework (#1184): - Change how Inputfile is handled internally - This allows support for specifying the thumbnails of photos and videos using the thumb= argument in the different send_ methods. - Also allows Bot.send_media_group to actually finally send more than one media. - Add thumb to Audio, Video and Videonote - Add Bot.edit_message_media together with InputMediaAnimation, InputMediaAudio, and inputMediaDocument. Other Bot API 4.0 changes: - Add forusquare_type to Venue, InlineQueryResultVenue, InputVenueMessageContent, and Bot.send_venue. (#1170) - Add vCard support by adding vcard field to Contact, InlineQueryResultContact, InputContactMessageContent, and Bot.send_contact. (#1166) - Support new message entities: CASHTAG and PHONE_NUMBER. (#1179) - Cashtag seems to be things like $USD and $GBP, but it seems telegram doesn't currently send them to bots. - Phone number also seems to have limited support for now - Add Bot.send_animation, add width, height, and duration to Animation, and add Filters.animation. (#1172) Co-authored-by: Jasmin Bom <jsmnbom@gmail.com> Co-authored-by: code1mountain <32801117+code1mountain@users.noreply.github.com> Co-authored-by: Eldinnie <pieter.schutz+github@gmail.com> Co-authored-by: mathefreak1 <mathefreak@hi2.in>
2018-08-29 14:18:58 +02:00
monkeypatch.setattr(user.get_bot(), "send_animation", make_assertion)
assert await user.send_animation("test_animation")
Bot API 4.0 (#1168) Telegram Passport (#1174): - Add full support for telegram passport. - New types: PassportData, PassportFile, EncryptedPassportElement, EncryptedCredentials, PassportElementError, PassportElementErrorDataField, PassportElementErrorFrontSide, PassportElementErrorReverseSide, PassportElementErrorSelfie, PassportElementErrorFile and PassportElementErrorFiles. - New bot method: set_passport_data_errors - New filter: Filters.passport_data - Field passport_data field on Message - PassportData is automagically decrypted when you specify your private key when creating Updater or Bot. - PassportFiles is also automagically decrypted as you download/retrieve them. - See new passportbot.py example for details on how to use, or go to our telegram passport wiki page for more info - NOTE: Passport decryption requires new dependency `cryptography`. Inputfile rework (#1184): - Change how Inputfile is handled internally - This allows support for specifying the thumbnails of photos and videos using the thumb= argument in the different send_ methods. - Also allows Bot.send_media_group to actually finally send more than one media. - Add thumb to Audio, Video and Videonote - Add Bot.edit_message_media together with InputMediaAnimation, InputMediaAudio, and inputMediaDocument. Other Bot API 4.0 changes: - Add forusquare_type to Venue, InlineQueryResultVenue, InputVenueMessageContent, and Bot.send_venue. (#1170) - Add vCard support by adding vcard field to Contact, InlineQueryResultContact, InputContactMessageContent, and Bot.send_contact. (#1166) - Support new message entities: CASHTAG and PHONE_NUMBER. (#1179) - Cashtag seems to be things like $USD and $GBP, but it seems telegram doesn't currently send them to bots. - Phone number also seems to have limited support for now - Add Bot.send_animation, add width, height, and duration to Animation, and add Filters.animation. (#1172) Co-authored-by: Jasmin Bom <jsmnbom@gmail.com> Co-authored-by: code1mountain <32801117+code1mountain@users.noreply.github.com> Co-authored-by: Eldinnie <pieter.schutz+github@gmail.com> Co-authored-by: mathefreak1 <mathefreak@hi2.in>
2018-08-29 14:18:58 +02:00
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])
2020-05-27 21:59:49 +02:00
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 = '<a href="tg://user?id={}">{}</a>'
assert user.mention_html() == expected.format(user.id, user.full_name)
assert user.mention_html("the<b>name\u2022") == expected.format(
user.id, "the&lt;b&gt;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):
API 4.5 (#1508) * Allow for nested MessageEntities in Message._parse_markdown/html, adjust tests * remove testing relict * Use MessageEntitys new equality check (#1465) * Remove unused variable * Update to custom_title feature and slow_mode_delay option Changes: - custom_title for ChatMember - new method setChatAdministratorCustomTitle for Bot - new slow_mode_delay for Chat Update due to new API future `custom_title` from API 4.5 (https://core.telegram.org/bots/api#december-31-2019) * Minor typo fix * Comply with Flake8 * Add new MessageEntities and MarkdownV2 * Added file_unique_id attrs from API 4.5 and updated tests for it * Fixed test and checked using flake8 * Fixed ChatPhoto documentation * Fix Flake8 * Add setChatAdminCstmTitle to Bot * Rename MDV2 methods * Change files id attrs to unique id * correct id_attrs for chat_photo * Revert "temporarily skip tests failing b/c missing api 4.5 (#1738)" This reverts commit 7cde6ca268754f029484e6cbded4363117df2e9b. * Fix text_markdown_v2 for monospace and text_links * closing remarks from pieter * Minor fix in escape_markdown, improve tests for it * Fix offset bug in Message._parse_* * Add test_chatphoto.py * remove debug print from test_message.py * try making codecov happy * Update readme * all hail codecov * Improve Link handling for MarkdownV1 and adjust tests. Closes #1654 * Dont use beginning of pre-entity as language in _parse_markdown * Remove debug print * Dummy commit to try fix codecov Co-authored-by: Hoi Dmytro <dmytro.hoi@gmail.com> Co-authored-by: Dmytro Hoi <code@dmytrohoi.com> Co-authored-by: poolitzer <25934244+poolitzer@users.noreply.github.com>
2020-03-28 16:37:26 +01:00
user.first_name = "first{name"
user.last_name = "last_name"
expected = "[{}](tg://user?id={})"
API 4.5 (#1508) * Allow for nested MessageEntities in Message._parse_markdown/html, adjust tests * remove testing relict * Use MessageEntitys new equality check (#1465) * Remove unused variable * Update to custom_title feature and slow_mode_delay option Changes: - custom_title for ChatMember - new method setChatAdministratorCustomTitle for Bot - new slow_mode_delay for Chat Update due to new API future `custom_title` from API 4.5 (https://core.telegram.org/bots/api#december-31-2019) * Minor typo fix * Comply with Flake8 * Add new MessageEntities and MarkdownV2 * Added file_unique_id attrs from API 4.5 and updated tests for it * Fixed test and checked using flake8 * Fixed ChatPhoto documentation * Fix Flake8 * Add setChatAdminCstmTitle to Bot * Rename MDV2 methods * Change files id attrs to unique id * correct id_attrs for chat_photo * Revert "temporarily skip tests failing b/c missing api 4.5 (#1738)" This reverts commit 7cde6ca268754f029484e6cbded4363117df2e9b. * Fix text_markdown_v2 for monospace and text_links * closing remarks from pieter * Minor fix in escape_markdown, improve tests for it * Fix offset bug in Message._parse_* * Add test_chatphoto.py * remove debug print from test_message.py * try making codecov happy * Update readme * all hail codecov * Improve Link handling for MarkdownV1 and adjust tests. Closes #1654 * Dont use beginning of pre-entity as language in _parse_markdown * Remove debug print * Dummy commit to try fix codecov Co-authored-by: Hoi Dmytro <dmytro.hoi@gmail.com> Co-authored-by: Dmytro Hoi <code@dmytrohoi.com> Co-authored-by: poolitzer <25934244+poolitzer@users.noreply.github.com>
2020-03-28 16:37:26 +01:00
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
)
API 4.5 (#1508) * Allow for nested MessageEntities in Message._parse_markdown/html, adjust tests * remove testing relict * Use MessageEntitys new equality check (#1465) * Remove unused variable * Update to custom_title feature and slow_mode_delay option Changes: - custom_title for ChatMember - new method setChatAdministratorCustomTitle for Bot - new slow_mode_delay for Chat Update due to new API future `custom_title` from API 4.5 (https://core.telegram.org/bots/api#december-31-2019) * Minor typo fix * Comply with Flake8 * Add new MessageEntities and MarkdownV2 * Added file_unique_id attrs from API 4.5 and updated tests for it * Fixed test and checked using flake8 * Fixed ChatPhoto documentation * Fix Flake8 * Add setChatAdminCstmTitle to Bot * Rename MDV2 methods * Change files id attrs to unique id * correct id_attrs for chat_photo * Revert "temporarily skip tests failing b/c missing api 4.5 (#1738)" This reverts commit 7cde6ca268754f029484e6cbded4363117df2e9b. * Fix text_markdown_v2 for monospace and text_links * closing remarks from pieter * Minor fix in escape_markdown, improve tests for it * Fix offset bug in Message._parse_* * Add test_chatphoto.py * remove debug print from test_message.py * try making codecov happy * Update readme * all hail codecov * Improve Link handling for MarkdownV1 and adjust tests. Closes #1654 * Dont use beginning of pre-entity as language in _parse_markdown * Remove debug print * Dummy commit to try fix codecov Co-authored-by: Hoi Dmytro <dmytro.hoi@gmail.com> Co-authored-by: Dmytro Hoi <code@dmytrohoi.com> Co-authored-by: poolitzer <25934244+poolitzer@users.noreply.github.com>
2020-03-28 16:37:26 +01:00
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)
async def test_instance_method_send_gift(self, monkeypatch, user):
async def make_assertion(*_, **kwargs):
return (
kwargs["user_id"] == user.id
and kwargs["gift_id"] == "gift_id"
and kwargs["text"] == "text"
and kwargs["text_parse_mode"] == "text_parse_mode"
and kwargs["text_entities"] == "text_entities"
)
assert check_shortcut_signature(user.send_gift, Bot.send_gift, ["user_id"], [])
assert await check_shortcut_call(user.send_gift, user.get_bot(), "send_gift")
assert await check_defaults_handling(user.send_gift, user.get_bot())
monkeypatch.setattr(user.get_bot(), "send_gift", make_assertion)
assert await user.send_gift(
gift_id="gift_id",
text="text",
text_parse_mode="text_parse_mode",
text_entities="text_entities",
)