Merge remote-tracking branch 'origin/feature/bot-api-7.0' into feature/bot-api-7.0

This commit is contained in:
JRoot Junior 2023-12-30 21:28:16 +02:00
commit c658514817
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2
6 changed files with 153 additions and 0 deletions

View 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

View 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

View file

@ -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

View 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

View 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

View 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