2965f5d47d
Sync telesrv 00630bc (feat(botapi): support formatted text parse modes). Skipped telesrv docs changes per public sync rules.
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import AsyncMock
|
|
|
|
from aiogram.enums import ParseMode
|
|
|
|
|
|
MODULE_PATH = Path(__file__).with_name("demo.py")
|
|
SPEC = importlib.util.spec_from_file_location("bedolagaformat_demo", MODULE_PATH)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
demo = importlib.util.module_from_spec(SPEC)
|
|
sys.modules[SPEC.name] = demo
|
|
SPEC.loader.exec_module(demo)
|
|
|
|
|
|
class SentMessage:
|
|
def __init__(self, message_id: int) -> None:
|
|
self.message_id = message_id
|
|
|
|
|
|
class BedolagaFormatDemoTest(unittest.IsolatedAsyncioTestCase):
|
|
def test_format_samples_cover_default_and_explicit_modes(self) -> None:
|
|
samples = demo.format_samples("BEDOLAGA123")
|
|
self.assertEqual(
|
|
[sample.parse_mode for sample in samples],
|
|
[None, ParseMode.MARKDOWN, ParseMode.MARKDOWN_V2],
|
|
)
|
|
self.assertIn("<b>BEDOLAGA123 Default HTML</b>", samples[0].text)
|
|
self.assertIn("*BEDOLAGA123 Markdown*", samples[1].text)
|
|
self.assertIn("*BEDOLAGA123 MarkdownV2*", samples[2].text)
|
|
|
|
def test_markdown_v2_escapes_reserved_marker_characters(self) -> None:
|
|
samples = demo.format_samples("BEDOLAGA-FULL-20260720")
|
|
self.assertIn(r"BEDOLAGA\-FULL\-20260720", samples[2].text)
|
|
|
|
async def test_send_suite_omits_parse_mode_only_for_default_html(self) -> None:
|
|
bot = AsyncMock()
|
|
bot.send_message.side_effect = [SentMessage(11), SentMessage(12), SentMessage(13)]
|
|
|
|
message_ids = await demo.send_format_suite(bot, 1780243200, "BEDOLAGA123")
|
|
|
|
self.assertEqual(message_ids, [11, 12, 13])
|
|
calls = bot.send_message.await_args_list
|
|
self.assertNotIn("parse_mode", calls[0].kwargs)
|
|
self.assertEqual(calls[1].kwargs["parse_mode"], ParseMode.MARKDOWN)
|
|
self.assertEqual(calls[2].kwargs["parse_mode"], ParseMode.MARKDOWN_V2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|