mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-14 10:53:23 +00:00
30 lines
857 B
Python
30 lines
857 B
Python
|
|
from random import randint
|
||
|
|
|
||
|
|
from aiogram.methods import ForwardMessages
|
||
|
|
from aiogram.types import MessageId
|
||
|
|
from tests.mocked_bot import MockedBot
|
||
|
|
|
||
|
|
|
||
|
|
class TestForwardMessages:
|
||
|
|
async def test_bot_method(self, bot: MockedBot):
|
||
|
|
prepare_result = bot.add_result_for(
|
||
|
|
ForwardMessages,
|
||
|
|
ok=True,
|
||
|
|
result=[
|
||
|
|
MessageId(message_id=randint(100, 200)),
|
||
|
|
MessageId(message_id=randint(200, 300)),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
response: list[MessageId] = await bot.forward_messages(
|
||
|
|
chat_id=randint(10, 50),
|
||
|
|
from_chat_id=randint(50, 99),
|
||
|
|
message_ids=[
|
||
|
|
randint(400, 500),
|
||
|
|
randint(600, 700),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
request = bot.get_request()
|
||
|
|
assert request
|
||
|
|
assert response == prepare_result.result
|