mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
Merge remote-tracking branch 'origin/feature/bot-api-7.0' into feature/bot-api-7.0
This commit is contained in:
commit
c658514817
6 changed files with 153 additions and 0 deletions
28
tests/test_api/test_methods/test_copy_messages.py
Normal file
28
tests/test_api/test_methods/test_copy_messages.py
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
from random import randint
|
||||
from aiogram.methods import CopyMessages
|
||||
from aiogram.types import MessageId
|
||||
from tests.mocked_bot import MockedBot
|
||||
|
||||
|
||||
class TestCopyMessages:
|
||||
async def test_bot_method(self, bot: MockedBot):
|
||||
prepare_result = bot.add_result_for(
|
||||
CopyMessages,
|
||||
ok=True,
|
||||
result=[
|
||||
MessageId(message_id=randint(100, 200)),
|
||||
MessageId(message_id=randint(300, 400)),
|
||||
],
|
||||
)
|
||||
|
||||
response: list[MessageId] = await bot.copy_messages(
|
||||
chat_id=randint(1000, 9999),
|
||||
from_chat_id=randint(1000, 9999),
|
||||
message_ids=[
|
||||
randint(1000, 4999),
|
||||
randint(5000, 9999),
|
||||
],
|
||||
)
|
||||
request = bot.get_request()
|
||||
assert request
|
||||
assert response == prepare_result.result
|
||||
19
tests/test_api/test_methods/test_delete_messages.py
Normal file
19
tests/test_api/test_methods/test_delete_messages.py
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
from aiogram.methods import DeleteMessages
|
||||
from tests.mocked_bot import MockedBot
|
||||
|
||||
|
||||
class TestDeleteMessages:
|
||||
async def test_bot_method(self, bot: MockedBot):
|
||||
prepare_result = bot.add_result_for(
|
||||
DeleteMessages,
|
||||
ok=True,
|
||||
result=True,
|
||||
)
|
||||
|
||||
response: bool = await bot.delete_messages(
|
||||
chat_id=42,
|
||||
message_ids=[13, 77],
|
||||
)
|
||||
request = bot.get_request()
|
||||
assert request
|
||||
assert response == prepare_result.result
|
||||
|
|
@ -20,4 +20,5 @@ class TestForwardMessage:
|
|||
|
||||
response: Message = await bot.forward_message(chat_id=42, from_chat_id=42, message_id=42)
|
||||
request = bot.get_request()
|
||||
assert request
|
||||
assert response == prepare_result.result
|
||||
|
|
|
|||
28
tests/test_api/test_methods/test_forward_messages.py
Normal file
28
tests/test_api/test_methods/test_forward_messages.py
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
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
|
||||
52
tests/test_api/test_methods/test_get_user_chat_boosts.py
Normal file
52
tests/test_api/test_methods/test_get_user_chat_boosts.py
Normal file
|
|
@ -0,0 +1,52 @@
|
|||
from datetime import datetime, timedelta
|
||||
from random import randint
|
||||
|
||||
from aiogram.methods import GetUserChatBoosts
|
||||
from aiogram.types import (
|
||||
ChatBoost,
|
||||
ChatBoostSourceGiveaway,
|
||||
ChatBoostSourcePremium,
|
||||
User,
|
||||
UserChatBoosts,
|
||||
)
|
||||
from tests.mocked_bot import MockedBot
|
||||
|
||||
|
||||
class TestGetUserChatBoosts:
|
||||
async def test_bot_method(self, bot: MockedBot):
|
||||
now = datetime.now()
|
||||
user = User(
|
||||
id=randint(200, 500),
|
||||
is_bot=False,
|
||||
first_name="name",
|
||||
)
|
||||
prepare_result = bot.add_result_for(
|
||||
GetUserChatBoosts,
|
||||
ok=True,
|
||||
result=UserChatBoosts(
|
||||
boosts=[
|
||||
ChatBoost(
|
||||
boost_id="eggs",
|
||||
add_date=now - timedelta(days=7),
|
||||
expiration_date=now + timedelta(days=14),
|
||||
source=ChatBoostSourceGiveaway(
|
||||
giveaway_message_id=randint(100, 300),
|
||||
),
|
||||
),
|
||||
ChatBoost(
|
||||
boost_id="spam",
|
||||
add_date=now - timedelta(days=3),
|
||||
expiration_date=now + timedelta(days=21),
|
||||
source=ChatBoostSourcePremium(user=user),
|
||||
),
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
response: UserChatBoosts = await bot.get_user_chat_boosts(
|
||||
chat_id=randint(100, 200),
|
||||
user_id=user.id,
|
||||
)
|
||||
request = bot.get_request()
|
||||
assert request
|
||||
assert response == prepare_result.result
|
||||
25
tests/test_api/test_methods/test_set_message_reaction.py
Normal file
25
tests/test_api/test_methods/test_set_message_reaction.py
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
from random import randint
|
||||
|
||||
from aiogram.methods import SetMessageReaction
|
||||
from aiogram.types import ReactionTypeCustomEmoji
|
||||
from tests.mocked_bot import MockedBot
|
||||
|
||||
|
||||
class TestSetMessageReaction:
|
||||
async def test_bot_method(self, bot: MockedBot):
|
||||
prepare_result = bot.add_result_for(
|
||||
SetMessageReaction,
|
||||
ok=True,
|
||||
result=True,
|
||||
)
|
||||
|
||||
response: bool = await bot.set_message_reaction(
|
||||
chat_id=randint(200, 300),
|
||||
message_id=randint(100, 200),
|
||||
reaction=[
|
||||
ReactionTypeCustomEmoji(custom_emoji="qwerty"),
|
||||
],
|
||||
)
|
||||
request = bot.get_request()
|
||||
assert request
|
||||
assert response == prepare_result.result
|
||||
Loading…
Add table
Add a link
Reference in a new issue