Test parsing CallbackQuery message (inaccessible and accessible)

This commit is contained in:
JRoot Junior 2023-12-30 21:39:38 +02:00
parent 81f533b3a2
commit 66e69e3c93
No known key found for this signature in database
GPG key ID: 738964250D5FF6E2

View file

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