47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from telegram import error
|
|
|
|
from apps.telegram.client import TelegramClient
|
|
from apps.telegram.models import TelegramMessage
|
|
from apps.telegram.tasks import send_log_and_actions_message
|
|
|
|
|
|
@patch.object(TelegramClient, "send_raw_message", side_effect=error.BadRequest("Message to reply not found"))
|
|
@pytest.mark.django_db
|
|
def test_send_log_and_actions_replied_message_not_found(
|
|
mock_send_message,
|
|
make_organization_and_user,
|
|
make_telegram_user_connector,
|
|
make_alert_receive_channel,
|
|
make_alert_group,
|
|
make_alert,
|
|
make_telegram_message,
|
|
caplog,
|
|
):
|
|
# set up a user with Telegram account connected
|
|
organization, user = make_organization_and_user()
|
|
make_telegram_user_connector(user)
|
|
|
|
# create an alert group with an existing Telegram message in user's DM
|
|
alert_receive_channel = make_alert_receive_channel(organization)
|
|
alert_group = make_alert_group(alert_receive_channel)
|
|
make_alert(alert_group=alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
|
telegram_message = make_telegram_message(
|
|
alert_group=alert_group,
|
|
message_type=TelegramMessage.PERSONAL_MESSAGE,
|
|
chat_id=str(user.telegram_connection.telegram_chat_id),
|
|
message_id=123,
|
|
)
|
|
|
|
reply_to_message_id = 321
|
|
send_log_and_actions_message(
|
|
telegram_message.chat_id, "group_chat_id", telegram_message.message_id, reply_to_message_id
|
|
)
|
|
|
|
expected_msg = (
|
|
f"Could not send log and actions messages to Telegram group with id group_chat_id "
|
|
f"due to 'Message to reply not found'. alert_group {alert_group.pk}"
|
|
)
|
|
assert expected_msg in caplog.text
|