From 9154d435069d73e6ef87c80b3e8117ab2ca16e60 Mon Sep 17 00:00:00 2001
From: JRoot Junior <jroot.junior@gmail.com>
Date: Mon, 6 May 2024 23:04:14 +0300
Subject: [PATCH] Fixed tests

---
 aiogram/types/message.py                     |  2 ++
 tests/test_api/test_methods/test_get_chat.py | 16 ++++++++++++----
 tests/test_api/test_types/test_message.py    | 17 +++++++++++++++++
 3 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/aiogram/types/message.py b/aiogram/types/message.py
index 34eb2d1b..d96febc7 100644
--- a/aiogram/types/message.py
+++ b/aiogram/types/message.py
@@ -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
 
diff --git a/tests/test_api/test_methods/test_get_chat.py b/tests/test_api/test_methods/test_get_chat.py
index 902b03b3..9f67b0c7 100644
--- a/tests/test_api/test_methods/test_get_chat.py
+++ b/tests/test_api/test_methods/test_get_chat.py
@@ -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
diff --git a/tests/test_api/test_types/test_message.py b/tests/test_api/test_types/test_message.py
index a17228b4..fd65119f 100644
--- a/tests/test_api/test_types/test_message.py
+++ b/tests/test_api/test_types/test_message.py
@@ -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],
 ]