101 lines
3.8 KiB
Python
101 lines
3.8 KiB
Python
|
import pytest
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from apps.alerts.models import AlertGroupLogRecord, AlertReceiveChannel
|
||
|
|
from apps.mattermost.alert_group_representative import AlertGroupMattermostRepresentative
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_get_handler(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_mattermost_channel,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_alert_group_log_record,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(
|
||
|
|
organization, integration=AlertReceiveChannel.INTEGRATION_GRAFANA
|
||
|
|
)
|
||
|
|
make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
ack_alert_group = make_alert_group(
|
||
|
|
alert_receive_channel=alert_receive_channel,
|
||
|
|
acknowledged_at=timezone.now(),
|
||
|
|
acknowledged=True,
|
||
|
|
)
|
||
|
|
make_alert(alert_group=ack_alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
log_record = make_alert_group_log_record(ack_alert_group, type=AlertGroupLogRecord.TYPE_ACK, author=None)
|
||
|
|
handler = AlertGroupMattermostRepresentative(log_record=log_record).get_handler()
|
||
|
|
assert handler.__name__ == "on_alert_group_action"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_is_applicable_success(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_mattermost_channel,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_alert_group_log_record,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(
|
||
|
|
organization, integration=AlertReceiveChannel.INTEGRATION_GRAFANA
|
||
|
|
)
|
||
|
|
make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
ack_alert_group = make_alert_group(
|
||
|
|
alert_receive_channel=alert_receive_channel,
|
||
|
|
acknowledged_at=timezone.now(),
|
||
|
|
acknowledged=True,
|
||
|
|
)
|
||
|
|
make_alert(alert_group=ack_alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
log_record = make_alert_group_log_record(ack_alert_group, type=AlertGroupLogRecord.TYPE_ACK, author=None)
|
||
|
|
assert AlertGroupMattermostRepresentative(log_record=log_record).is_applicable()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_is_applicable_without_channels(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_alert_group_log_record,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(
|
||
|
|
organization, integration=AlertReceiveChannel.INTEGRATION_GRAFANA
|
||
|
|
)
|
||
|
|
ack_alert_group = make_alert_group(
|
||
|
|
alert_receive_channel=alert_receive_channel,
|
||
|
|
acknowledged_at=timezone.now(),
|
||
|
|
acknowledged=True,
|
||
|
|
)
|
||
|
|
make_alert(alert_group=ack_alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
log_record = make_alert_group_log_record(ack_alert_group, type=AlertGroupLogRecord.TYPE_ACK, author=None)
|
||
|
|
assert not AlertGroupMattermostRepresentative(log_record=log_record).is_applicable()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_is_applicable_invalid_type(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_mattermost_channel,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_alert_group_log_record,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(
|
||
|
|
organization, integration=AlertReceiveChannel.INTEGRATION_GRAFANA
|
||
|
|
)
|
||
|
|
make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
ack_alert_group = make_alert_group(
|
||
|
|
alert_receive_channel=alert_receive_channel,
|
||
|
|
acknowledged_at=timezone.now(),
|
||
|
|
acknowledged=True,
|
||
|
|
)
|
||
|
|
make_alert(alert_group=ack_alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
log_record = make_alert_group_log_record(ack_alert_group, type=AlertGroupLogRecord.TYPE_RE_INVITE, author=None)
|
||
|
|
assert not AlertGroupMattermostRepresentative(log_record=log_record).is_applicable()
|