python-telegram-bot/tests/test_message.py

2015 lines
81 KiB
Python
Raw Normal View History

#!/usr/bin/env python
#
# A library that provides a Python interface to the Telegram Bot API
2023-01-01 21:31:29 +01:00
# Copyright (C) 2015-2023
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
#
# 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/].
from copy import copy
from datetime import datetime
import pytest
from telegram import (
Animation,
Audio,
Bot,
Chat,
ChatShared,
Contact,
Dice,
Document,
Game,
Invoice,
Location,
Message,
MessageAutoDeleteTimerChanged,
MessageEntity,
PassportData,
PhotoSize,
Poll,
PollOption,
ProximityAlertTriggered,
Sticker,
Story,
SuccessfulPayment,
Update,
User,
UserShared,
Venue,
Video,
VideoChatEnded,
VideoChatParticipantsInvited,
VideoChatScheduled,
VideoChatStarted,
VideoNote,
Voice,
WebAppData,
)
from telegram._utils.datetime import UTC
from telegram.constants import ChatAction, ParseMode
from telegram.ext import Defaults
2023-02-22 20:19:46 +01:00
from tests._passport.test_passport import RAW_PASSPORT_DATA
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 message(bot):
message = Message(
message_id=TestMessageBase.id_,
date=TestMessageBase.date,
chat=copy(TestMessageBase.chat),
from_user=copy(TestMessageBase.from_user),
)
message.set_bot(bot)
message._unfreeze()
message.chat._unfreeze()
message.from_user._unfreeze()
return message
@pytest.fixture(
params=[
{"forward_from": User(99, "forward_user", False), "forward_date": datetime.utcnow()},
{
"forward_from_chat": Chat(-23, "channel"),
"forward_from_message_id": 101,
"forward_date": datetime.utcnow(),
},
{
"reply_to_message": Message(
50, datetime.utcnow(), Chat(13, "channel"), User(9, "i", False)
)
},
{"edit_date": datetime.utcnow()},
{
"text": "a text message",
"entities": [MessageEntity("bold", 10, 4), MessageEntity("italic", 16, 7)],
},
{
"caption": "A message caption",
"caption_entities": [MessageEntity("bold", 1, 1), MessageEntity("text_link", 4, 3)],
},
{"audio": Audio("audio_id", "unique_id", 12), "caption": "audio_file"},
{"document": Document("document_id", "unique_id"), "caption": "document_file"},
{
"animation": Animation("animation_id", "unique_id", 30, 30, 1),
"caption": "animation_file",
},
{
"game": Game(
"my_game",
"just my game",
[
PhotoSize("game_photo_id", "unique_id", 30, 30),
],
)
},
{"photo": [PhotoSize("photo_id", "unique_id", 50, 50)], "caption": "photo_file"},
{"sticker": Sticker("sticker_id", "unique_id", 50, 50, True, False, Sticker.REGULAR)},
{"story": Story()},
{"video": Video("video_id", "unique_id", 12, 12, 12), "caption": "video_file"},
{"voice": Voice("voice_id", "unique_id", 5)},
{"video_note": VideoNote("video_note_id", "unique_id", 20, 12)},
{"new_chat_members": [User(55, "new_user", False)]},
{"contact": Contact("phone_numner", "contact_name")},
{"location": Location(-23.691288, 46.788279)},
{"venue": Venue(Location(-23.691288, 46.788279), "some place", "right here")},
{"left_chat_member": User(33, "kicked", False)},
{"new_chat_title": "new title"},
{"new_chat_photo": [PhotoSize("photo_id", "unique_id", 50, 50)]},
{"delete_chat_photo": True},
{"group_chat_created": True},
{"supergroup_chat_created": True},
{"channel_chat_created": True},
{"message_auto_delete_timer_changed": MessageAutoDeleteTimerChanged(42)},
{"migrate_to_chat_id": -12345},
{"migrate_from_chat_id": -54321},
{
"pinned_message": Message(
7, datetime.utcnow(), Chat(13, "channel"), User(9, "i", False)
)
},
{"invoice": Invoice("my invoice", "invoice", "start", "EUR", 243)},
{
"successful_payment": SuccessfulPayment(
"EUR", 243, "payload", "charge_id", "provider_id", order_info={}
)
},
{"connected_website": "http://example.com/"},
{"forward_signature": "some_forward_sign"},
{"author_signature": "some_author_sign"},
{
"photo": [PhotoSize("photo_id", "unique_id", 50, 50)],
"caption": "photo_file",
"media_group_id": 1234443322222,
},
{"passport_data": PassportData.de_json(RAW_PASSPORT_DATA, None)},
{
"poll": Poll(
id="abc",
question="What is this?",
options=[PollOption(text="a", voter_count=1), PollOption(text="b", voter_count=2)],
is_closed=False,
total_voter_count=0,
is_anonymous=False,
type=Poll.REGULAR,
allows_multiple_answers=True,
explanation_entities=[],
)
},
{
"text": "a text message",
"reply_markup": {
"inline_keyboard": [
[
{"text": "start", "url": "http://google.com"},
{"text": "next", "callback_data": "abcd"},
],
[{"text": "Cancel", "callback_data": "Cancel"}],
]
},
},
{"dice": Dice(4, "🎲")},
{"via_bot": User(9, "A_Bot", True)},
{
"proximity_alert_triggered": ProximityAlertTriggered(
User(1, "John", False), User(2, "Doe", False), 42
)
},
{"video_chat_scheduled": VideoChatScheduled(datetime.utcnow())},
{"video_chat_started": VideoChatStarted()},
{"video_chat_ended": VideoChatEnded(100)},
API 5.1 (#2424) * Feat: New invite links * Fix: doc strings Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> * new dice, new admin privilege, revoke_messages, update and fix some docs * add missing param to shortcut * Add ChatMemberUpdated * Add voicechat related objects Signed-off-by: starry69 <starry369126@outlook.com> * add versionadd tags Signed-off-by: starry69 <starry369126@outlook.com> * Fix filter tests * Update tg.Update * ChatMemberHandler * Add versioning directives * add can_manage_voice_chats attr and fix docs Signed-off-by: starry69 <starry369126@outlook.com> * fix chat shortcut Signed-off-by: starry69 <starry369126@outlook.com> * address review * MADTC * Chat.message_auto_delete_time * Some doc fixes * address review Signed-off-by: starry69 <starry369126@outlook.com> * welp Signed-off-by: starry69 <starry369126@outlook.com> * Add voicechat related filters Signed-off-by: starry69 <starry369126@outlook.com> * Fix: Addressing review change place of version adding, added obj:True as doc string, changing how member limit is initiated * feat: adding chat shortcuts for invite links * fix: changing equality of chatinviteobjects * Non-test comments * Some test fixes * A bit more tests * Bump API version in both readmes * Increase coverage * Add Bot API Version in telegram.constants (#2429) * add bot api version in constants Signed-off-by: starry69 <starry369126@outlook.com> * addressing review Signed-off-by: starry69 <starry369126@outlook.com> * add versioning directive Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> * pre-commit & coverage Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> Co-authored-by: Harshil <ilovebhagwan@gmail.com> Co-authored-by: starry69 <starry369126@outlook.com>
2021-03-14 16:41:35 +01:00
{
"video_chat_participants_invited": VideoChatParticipantsInvited(
[User(1, "Rem", False), User(2, "Emilia", False)]
API 5.1 (#2424) * Feat: New invite links * Fix: doc strings Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> * new dice, new admin privilege, revoke_messages, update and fix some docs * add missing param to shortcut * Add ChatMemberUpdated * Add voicechat related objects Signed-off-by: starry69 <starry369126@outlook.com> * add versionadd tags Signed-off-by: starry69 <starry369126@outlook.com> * Fix filter tests * Update tg.Update * ChatMemberHandler * Add versioning directives * add can_manage_voice_chats attr and fix docs Signed-off-by: starry69 <starry369126@outlook.com> * fix chat shortcut Signed-off-by: starry69 <starry369126@outlook.com> * address review * MADTC * Chat.message_auto_delete_time * Some doc fixes * address review Signed-off-by: starry69 <starry369126@outlook.com> * welp Signed-off-by: starry69 <starry369126@outlook.com> * Add voicechat related filters Signed-off-by: starry69 <starry369126@outlook.com> * Fix: Addressing review change place of version adding, added obj:True as doc string, changing how member limit is initiated * feat: adding chat shortcuts for invite links * fix: changing equality of chatinviteobjects * Non-test comments * Some test fixes * A bit more tests * Bump API version in both readmes * Increase coverage * Add Bot API Version in telegram.constants (#2429) * add bot api version in constants Signed-off-by: starry69 <starry369126@outlook.com> * addressing review Signed-off-by: starry69 <starry369126@outlook.com> * add versioning directive Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> * pre-commit & coverage Co-authored-by: Bibo-Joshi <hinrich.mahler@freenet.de> Co-authored-by: Harshil <ilovebhagwan@gmail.com> Co-authored-by: starry69 <starry369126@outlook.com>
2021-03-14 16:41:35 +01:00
)
},
{"sender_chat": Chat(-123, "discussion_channel")},
{"is_automatic_forward": True},
{"has_protected_content": True},
{
"entities": [
MessageEntity(MessageEntity.BOLD, 0, 1),
MessageEntity(MessageEntity.TEXT_LINK, 2, 3, url="https://ptb.org"),
]
},
{"web_app_data": WebAppData("some_data", "some_button_text")},
{"message_thread_id": 123},
{"user_shared": UserShared(1, 2)},
{"chat_shared": ChatShared(3, 4)},
],
ids=[
"forwarded_user",
"forwarded_channel",
"reply",
"edited",
"text",
"caption_entities",
"audio",
"document",
"animation",
"game",
"photo",
"sticker",
"story",
"video",
"voice",
"video_note",
"new_members",
"contact",
"location",
"venue",
"left_member",
"new_title",
"new_photo",
"delete_photo",
"group_created",
"supergroup_created",
"channel_created",
"message_auto_delete_timer_changed",
"migrated_to",
"migrated_from",
"pinned",
"invoice",
"successful_payment",
"connected_website",
"forward_signature",
"author_signature",
"photo_from_media_group",
"passport_data",
"poll",
"reply_markup",
"dice",
"via_bot",
"proximity_alert_triggered",
"video_chat_scheduled",
"video_chat_started",
"video_chat_ended",
"video_chat_participants_invited",
"sender_chat",
"is_automatic_forward",
"has_protected_content",
"entities",
"web_app_data",
"message_thread_id",
"user_shared",
"chat_shared",
],
)
def message_params(bot, request):
message = Message(
message_id=TestMessageBase.id_,
from_user=TestMessageBase.from_user,
date=TestMessageBase.date,
chat=TestMessageBase.chat,
**request.param,
)
message.set_bot(bot)
return message
class TestMessageBase:
id_ = 1
from_user = User(2, "testuser", False)
date = datetime.utcnow()
chat = Chat(3, "private")
test_entities = [
{"length": 4, "offset": 10, "type": "bold"},
{"length": 3, "offset": 16, "type": "italic"},
{"length": 3, "offset": 20, "type": "italic"},
{"length": 4, "offset": 25, "type": "code"},
{"length": 5, "offset": 31, "type": "text_link", "url": "http://github.com/ab_"},
{
"length": 12,
"offset": 38,
"type": "text_mention",
"user": User(123456789, "mentioned user", False),
},
{"length": 3, "offset": 55, "type": "pre", "language": "python"},
{"length": 21, "offset": 60, "type": "url"},
]
test_text = "Test for <bold, ita_lic, code, links, text-mention and pre. http://google.com/ab_"
test_entities_v2 = [
{"length": 4, "offset": 0, "type": "underline"},
{"length": 4, "offset": 10, "type": "bold"},
{"length": 7, "offset": 16, "type": "italic"},
{"length": 6, "offset": 25, "type": "code"},
{"length": 5, "offset": 33, "type": "text_link", "url": r"http://github.com/abc\)def"},
{
"length": 12,
"offset": 40,
"type": "text_mention",
"user": User(123456789, "mentioned user", False),
},
{"length": 5, "offset": 57, "type": "pre"},
{"length": 17, "offset": 64, "type": "url"},
{"length": 41, "offset": 86, "type": "italic"},
{"length": 29, "offset": 91, "type": "bold"},
{"length": 9, "offset": 101, "type": "strikethrough"},
{"length": 10, "offset": 129, "type": "pre", "language": "python"},
{"length": 7, "offset": 141, "type": "spoiler"},
{"length": 2, "offset": 150, "type": "custom_emoji", "custom_emoji_id": "1"},
]
test_text_v2 = (
r"Test for <bold, ita_lic, \`code, links, text-mention and `\pre. "
"http://google.com and bold nested in strk>trgh nested in italic. Python pre. Spoiled. 👍."
)
test_message = Message(
message_id=1,
from_user=None,
date=None,
chat=None,
text=test_text,
entities=[MessageEntity(**e) for e in test_entities],
caption=test_text,
caption_entities=[MessageEntity(**e) for e in test_entities],
)
test_message_v2 = Message(
message_id=1,
from_user=None,
date=None,
chat=None,
text=test_text_v2,
entities=[MessageEntity(**e) for e in test_entities_v2],
caption=test_text_v2,
caption_entities=[MessageEntity(**e) for e in test_entities_v2],
)
class TestMessageWithoutRequest(TestMessageBase):
2023-02-22 20:19:46 +01:00
def test_slot_behaviour(self, message):
for attr in message.__slots__:
assert getattr(message, attr, "err") != "err", f"got extra slot '{attr}'"
assert len(mro_slots(message)) == len(set(mro_slots(message))), "duplicate slot"
def test_all_possibilities_de_json_and_to_dict(self, bot, message_params):
new = Message.de_json(message_params.to_dict(), bot)
assert new.api_kwargs == {}
assert new.to_dict() == message_params.to_dict()
# Checking that none of the attributes are dicts is a best effort approach to ensure that
# de_json converts everything to proper classes without having to write special tests for
# every single case
for slot in new.__slots__:
assert not isinstance(new[slot], dict)
def test_de_json_localization(self, bot, raw_bot, tz_bot):
json_dict = {
"message_id": 12,
"from_user": None,
"date": int(datetime.now().timestamp()),
"chat": None,
"edit_date": int(datetime.now().timestamp()),
"forward_date": int(datetime.now().timestamp()),
}
message_raw = Message.de_json(json_dict, raw_bot)
message_bot = Message.de_json(json_dict, bot)
message_tz = Message.de_json(json_dict, tz_bot)
# comparing utcoffsets because comparing timezones is unpredicatable
date_offset = message_tz.date.utcoffset()
date_tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(message_tz.date.replace(tzinfo=None))
edit_date_offset = message_tz.edit_date.utcoffset()
edit_date_tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(
message_tz.edit_date.replace(tzinfo=None)
)
forward_date_offset = message_tz.forward_date.utcoffset()
forward_date_tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(
message_tz.forward_date.replace(tzinfo=None)
)
assert message_raw.date.tzinfo == UTC
assert message_bot.date.tzinfo == UTC
assert date_offset == date_tz_bot_offset
assert message_raw.edit_date.tzinfo == UTC
assert message_bot.edit_date.tzinfo == UTC
assert edit_date_offset == edit_date_tz_bot_offset
assert message_raw.forward_date.tzinfo == UTC
assert message_bot.forward_date.tzinfo == UTC
assert forward_date_offset == forward_date_tz_bot_offset
def test_equality(self):
id_ = 1
a = Message(id_, self.date, self.chat, from_user=self.from_user)
b = Message(id_, self.date, self.chat, from_user=self.from_user)
c = Message(id_, self.date, Chat(123, Chat.GROUP), from_user=User(0, "", False))
d = Message(0, self.date, self.chat, from_user=self.from_user)
e = Update(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)
async def test_parse_entity(self):
text = (
b"\\U0001f469\\u200d\\U0001f469\\u200d\\U0001f467"
b"\\u200d\\U0001f467\\U0001f431http://google.com"
).decode("unicode-escape")
entity = MessageEntity(type=MessageEntity.URL, offset=13, length=17)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[entity],
)
assert message.parse_entity(entity) == "http://google.com"
with pytest.raises(RuntimeError, match="Message has no"):
Message(message_id=1, date=self.date, chat=self.chat).parse_entity(entity)
async def test_parse_caption_entity(self):
caption = (
b"\\U0001f469\\u200d\\U0001f469\\u200d\\U0001f467"
b"\\u200d\\U0001f467\\U0001f431http://google.com"
).decode("unicode-escape")
2017-10-14 20:03:02 +02:00
entity = MessageEntity(type=MessageEntity.URL, offset=13, length=17)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[entity],
)
assert message.parse_caption_entity(entity) == "http://google.com"
2017-10-14 20:03:02 +02:00
with pytest.raises(RuntimeError, match="Message has no"):
Message(message_id=1, date=self.date, chat=self.chat).parse_entity(entity)
async def test_parse_entities(self):
text = (
b"\\U0001f469\\u200d\\U0001f469\\u200d\\U0001f467"
b"\\u200d\\U0001f467\\U0001f431http://google.com"
).decode("unicode-escape")
entity = MessageEntity(type=MessageEntity.URL, offset=13, length=17)
entity_2 = MessageEntity(type=MessageEntity.BOLD, offset=13, length=1)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[entity_2, entity],
)
assert message.parse_entities(MessageEntity.URL) == {entity: "http://google.com"}
assert message.parse_entities() == {entity: "http://google.com", entity_2: "h"}
async def test_parse_caption_entities(self):
text = (
b"\\U0001f469\\u200d\\U0001f469\\u200d\\U0001f467"
b"\\u200d\\U0001f467\\U0001f431http://google.com"
).decode("unicode-escape")
2017-10-14 20:03:02 +02:00
entity = MessageEntity(type=MessageEntity.URL, offset=13, length=17)
entity_2 = MessageEntity(type=MessageEntity.BOLD, offset=13, length=1)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=text,
caption_entities=[entity_2, entity],
)
assert message.parse_caption_entities(MessageEntity.URL) == {entity: "http://google.com"}
assert message.parse_caption_entities() == {
entity: "http://google.com",
entity_2: "h",
}
2017-10-14 20:03:02 +02:00
def test_text_html_simple(self):
test_html_string = (
"<u>Test</u> for &lt;<b>bold</b>, <i>ita_lic</i>, "
r"<code>\`code</code>, "
r'<a href="http://github.com/abc\)def">links</a>, '
'<a href="tg://user?id=123456789">text-mention</a> and '
r"<pre>`\pre</pre>. http://google.com "
"and <i>bold <b>nested in <s>strk&gt;trgh</s> nested in</b> italic</i>. "
2022-01-03 08:13:33 +01:00
'<pre><code class="python">Python pre</code></pre>. '
'<span class="tg-spoiler">Spoiled</span>. '
'<tg-emoji emoji-id="1">👍</tg-emoji>.'
)
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
text_html = self.test_message_v2.text_html
assert text_html == test_html_string
def test_text_html_empty(self, message):
message.text = None
message.caption = "test"
assert message.text_html is None
def test_text_html_urled(self):
test_html_string = (
"<u>Test</u> for &lt;<b>bold</b>, <i>ita_lic</i>, "
r"<code>\`code</code>, "
r'<a href="http://github.com/abc\)def">links</a>, '
'<a href="tg://user?id=123456789">text-mention</a> and '
r'<pre>`\pre</pre>. <a href="http://google.com">http://google.com</a> '
"and <i>bold <b>nested in <s>strk&gt;trgh</s> nested in</b> italic</i>. "
2022-01-03 08:13:33 +01:00
'<pre><code class="python">Python pre</code></pre>. '
'<span class="tg-spoiler">Spoiled</span>. '
'<tg-emoji emoji-id="1">👍</tg-emoji>.'
)
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
text_html = self.test_message_v2.text_html_urled
assert text_html == test_html_string
def test_text_markdown_simple(self):
test_md_string = (
r"Test for <*bold*, _ita_\__lic_, `code`, "
"[links](http://github.com/ab_), "
"[text-mention](tg://user?id=123456789) and ```python\npre```. "
r"http://google.com/ab\_"
)
text_markdown = self.test_message.text_markdown
assert text_markdown == test_md_string
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
def test_text_markdown_v2_simple(self):
test_md_string = (
r"__Test__ for <*bold*, _ita\_lic_, `\\\`code`, "
"[links](http://github.com/abc\\\\\\)def), "
"[text\\-mention](tg://user?id=123456789) and ```\\`\\\\pre```\\. "
r"http://google\.com and _bold *nested in ~strk\>trgh~ nested in* italic_\. "
"```python\nPython pre```\\. ||Spoiled||\\. ![👍](tg://emoji?id=1)\\."
)
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
text_markdown = self.test_message_v2.text_markdown_v2
assert text_markdown == test_md_string
def test_text_markdown_new_in_v2(self, message):
message.text = "test"
message.entities = [
MessageEntity(MessageEntity.BOLD, offset=0, length=4),
MessageEntity(MessageEntity.ITALIC, offset=0, length=4),
]
2023-03-25 19:18:04 +01:00
with pytest.raises(ValueError, match="Nested entities are not supported for"):
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 message.text_markdown
message.entities = [MessageEntity(MessageEntity.UNDERLINE, offset=0, length=4)]
2023-03-25 19:18:04 +01:00
with pytest.raises(ValueError, match="Underline entities are not supported for"):
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
message.text_markdown
message.entities = [MessageEntity(MessageEntity.STRIKETHROUGH, offset=0, length=4)]
2023-03-25 19:18:04 +01:00
with pytest.raises(ValueError, match="Strikethrough entities are not supported for"):
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
message.text_markdown
2022-01-03 08:13:33 +01:00
message.entities = [MessageEntity(MessageEntity.SPOILER, offset=0, length=4)]
2023-03-25 19:18:04 +01:00
with pytest.raises(ValueError, match="Spoiler entities are not supported for"):
2022-01-03 08:13:33 +01:00
message.text_markdown
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
message.entities = []
def test_text_markdown_empty(self, message):
message.text = None
message.caption = "test"
assert message.text_markdown is None
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 message.text_markdown_v2 is None
def test_text_markdown_urled(self):
test_md_string = (
r"Test for <*bold*, _ita_\__lic_, `code`, "
"[links](http://github.com/ab_), "
"[text-mention](tg://user?id=123456789) and ```python\npre```. "
"[http://google.com/ab_](http://google.com/ab_)"
)
text_markdown = self.test_message.text_markdown_urled
assert text_markdown == test_md_string
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
def test_text_markdown_v2_urled(self):
test_md_string = (
r"__Test__ for <*bold*, _ita\_lic_, `\\\`code`, "
"[links](http://github.com/abc\\\\\\)def), "
"[text\\-mention](tg://user?id=123456789) and ```\\`\\\\pre```\\. "
r"[http://google\.com](http://google.com) and _bold *nested in ~strk\>trgh~ "
"nested in* italic_\\. ```python\nPython pre```\\. ||Spoiled||\\. "
"![👍](tg://emoji?id=1)\\."
)
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
text_markdown = self.test_message_v2.text_markdown_v2_urled
assert text_markdown == test_md_string
def test_text_html_emoji(self):
text = b"\\U0001f469\\u200d\\U0001f469\\u200d ABC".decode("unicode-escape")
expected = b"\\U0001f469\\u200d\\U0001f469\\u200d <b>ABC</b>".decode("unicode-escape")
bold_entity = MessageEntity(type=MessageEntity.BOLD, offset=7, length=3)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[bold_entity],
)
assert expected == message.text_html
def test_text_markdown_emoji(self):
text = b"\\U0001f469\\u200d\\U0001f469\\u200d ABC".decode("unicode-escape")
expected = b"\\U0001f469\\u200d\\U0001f469\\u200d *ABC*".decode("unicode-escape")
bold_entity = MessageEntity(type=MessageEntity.BOLD, offset=7, length=3)
message = Message(
1, self.date, self.chat, self.from_user, text=text, entities=[bold_entity]
)
assert expected == message.text_markdown
@pytest.mark.parametrize(
"type_",
argvalues=[
"text_markdown",
"text_markdown_urled",
],
)
def test_text_custom_emoji_md_v1(self, type_, recwarn):
text = "Look a custom emoji: 😎"
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[emoji_entity],
)
with pytest.raises(ValueError, match="Custom emoji entities are not supported for"):
getattr(message, type_)
@pytest.mark.parametrize(
"type_",
argvalues=[
"text_markdown_v2",
"text_markdown_v2_urled",
],
)
def test_text_custom_emoji_md_v2(self, type_):
text = "Look a custom emoji: 😎"
expected = "Look a custom emoji: ![😎](tg://emoji?id=5472409228461217725)"
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[emoji_entity],
)
assert expected == message[type_]
@pytest.mark.parametrize(
"type_",
argvalues=[
"text_html",
"text_html_urled",
],
)
def test_text_custom_emoji_html(self, type_):
text = "Look a custom emoji: 😎"
expected = 'Look a custom emoji: <tg-emoji emoji-id="5472409228461217725">😎</tg-emoji>'
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
text=text,
entities=[emoji_entity],
)
assert expected == message[type_]
def test_caption_html_simple(self):
test_html_string = (
"<u>Test</u> for &lt;<b>bold</b>, <i>ita_lic</i>, "
r"<code>\`code</code>, "
r'<a href="http://github.com/abc\)def">links</a>, '
'<a href="tg://user?id=123456789">text-mention</a> and '
r"<pre>`\pre</pre>. http://google.com "
"and <i>bold <b>nested in <s>strk&gt;trgh</s> nested in</b> italic</i>. "
2022-01-03 08:13:33 +01:00
'<pre><code class="python">Python pre</code></pre>. '
'<span class="tg-spoiler">Spoiled</span>. '
'<tg-emoji emoji-id="1">👍</tg-emoji>.'
)
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
caption_html = self.test_message_v2.caption_html
assert caption_html == test_html_string
def test_caption_html_empty(self, message):
message.text = "test"
message.caption = None
assert message.caption_html is None
def test_caption_html_urled(self):
test_html_string = (
"<u>Test</u> for &lt;<b>bold</b>, <i>ita_lic</i>, "
r"<code>\`code</code>, "
r'<a href="http://github.com/abc\)def">links</a>, '
'<a href="tg://user?id=123456789">text-mention</a> and '
r'<pre>`\pre</pre>. <a href="http://google.com">http://google.com</a> '
"and <i>bold <b>nested in <s>strk&gt;trgh</s> nested in</b> italic</i>. "
2022-01-03 08:13:33 +01:00
'<pre><code class="python">Python pre</code></pre>. '
'<span class="tg-spoiler">Spoiled</span>. '
'<tg-emoji emoji-id="1">👍</tg-emoji>.'
)
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
caption_html = self.test_message_v2.caption_html_urled
assert caption_html == test_html_string
def test_caption_markdown_simple(self):
test_md_string = (
r"Test for <*bold*, _ita_\__lic_, `code`, "
"[links](http://github.com/ab_), "
"[text-mention](tg://user?id=123456789) and ```python\npre```. "
r"http://google.com/ab\_"
)
caption_markdown = self.test_message.caption_markdown
assert caption_markdown == test_md_string
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
def test_caption_markdown_v2_simple(self):
test_md_string = (
r"__Test__ for <*bold*, _ita\_lic_, `\\\`code`, "
"[links](http://github.com/abc\\\\\\)def), "
"[text\\-mention](tg://user?id=123456789) and ```\\`\\\\pre```\\. "
r"http://google\.com and _bold *nested in ~strk\>trgh~ nested in* italic_\. "
"```python\nPython pre```\\. ||Spoiled||\\. ![👍](tg://emoji?id=1)\\."
)
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
caption_markdown = self.test_message_v2.caption_markdown_v2
assert caption_markdown == test_md_string
def test_caption_markdown_empty(self, message):
message.text = "test"
message.caption = None
assert message.caption_markdown is None
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 message.caption_markdown_v2 is None
def test_caption_markdown_urled(self):
test_md_string = (
r"Test for <*bold*, _ita_\__lic_, `code`, "
"[links](http://github.com/ab_), "
"[text-mention](tg://user?id=123456789) and ```python\npre```. "
"[http://google.com/ab_](http://google.com/ab_)"
)
caption_markdown = self.test_message.caption_markdown_urled
assert caption_markdown == test_md_string
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
def test_caption_markdown_v2_urled(self):
test_md_string = (
r"__Test__ for <*bold*, _ita\_lic_, `\\\`code`, "
"[links](http://github.com/abc\\\\\\)def), "
"[text\\-mention](tg://user?id=123456789) and ```\\`\\\\pre```\\. "
r"[http://google\.com](http://google.com) and _bold *nested in ~strk\>trgh~ "
"nested in* italic_\\. ```python\nPython pre```\\. ||Spoiled||\\. "
"![👍](tg://emoji?id=1)\\."
)
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
caption_markdown = self.test_message_v2.caption_markdown_v2_urled
assert caption_markdown == test_md_string
def test_caption_html_emoji(self):
caption = b"\\U0001f469\\u200d\\U0001f469\\u200d ABC".decode("unicode-escape")
expected = b"\\U0001f469\\u200d\\U0001f469\\u200d <b>ABC</b>".decode("unicode-escape")
bold_entity = MessageEntity(type=MessageEntity.BOLD, offset=7, length=3)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[bold_entity],
)
assert expected == message.caption_html
def test_caption_markdown_emoji(self):
caption = b"\\U0001f469\\u200d\\U0001f469\\u200d ABC".decode("unicode-escape")
expected = b"\\U0001f469\\u200d\\U0001f469\\u200d *ABC*".decode("unicode-escape")
bold_entity = MessageEntity(type=MessageEntity.BOLD, offset=7, length=3)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[bold_entity],
)
assert expected == message.caption_markdown
@pytest.mark.parametrize(
"type_",
argvalues=[
"caption_markdown",
"caption_markdown_urled",
],
)
def test_caption_custom_emoji_md_v1(self, type_, recwarn):
caption = "Look a custom emoji: 😎"
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[emoji_entity],
)
with pytest.raises(ValueError, match="Custom emoji entities are not supported for"):
getattr(message, type_)
@pytest.mark.parametrize(
"type_",
argvalues=[
"caption_markdown_v2",
"caption_markdown_v2_urled",
],
)
def test_caption_custom_emoji_md_v2(self, type_):
caption = "Look a custom emoji: 😎"
expected = "Look a custom emoji: ![😎](tg://emoji?id=5472409228461217725)"
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[emoji_entity],
)
assert expected == message[type_]
@pytest.mark.parametrize(
"type_",
argvalues=[
"caption_html",
"caption_html_urled",
],
)
def test_caption_custom_emoji_html(self, type_):
caption = "Look a custom emoji: 😎"
expected = 'Look a custom emoji: <tg-emoji emoji-id="5472409228461217725">😎</tg-emoji>'
emoji_entity = MessageEntity(
type=MessageEntity.CUSTOM_EMOJI,
offset=21,
length=2,
custom_emoji_id="5472409228461217725",
)
message = Message(
1,
from_user=self.from_user,
date=self.date,
chat=self.chat,
caption=caption,
caption_entities=[emoji_entity],
)
assert expected == message[type_]
async def test_parse_entities_url_emoji(self):
url = b"http://github.com/?unicode=\\u2713\\U0001f469".decode("unicode-escape")
text = "some url"
link_entity = MessageEntity(type=MessageEntity.URL, offset=0, length=8, url=url)
message = Message(
1, self.from_user, self.date, self.chat, text=text, entities=[link_entity]
)
assert message.parse_entities() == {link_entity: text}
assert next(iter(message.parse_entities())).url == url
def test_chat_id(self, message):
assert message.chat_id == message.chat.id
def test_id(self, message):
assert message.message_id == message.id
@pytest.mark.parametrize("type_", argvalues=[Chat.SUPERGROUP, Chat.CHANNEL])
def test_link_with_username(self, message, type_):
message.chat.username = "username"
message.chat.type = type_
assert message.link == f"https://t.me/{message.chat.username}/{message.message_id}"
@pytest.mark.parametrize(
2023-03-25 19:18:04 +01:00
("type_", "id_"), argvalues=[(Chat.CHANNEL, -1003), (Chat.SUPERGROUP, -1003)]
)
def test_link_with_id(self, message, type_, id_):
message.chat.username = None
message.chat.id = id_
message.chat.type = type_
# The leading - for group ids/ -100 for supergroup ids isn't supposed to be in the link
assert message.link == f"https://t.me/c/{3}/{message.message_id}"
def test_link_with_topics(self, message):
message.chat.username = None
message.chat.id = -1003
message.is_topic_message = True
message.message_thread_id = 123
assert message.link == f"https://t.me/c/3/{message.message_id}?thread=123"
def test_link_with_reply(self, message):
message.chat.username = None
message.chat.id = -1003
message.reply_to_message = Message(7, self.from_user, self.date, self.chat, text="Reply")
message.message_thread_id = 123
assert message.link == f"https://t.me/c/3/{message.message_id}?thread=123"
2023-03-25 19:18:04 +01:00
@pytest.mark.parametrize(("id_", "username"), argvalues=[(None, "username"), (-3, None)])
def test_link_private_chats(self, message, id_, username):
message.chat.type = Chat.PRIVATE
message.chat.id = id_
message.chat.username = username
assert message.link is None
message.chat.type = Chat.GROUP
assert message.link is None
def test_effective_attachment(self, message_params):
# This list is hard coded on purpose because just using constants.MessageAttachmentType
# (which is used in Message.effective_message) wouldn't find any mistakes
expected_attachment_types = [
"animation",
"audio",
"contact",
"dice",
"document",
"game",
"invoice",
"location",
"passport_data",
"photo",
"poll",
"sticker",
"story",
"successful_payment",
"video",
"video_note",
"voice",
"venue",
]
for _ in range(3):
# We run the same test multiple times to make sure that the caching is tested
attachment = message_params.effective_attachment
if attachment:
condition = any(
message_params[message_type] is attachment
for message_type in expected_attachment_types
)
assert condition, "Got effective_attachment for unexpected type"
else:
condition = any(
message_params[message_type] for message_type in expected_attachment_types
)
assert not condition, "effective_attachment was None even though it should not be"
async def test_reply_text(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
text = kwargs["text"] == "test"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and text and reply
assert check_shortcut_signature(
Message.reply_text, Bot.send_message, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_text, message.get_bot(), "send_message")
assert await check_defaults_handling(message.reply_text, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_message", make_assertion)
assert await message.reply_text("test")
assert await message.reply_text("test", quote=True)
assert await message.reply_text("test", reply_to_message_id=message.message_id, quote=True)
async def test_reply_markdown(self, monkeypatch, message):
test_md_string = (
r"Test for <*bold*, _ita_\__lic_, `code`, "
"[links](http://github.com/ab_), "
"[text-mention](tg://user?id=123456789) and ```python\npre```. "
r"http://google.com/ab\_"
)
async def make_assertion(*_, **kwargs):
cid = kwargs["chat_id"] == message.chat_id
markdown_text = kwargs["text"] == test_md_string
markdown_enabled = kwargs["parse_mode"] == ParseMode.MARKDOWN
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return all([cid, markdown_text, reply, markdown_enabled])
assert check_shortcut_signature(
Message.reply_markdown, Bot.send_message, ["chat_id", "parse_mode"], ["quote"]
)
assert await check_shortcut_call(message.reply_text, message.get_bot(), "send_message")
assert await check_defaults_handling(message.reply_text, message.get_bot())
text_markdown = self.test_message.text_markdown
assert text_markdown == test_md_string
monkeypatch.setattr(message.get_bot(), "send_message", make_assertion)
assert await message.reply_markdown(self.test_message.text_markdown)
assert await message.reply_markdown(self.test_message.text_markdown, quote=True)
assert await message.reply_markdown(
self.test_message.text_markdown, reply_to_message_id=message.message_id, quote=True
)
async def test_reply_markdown_v2(self, monkeypatch, message):
test_md_string = (
r"__Test__ for <*bold*, _ita\_lic_, `\\\`code`, "
"[links](http://github.com/abc\\\\\\)def), "
"[text\\-mention](tg://user?id=123456789) and ```\\`\\\\pre```\\. "
r"http://google\.com and _bold *nested in ~strk\>trgh~ nested in* italic_\. "
"```python\nPython pre```\\. ||Spoiled||\\. ![👍](tg://emoji?id=1)\\."
)
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
async def make_assertion(*_, **kwargs):
cid = kwargs["chat_id"] == message.chat_id
markdown_text = kwargs["text"] == test_md_string
markdown_enabled = kwargs["parse_mode"] == ParseMode.MARKDOWN_V2
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_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
else:
reply = True
return all([cid, markdown_text, reply, markdown_enabled])
assert check_shortcut_signature(
Message.reply_markdown_v2, Bot.send_message, ["chat_id", "parse_mode"], ["quote"]
)
assert await check_shortcut_call(message.reply_text, message.get_bot(), "send_message")
assert await check_defaults_handling(message.reply_text, message.get_bot())
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
text_markdown = self.test_message_v2.text_markdown_v2
assert text_markdown == test_md_string
monkeypatch.setattr(message.get_bot(), "send_message", make_assertion)
assert await message.reply_markdown_v2(self.test_message_v2.text_markdown_v2)
assert await message.reply_markdown_v2(self.test_message_v2.text_markdown_v2, quote=True)
assert await message.reply_markdown_v2(
self.test_message_v2.text_markdown_v2,
reply_to_message_id=message.message_id,
quote=True,
)
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
async def test_reply_html(self, monkeypatch, message):
test_html_string = (
"<u>Test</u> for &lt;<b>bold</b>, <i>ita_lic</i>, "
r"<code>\`code</code>, "
r'<a href="http://github.com/abc\)def">links</a>, '
'<a href="tg://user?id=123456789">text-mention</a> and '
r"<pre>`\pre</pre>. http://google.com "
"and <i>bold <b>nested in <s>strk&gt;trgh</s> nested in</b> italic</i>. "
2022-01-03 08:13:33 +01:00
'<pre><code class="python">Python pre</code></pre>. '
'<span class="tg-spoiler">Spoiled</span>. '
'<tg-emoji emoji-id="1">👍</tg-emoji>.'
)
async def make_assertion(*_, **kwargs):
cid = kwargs["chat_id"] == message.chat_id
html_text = kwargs["text"] == test_html_string
html_enabled = kwargs["parse_mode"] == ParseMode.HTML
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return all([cid, html_text, reply, html_enabled])
assert check_shortcut_signature(
Message.reply_html, Bot.send_message, ["chat_id", "parse_mode"], ["quote"]
)
assert await check_shortcut_call(message.reply_text, message.get_bot(), "send_message")
assert await check_defaults_handling(message.reply_text, message.get_bot())
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
text_html = self.test_message_v2.text_html
assert text_html == test_html_string
monkeypatch.setattr(message.get_bot(), "send_message", make_assertion)
assert await message.reply_html(self.test_message_v2.text_html)
assert await message.reply_html(self.test_message_v2.text_html, quote=True)
assert await message.reply_html(
self.test_message_v2.text_html, reply_to_message_id=message.message_id, quote=True
)
async def test_reply_media_group(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
media = kwargs["media"] == "reply_media_group"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and media and reply
assert check_shortcut_signature(
Message.reply_media_group, Bot.send_media_group, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(
message.reply_media_group, message.get_bot(), "send_media_group"
)
assert await check_defaults_handling(message.reply_media_group, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_media_group", make_assertion)
assert await message.reply_media_group(media="reply_media_group")
assert await message.reply_media_group(media="reply_media_group", quote=True)
async def test_reply_photo(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
photo = kwargs["photo"] == "test_photo"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and photo and reply
assert check_shortcut_signature(
Message.reply_photo, Bot.send_photo, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_photo, message.get_bot(), "send_photo")
assert await check_defaults_handling(message.reply_photo, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_photo", make_assertion)
assert await message.reply_photo(photo="test_photo")
assert await message.reply_photo(photo="test_photo", quote=True)
async def test_reply_audio(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
audio = kwargs["audio"] == "test_audio"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and audio and reply
assert check_shortcut_signature(
Message.reply_audio, Bot.send_audio, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_audio, message.get_bot(), "send_audio")
assert await check_defaults_handling(message.reply_audio, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_audio", make_assertion)
assert await message.reply_audio(audio="test_audio")
assert await message.reply_audio(audio="test_audio", quote=True)
async def test_reply_document(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
document = kwargs["document"] == "test_document"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and document and reply
assert check_shortcut_signature(
Message.reply_document, Bot.send_document, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(
message.reply_document, message.get_bot(), "send_document"
)
assert await check_defaults_handling(message.reply_document, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_document", make_assertion)
assert await message.reply_document(document="test_document")
assert await message.reply_document(document="test_document", quote=True)
async def test_reply_animation(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
animation = kwargs["animation"] == "test_animation"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
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
else:
reply = True
return id_ and animation and reply
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
assert check_shortcut_signature(
Message.reply_animation, Bot.send_animation, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(
message.reply_animation, message.get_bot(), "send_animation"
)
assert await check_defaults_handling(message.reply_animation, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_animation", make_assertion)
assert await message.reply_animation(animation="test_animation")
assert await message.reply_animation(animation="test_animation", quote=True)
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_reply_sticker(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
sticker = kwargs["sticker"] == "test_sticker"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and sticker and reply
assert check_shortcut_signature(
Message.reply_sticker, Bot.send_sticker, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_sticker, message.get_bot(), "send_sticker")
assert await check_defaults_handling(message.reply_sticker, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_sticker", make_assertion)
assert await message.reply_sticker(sticker="test_sticker")
assert await message.reply_sticker(sticker="test_sticker", quote=True)
async def test_reply_video(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
video = kwargs["video"] == "test_video"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and video and reply
assert check_shortcut_signature(
Message.reply_video, Bot.send_video, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_video, message.get_bot(), "send_video")
assert await check_defaults_handling(message.reply_video, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_video", make_assertion)
assert await message.reply_video(video="test_video")
assert await message.reply_video(video="test_video", quote=True)
async def test_reply_video_note(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
video_note = kwargs["video_note"] == "test_video_note"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and video_note and reply
assert check_shortcut_signature(
Message.reply_video_note, Bot.send_video_note, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(
message.reply_video_note, message.get_bot(), "send_video_note"
)
assert await check_defaults_handling(message.reply_video_note, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_video_note", make_assertion)
assert await message.reply_video_note(video_note="test_video_note")
assert await message.reply_video_note(video_note="test_video_note", quote=True)
async def test_reply_voice(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
voice = kwargs["voice"] == "test_voice"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and voice and reply
assert check_shortcut_signature(
Message.reply_voice, Bot.send_voice, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_voice, message.get_bot(), "send_voice")
assert await check_defaults_handling(message.reply_voice, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_voice", make_assertion)
assert await message.reply_voice(voice="test_voice")
assert await message.reply_voice(voice="test_voice", quote=True)
async def test_reply_location(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
location = kwargs["location"] == "test_location"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and location and reply
assert check_shortcut_signature(
Message.reply_location, Bot.send_location, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(
message.reply_location, message.get_bot(), "send_location"
)
assert await check_defaults_handling(message.reply_location, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_location", make_assertion)
assert await message.reply_location(location="test_location")
assert await message.reply_location(location="test_location", quote=True)
async def test_reply_venue(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
venue = kwargs["venue"] == "test_venue"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and venue and reply
assert check_shortcut_signature(
Message.reply_venue, Bot.send_venue, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_venue, message.get_bot(), "send_venue")
assert await check_defaults_handling(message.reply_venue, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_venue", make_assertion)
assert await message.reply_venue(venue="test_venue")
assert await message.reply_venue(venue="test_venue", quote=True)
async def test_reply_contact(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
contact = kwargs["contact"] == "test_contact"
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and contact and reply
assert check_shortcut_signature(
Message.reply_contact, Bot.send_contact, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_contact, message.get_bot(), "send_contact")
assert await check_defaults_handling(message.reply_contact, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_contact", make_assertion)
assert await message.reply_contact(contact="test_contact")
assert await message.reply_contact(contact="test_contact", quote=True)
async def test_reply_poll(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
question = kwargs["question"] == "test_poll"
options = kwargs["options"] == ["1", "2", "3"]
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
2019-08-23 21:20:41 +02:00
else:
reply = True
return id_ and question and options and reply
assert check_shortcut_signature(Message.reply_poll, Bot.send_poll, ["chat_id"], ["quote"])
assert await check_shortcut_call(message.reply_poll, message.get_bot(), "send_poll")
assert await check_defaults_handling(message.reply_poll, message.get_bot())
2019-08-23 21:20:41 +02:00
monkeypatch.setattr(message.get_bot(), "send_poll", make_assertion)
assert await message.reply_poll(question="test_poll", options=["1", "2", "3"])
assert await message.reply_poll(question="test_poll", quote=True, options=["1", "2", "3"])
async def test_reply_dice(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
contact = kwargs["disable_notification"] is True
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
return id_ and contact and reply
assert check_shortcut_signature(Message.reply_dice, Bot.send_dice, ["chat_id"], ["quote"])
assert await check_shortcut_call(message.reply_dice, message.get_bot(), "send_dice")
assert await check_defaults_handling(message.reply_dice, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_dice", make_assertion)
assert await message.reply_dice(disable_notification=True)
assert await message.reply_dice(disable_notification=True, quote=True)
2019-08-23 21:20:41 +02:00
async def test_reply_action(self, monkeypatch, message: Message):
async def make_assertion(*_, **kwargs):
id_ = kwargs["chat_id"] == message.chat_id
action = kwargs["action"] == ChatAction.TYPING
return id_ and action
assert check_shortcut_signature(
Message.reply_chat_action, Bot.send_chat_action, ["chat_id"], []
)
assert await check_shortcut_call(
message.reply_chat_action, message.get_bot(), "send_chat_action"
)
assert await check_defaults_handling(message.reply_chat_action, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_chat_action", make_assertion)
assert await message.reply_chat_action(action=ChatAction.TYPING)
async def test_reply_game(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id and kwargs["game_short_name"] == "test_game"
)
assert check_shortcut_signature(Message.reply_game, Bot.send_game, ["chat_id"], ["quote"])
assert await check_shortcut_call(message.reply_game, message.get_bot(), "send_game")
assert await check_defaults_handling(message.reply_game, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_game", make_assertion)
assert await message.reply_game(game_short_name="test_game")
assert await message.reply_game(game_short_name="test_game", quote=True)
async def test_reply_invoice(self, monkeypatch, message):
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"] == message.chat_id and args
assert check_shortcut_signature(
Message.reply_invoice, Bot.send_invoice, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.reply_invoice, message.get_bot(), "send_invoice")
assert await check_defaults_handling(message.reply_invoice, message.get_bot())
monkeypatch.setattr(message.get_bot(), "send_invoice", make_assertion)
assert await message.reply_invoice(
"title",
"description",
"payload",
"provider_token",
"currency",
"prices",
)
assert await message.reply_invoice(
"title",
"description",
"payload",
"provider_token",
"currency",
"prices",
quote=True,
)
2023-03-25 19:18:04 +01:00
@pytest.mark.parametrize(("disable_notification", "protected"), [(False, True), (True, False)])
async def test_forward(self, monkeypatch, message, disable_notification, protected):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == 123456
from_chat = kwargs["from_chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
notification = kwargs["disable_notification"] == disable_notification
protected_cont = kwargs["protect_content"] == protected
2022-01-03 08:13:33 +01:00
return chat_id and from_chat and message_id and notification and protected_cont
assert check_shortcut_signature(
Message.forward, Bot.forward_message, ["from_chat_id", "message_id"], []
)
assert await check_shortcut_call(message.forward, message.get_bot(), "forward_message")
assert await check_defaults_handling(message.forward, message.get_bot())
monkeypatch.setattr(message.get_bot(), "forward_message", make_assertion)
assert await message.forward(
2022-01-03 08:13:33 +01:00
123456, disable_notification=disable_notification, protect_content=protected
)
assert not await message.forward(635241)
2023-03-25 19:18:04 +01:00
@pytest.mark.parametrize(("disable_notification", "protected"), [(True, False), (False, True)])
async def test_copy(self, monkeypatch, message, disable_notification, protected):
keyboard = [[1, 2]]
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == 123456
from_chat = kwargs["from_chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
notification = kwargs["disable_notification"] == disable_notification
protected_cont = kwargs["protect_content"] == protected
if kwargs.get("reply_markup") is not None:
reply_markup = kwargs["reply_markup"] is keyboard
else:
reply_markup = True
2022-01-03 08:13:33 +01:00
return (
chat_id
and from_chat
and message_id
and notification
and reply_markup
and protected_cont
)
assert check_shortcut_signature(
Message.copy, Bot.copy_message, ["from_chat_id", "message_id"], []
)
assert await check_shortcut_call(message.copy, message.get_bot(), "copy_message")
assert await check_defaults_handling(message.copy, message.get_bot())
monkeypatch.setattr(message.get_bot(), "copy_message", make_assertion)
assert await message.copy(
2022-01-03 08:13:33 +01:00
123456, disable_notification=disable_notification, protect_content=protected
)
assert await message.copy(
2022-01-03 08:13:33 +01:00
123456,
reply_markup=keyboard,
disable_notification=disable_notification,
protect_content=protected,
)
assert not await message.copy(635241)
2023-03-25 19:18:04 +01:00
@pytest.mark.parametrize(("disable_notification", "protected"), [(True, False), (False, True)])
async def test_reply_copy(self, monkeypatch, message, disable_notification, protected):
keyboard = [[1, 2]]
async def make_assertion(*_, **kwargs):
chat_id = kwargs["from_chat_id"] == 123456
from_chat = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == 456789
notification = kwargs["disable_notification"] == disable_notification
is_protected = kwargs["protect_content"] == protected
if kwargs.get("reply_markup") is not None:
reply_markup = kwargs["reply_markup"] is keyboard
else:
reply_markup = True
if kwargs.get("reply_to_message_id") is not None:
reply = kwargs["reply_to_message_id"] == message.message_id
else:
reply = True
2022-01-03 08:13:33 +01:00
return (
chat_id
and from_chat
and message_id
and notification
and reply_markup
and reply
and is_protected
)
assert check_shortcut_signature(
Message.reply_copy, Bot.copy_message, ["chat_id"], ["quote"]
)
assert await check_shortcut_call(message.copy, message.get_bot(), "copy_message")
assert await check_defaults_handling(message.copy, message.get_bot())
monkeypatch.setattr(message.get_bot(), "copy_message", make_assertion)
assert await message.reply_copy(
2022-01-03 08:13:33 +01:00
123456, 456789, disable_notification=disable_notification, protect_content=protected
)
assert await message.reply_copy(
2022-01-03 08:13:33 +01:00
123456,
456789,
reply_markup=keyboard,
disable_notification=disable_notification,
protect_content=protected,
)
assert await message.reply_copy(
2022-01-03 08:13:33 +01:00
123456,
456789,
quote=True,
disable_notification=disable_notification,
protect_content=protected,
)
assert await message.reply_copy(
123456,
456789,
quote=True,
reply_to_message_id=message.message_id,
disable_notification=disable_notification,
2022-01-03 08:13:33 +01:00
protect_content=protected,
)
async def test_edit_text(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
text = kwargs["text"] == "test"
return chat_id and message_id and text
assert check_shortcut_signature(
Message.edit_text,
Bot.edit_message_text,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.edit_text,
message.get_bot(),
"edit_message_text",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.edit_text, message.get_bot())
monkeypatch.setattr(message.get_bot(), "edit_message_text", make_assertion)
assert await message.edit_text(text="test")
async def test_edit_caption(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
caption = kwargs["caption"] == "new caption"
return chat_id and message_id and caption
assert check_shortcut_signature(
Message.edit_caption,
Bot.edit_message_caption,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.edit_caption,
message.get_bot(),
"edit_message_caption",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.edit_caption, message.get_bot())
monkeypatch.setattr(message.get_bot(), "edit_message_caption", make_assertion)
assert await message.edit_caption(caption="new caption")
async def test_edit_media(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
media = kwargs["media"] == "my_media"
return chat_id and message_id and media
assert check_shortcut_signature(
Message.edit_media,
Bot.edit_message_media,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.edit_media,
message.get_bot(),
"edit_message_media",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.edit_media, message.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(message.get_bot(), "edit_message_media", make_assertion)
assert await message.edit_media("my_media")
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_edit_reply_markup(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
reply_markup = kwargs["reply_markup"] == [["1", "2"]]
return chat_id and message_id and reply_markup
assert check_shortcut_signature(
Message.edit_reply_markup,
Bot.edit_message_reply_markup,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.edit_reply_markup,
message.get_bot(),
"edit_message_reply_markup",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.edit_reply_markup, message.get_bot())
monkeypatch.setattr(message.get_bot(), "edit_message_reply_markup", make_assertion)
assert await message.edit_reply_markup(reply_markup=[["1", "2"]])
async def test_edit_live_location(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
latitude = kwargs["latitude"] == 1
longitude = kwargs["longitude"] == 2
return chat_id and message_id and longitude and latitude
assert check_shortcut_signature(
Message.edit_live_location,
Bot.edit_message_live_location,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.edit_live_location,
message.get_bot(),
"edit_message_live_location",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.edit_live_location, message.get_bot())
monkeypatch.setattr(message.get_bot(), "edit_message_live_location", make_assertion)
assert await message.edit_live_location(latitude=1, longitude=2)
async def test_stop_live_location(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
return chat_id and message_id
assert check_shortcut_signature(
Message.stop_live_location,
Bot.stop_message_live_location,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.stop_live_location,
message.get_bot(),
"stop_message_live_location",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.stop_live_location, message.get_bot())
monkeypatch.setattr(message.get_bot(), "stop_message_live_location", make_assertion)
assert await message.stop_live_location()
async def test_set_game_score(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
user_id = kwargs["user_id"] == 1
score = kwargs["score"] == 2
return chat_id and message_id and user_id and score
assert check_shortcut_signature(
Message.set_game_score,
Bot.set_game_score,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.set_game_score,
message.get_bot(),
"set_game_score",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.set_game_score, message.get_bot())
monkeypatch.setattr(message.get_bot(), "set_game_score", make_assertion)
assert await message.set_game_score(user_id=1, score=2)
async def test_get_game_high_scores(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
user_id = kwargs["user_id"] == 1
return chat_id and message_id and user_id
assert check_shortcut_signature(
Message.get_game_high_scores,
Bot.get_game_high_scores,
["chat_id", "message_id", "inline_message_id"],
[],
)
assert await check_shortcut_call(
message.get_game_high_scores,
message.get_bot(),
"get_game_high_scores",
skip_params=["inline_message_id"],
shortcut_kwargs=["message_id", "chat_id"],
)
assert await check_defaults_handling(message.get_game_high_scores, message.get_bot())
monkeypatch.setattr(message.get_bot(), "get_game_high_scores", make_assertion)
assert await message.get_game_high_scores(user_id=1)
async def test_delete(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
return chat_id and message_id
assert check_shortcut_signature(
Message.delete, Bot.delete_message, ["chat_id", "message_id"], []
)
assert await check_shortcut_call(message.delete, message.get_bot(), "delete_message")
assert await check_defaults_handling(message.delete, message.get_bot())
monkeypatch.setattr(message.get_bot(), "delete_message", make_assertion)
assert await message.delete()
async def test_stop_poll(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
return chat_id and message_id
assert check_shortcut_signature(
Message.stop_poll, Bot.stop_poll, ["chat_id", "message_id"], []
)
assert await check_shortcut_call(message.stop_poll, message.get_bot(), "stop_poll")
assert await check_defaults_handling(message.stop_poll, message.get_bot())
monkeypatch.setattr(message.get_bot(), "stop_poll", make_assertion)
assert await message.stop_poll()
async def test_pin(self, monkeypatch, message):
async def make_assertion(*args, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
return chat_id and message_id
assert check_shortcut_signature(
Message.pin, Bot.pin_chat_message, ["chat_id", "message_id"], []
)
assert await check_shortcut_call(message.pin, message.get_bot(), "pin_chat_message")
assert await check_defaults_handling(message.pin, message.get_bot())
monkeypatch.setattr(message.get_bot(), "pin_chat_message", make_assertion)
assert await message.pin()
async def test_unpin(self, monkeypatch, message):
async def make_assertion(*args, **kwargs):
chat_id = kwargs["chat_id"] == message.chat_id
message_id = kwargs["message_id"] == message.message_id
return chat_id and message_id
assert check_shortcut_signature(
Message.unpin, Bot.unpin_chat_message, ["chat_id", "message_id"], []
)
assert await check_shortcut_call(
message.unpin,
message.get_bot(),
"unpin_chat_message",
shortcut_kwargs=["chat_id", "message_id"],
)
assert await check_defaults_handling(message.unpin, message.get_bot())
monkeypatch.setattr(message.get_bot(), "unpin_chat_message", make_assertion)
assert await message.unpin()
def test_default_quote(self, message):
message.get_bot()._defaults = Defaults()
try:
message.get_bot().defaults._quote = False
assert message._quote(None, None) is None
message.get_bot().defaults._quote = True
assert message._quote(None, None) == message.message_id
message.get_bot().defaults._quote = None
message.chat.type = Chat.PRIVATE
assert message._quote(None, None) is None
message.chat.type = Chat.GROUP
assert message._quote(None, None)
finally:
message.get_bot()._defaults = None
async def test_edit_forum_topic(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id
and kwargs["message_thread_id"] == message.message_thread_id
and kwargs["name"] == "New Name"
and kwargs["icon_custom_emoji_id"] == "12345"
)
assert check_shortcut_signature(
Message.edit_forum_topic, Bot.edit_forum_topic, ["chat_id", "message_thread_id"], []
)
assert await check_shortcut_call(
message.edit_forum_topic,
message.get_bot(),
"edit_forum_topic",
shortcut_kwargs=["chat_id", "message_thread_id"],
)
assert await check_defaults_handling(message.edit_forum_topic, message.get_bot())
monkeypatch.setattr(message.get_bot(), "edit_forum_topic", make_assertion)
assert await message.edit_forum_topic(name="New Name", icon_custom_emoji_id="12345")
async def test_close_forum_topic(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id
and kwargs["message_thread_id"] == message.message_thread_id
)
assert check_shortcut_signature(
Message.close_forum_topic, Bot.close_forum_topic, ["chat_id", "message_thread_id"], []
)
assert await check_shortcut_call(
message.close_forum_topic,
message.get_bot(),
"close_forum_topic",
shortcut_kwargs=["chat_id", "message_thread_id"],
)
assert await check_defaults_handling(message.close_forum_topic, message.get_bot())
monkeypatch.setattr(message.get_bot(), "close_forum_topic", make_assertion)
assert await message.close_forum_topic()
async def test_reopen_forum_topic(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id
and kwargs["message_thread_id"] == message.message_thread_id
)
assert check_shortcut_signature(
Message.reopen_forum_topic,
Bot.reopen_forum_topic,
["chat_id", "message_thread_id"],
[],
)
assert await check_shortcut_call(
message.reopen_forum_topic,
message.get_bot(),
"reopen_forum_topic",
shortcut_kwargs=["chat_id", "message_thread_id"],
)
assert await check_defaults_handling(message.reopen_forum_topic, message.get_bot())
monkeypatch.setattr(message.get_bot(), "reopen_forum_topic", make_assertion)
assert await message.reopen_forum_topic()
async def test_delete_forum_topic(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id
and kwargs["message_thread_id"] == message.message_thread_id
)
assert check_shortcut_signature(
Message.delete_forum_topic,
Bot.delete_forum_topic,
["chat_id", "message_thread_id"],
[],
)
assert await check_shortcut_call(
message.delete_forum_topic,
message.get_bot(),
"delete_forum_topic",
shortcut_kwargs=["chat_id", "message_thread_id"],
)
assert await check_defaults_handling(message.delete_forum_topic, message.get_bot())
monkeypatch.setattr(message.get_bot(), "delete_forum_topic", make_assertion)
assert await message.delete_forum_topic()
async def test_unpin_all_forum_topic_messages(self, monkeypatch, message):
async def make_assertion(*_, **kwargs):
return (
kwargs["chat_id"] == message.chat_id
and kwargs["message_thread_id"] == message.message_thread_id
)
assert check_shortcut_signature(
Message.unpin_all_forum_topic_messages,
Bot.unpin_all_forum_topic_messages,
["chat_id", "message_thread_id"],
[],
)
assert await check_shortcut_call(
message.unpin_all_forum_topic_messages,
message.get_bot(),
"unpin_all_forum_topic_messages",
shortcut_kwargs=["chat_id", "message_thread_id"],
)
assert await check_defaults_handling(
message.unpin_all_forum_topic_messages, message.get_bot()
)
monkeypatch.setattr(message.get_bot(), "unpin_all_forum_topic_messages", make_assertion)
assert await message.unpin_all_forum_topic_messages()