Fixed tests

This commit is contained in:
JRoot Junior 2024-05-06 23:04:14 +03:00
parent 3228954def
commit 9154d43506
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2
3 changed files with 31 additions and 4 deletions

View file

@ -630,6 +630,8 @@ class Message(MaybeInaccessibleMessage):
return ContentType.STORY
if self.write_access_allowed:
return ContentType.WRITE_ACCESS_ALLOWED
if self.chat_background_set:
return ContentType.CHAT_BACKGROUND_SET
if self.boost_added:
return ContentType.BOOST_ADDED

View file

@ -1,14 +1,22 @@
from aiogram.methods import GetChat, Request
from aiogram.types import Chat
from aiogram.methods import GetChat
from aiogram.types import ChatFullInfo
from tests.mocked_bot import MockedBot
class TestGetChat:
async def test_bot_method(self, bot: MockedBot):
prepare_result = bot.add_result_for(
GetChat, ok=True, result=Chat(id=-42, type="channel", title="chat")
GetChat,
ok=True,
result=ChatFullInfo(
id=-42,
type="channel",
title="chat",
accent_color_id=0,
max_reaction_count=0,
),
)
response: Chat = await bot.get_chat(chat_id=-42)
response: ChatFullInfo = await bot.get_chat(chat_id=-42)
request = bot.get_request()
assert response == prepare_result.result

View file

@ -88,6 +88,9 @@ from aiogram.types import (
Voice,
WebAppData,
WriteAccessAllowed,
ChatBackground,
BackgroundTypeFill,
BackgroundFillSolid,
)
from aiogram.types.message import ContentType, Message
@ -553,6 +556,18 @@ TEST_MESSAGE_BOOST_ADDED = Message(
from_user=User(id=42, is_bot=False, first_name="User"),
boost_added=ChatBoostAdded(boost_count=1),
)
TEST_CHAT_BACKGROUND_SET = Message(
message_id=42,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
from_user=User(id=42, is_bot=False, first_name="User"),
chat_background_set=ChatBackground(
type=BackgroundTypeFill(
fill=BackgroundFillSolid(color=0x000000),
dark_theme_dimming=0,
)
),
)
TEST_MESSAGE_UNKNOWN = Message(
message_id=42,
date=datetime.datetime.now(),
@ -620,6 +635,7 @@ MESSAGES_AND_CONTENT_TYPES = [
[TEST_MESSAGE_GENERAL_FORUM_TOPIC_UNHIDDEN, ContentType.GENERAL_FORUM_TOPIC_UNHIDDEN],
[TEST_MESSAGE_WRITE_ACCESS_ALLOWED, ContentType.WRITE_ACCESS_ALLOWED],
[TEST_MESSAGE_BOOST_ADDED, ContentType.BOOST_ADDED],
[TEST_CHAT_BACKGROUND_SET, ContentType.CHAT_BACKGROUND_SET],
[TEST_MESSAGE_UNKNOWN, ContentType.UNKNOWN],
]
@ -678,6 +694,7 @@ MESSAGES_AND_COPY_METHODS = [
[TEST_MESSAGE_GIVEAWAY, None],
[TEST_MESSAGE_GIVEAWAY_WINNERS, None],
[TEST_MESSAGE_BOOST_ADDED, None],
[TEST_CHAT_BACKGROUND_SET, None],
[TEST_MESSAGE_UNKNOWN, None],
]