banteg
129 lines
3.6 KiB
Python
129 lines
3.6 KiB
Python
from takopi.telegram import (
|
|
TelegramCallbackQuery,
|
|
TelegramIncomingMessage,
|
|
parse_incoming_update,
|
|
)
|
|
|
|
|
|
def test_parse_incoming_update_maps_fields() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"message": {
|
|
"message_id": 10,
|
|
"text": "hello",
|
|
"chat": {"id": 123, "type": "supergroup", "is_forum": True},
|
|
"from": {"id": 99},
|
|
"reply_to_message": {"message_id": 5, "text": "prev"},
|
|
},
|
|
}
|
|
|
|
msg = parse_incoming_update(update, chat_id=123)
|
|
assert msg is not None
|
|
assert isinstance(msg, TelegramIncomingMessage)
|
|
assert msg.transport == "telegram"
|
|
assert msg.chat_id == 123
|
|
assert msg.message_id == 10
|
|
assert msg.text == "hello"
|
|
assert msg.reply_to_message_id == 5
|
|
assert msg.reply_to_text == "prev"
|
|
assert msg.sender_id == 99
|
|
assert msg.thread_id is None
|
|
assert msg.is_topic_message is None
|
|
assert msg.chat_type == "supergroup"
|
|
assert msg.is_forum is True
|
|
assert msg.voice is None
|
|
assert msg.raw == update["message"]
|
|
|
|
|
|
def test_parse_incoming_update_filters_non_matching_chat() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"message": {
|
|
"message_id": 10,
|
|
"text": "hello",
|
|
"chat": {"id": 123},
|
|
},
|
|
}
|
|
|
|
assert parse_incoming_update(update, chat_id=999) is None
|
|
|
|
|
|
def test_parse_incoming_update_filters_non_text_and_non_voice() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"message": {"message_id": 10, "chat": {"id": 123}, "photo": []},
|
|
}
|
|
|
|
assert parse_incoming_update(update, chat_id=123) is None
|
|
|
|
|
|
def test_parse_incoming_update_voice_message() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"message": {
|
|
"message_id": 10,
|
|
"chat": {"id": 123},
|
|
"voice": {
|
|
"file_id": "voice-id",
|
|
"file_unique_id": "uniq",
|
|
"duration": 3,
|
|
"mime_type": "audio/ogg",
|
|
"file_size": 1234,
|
|
},
|
|
},
|
|
}
|
|
|
|
msg = parse_incoming_update(update, chat_id=123)
|
|
assert msg is not None
|
|
assert isinstance(msg, TelegramIncomingMessage)
|
|
assert msg.text == ""
|
|
assert msg.voice is not None
|
|
assert msg.voice.file_id == "voice-id"
|
|
assert msg.voice.mime_type == "audio/ogg"
|
|
assert msg.voice.file_size == 1234
|
|
assert msg.voice.duration == 3
|
|
|
|
|
|
def test_parse_incoming_update_callback_query() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"callback_query": {
|
|
"id": "cbq-1",
|
|
"data": "takopi:cancel",
|
|
"from": {"id": 321},
|
|
"message": {
|
|
"message_id": 55,
|
|
"chat": {"id": 123},
|
|
},
|
|
},
|
|
}
|
|
|
|
msg = parse_incoming_update(update, chat_id=123)
|
|
assert isinstance(msg, TelegramCallbackQuery)
|
|
assert msg.transport == "telegram"
|
|
assert msg.chat_id == 123
|
|
assert msg.message_id == 55
|
|
assert msg.callback_query_id == "cbq-1"
|
|
assert msg.data == "takopi:cancel"
|
|
assert msg.sender_id == 321
|
|
|
|
|
|
def test_parse_incoming_update_topic_fields() -> None:
|
|
update = {
|
|
"update_id": 1,
|
|
"message": {
|
|
"message_id": 10,
|
|
"text": "hello",
|
|
"message_thread_id": 77,
|
|
"is_topic_message": True,
|
|
"chat": {"id": -100, "type": "supergroup", "is_forum": True},
|
|
},
|
|
}
|
|
|
|
msg = parse_incoming_update(update, chat_id=-100)
|
|
assert isinstance(msg, TelegramIncomingMessage)
|
|
assert msg.thread_id == 77
|
|
assert msg.is_topic_message is True
|
|
assert msg.chat_type == "supergroup"
|
|
assert msg.is_forum is True
|