From 66e69e3c937dd96c0333fbba4f419f3f1a523ebf Mon Sep 17 00:00:00 2001 From: JRoot Junior Date: Sat, 30 Dec 2023 21:39:38 +0200 Subject: [PATCH] Test parsing CallbackQuery message (inaccessible and accessible) --- .../test_types/test_callback_query.py | 33 ++++++++++++++++++- 1 file changed, 32 insertions(+), 1 deletion(-) diff --git a/tests/test_api/test_types/test_callback_query.py b/tests/test_api/test_types/test_callback_query.py index 3eae5a07..a0569911 100644 --- a/tests/test_api/test_types/test_callback_query.py +++ b/tests/test_api/test_types/test_callback_query.py @@ -1,5 +1,5 @@ from aiogram.methods import AnswerCallbackQuery -from aiogram.types import CallbackQuery, User +from aiogram.types import CallbackQuery, User, Message, InaccessibleMessage class TestCallbackQuery: @@ -17,3 +17,34 @@ class TestCallbackQuery: for key, value in kwargs.items(): assert getattr(api_method, key) == value + + def test_parse_message(self): + data = { + "id": "id", + "from": {"id": 42, "is_bot": False, "first_name": "name"}, + "message": { + "message_id": 123, + "date": 1234567890, + "chat": {"id": 42, "type": "private"}, + }, + "chat_instance": "chat", + "data": "data", + } + callback_query = CallbackQuery.model_validate(data) + assert isinstance(callback_query.message, Message) + + def test_parse_inaccessible_message(self): + data = { + "id": "id", + "from": {"id": 42, "is_bot": False, "first_name": "name"}, + "message": { + "message_id": 123, + "date": 0, + "chat": {"id": 42, "type": "private"}, + }, + "chat_instance": "chat", + "data": "data", + } + callback_query = CallbackQuery.model_validate(data) + + assert isinstance(callback_query.message, InaccessibleMessage)