91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from apps.mattermost.models import MattermostChannel
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_get_channel_for_alert_group(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_channel_filter,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_mattermost_channel,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(organization)
|
||
|
|
make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
channel = make_mattermost_channel(organization=organization)
|
||
|
|
channel_filter = make_channel_filter(
|
||
|
|
alert_receive_channel,
|
||
|
|
notification_backends={"MATTERMOST": {"channel": channel.public_primary_key, "enabled": True}},
|
||
|
|
)
|
||
|
|
|
||
|
|
alert_group = make_alert_group(alert_receive_channel, channel_filter=channel_filter)
|
||
|
|
make_alert(alert_group=alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
|
||
|
|
ch = MattermostChannel.get_channel_for_alert_group(alert_group)
|
||
|
|
assert ch.public_primary_key == channel.public_primary_key
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_get_mattermost_channel_disabled_for_route(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_channel_filter,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_mattermost_channel,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(organization)
|
||
|
|
channel = make_mattermost_channel(organization=organization)
|
||
|
|
channel_filter = make_channel_filter(
|
||
|
|
alert_receive_channel,
|
||
|
|
notification_backends={"MATTERMOST": {"channel": channel.public_primary_key, "enabled": False}},
|
||
|
|
)
|
||
|
|
|
||
|
|
alert_group = make_alert_group(alert_receive_channel, channel_filter=channel_filter)
|
||
|
|
make_alert(alert_group=alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
|
||
|
|
ch = MattermostChannel.get_channel_for_alert_group(alert_group)
|
||
|
|
assert ch is None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_get_mattermost_channel_invalid_route_channel(
|
||
|
|
make_organization,
|
||
|
|
make_alert_receive_channel,
|
||
|
|
make_channel_filter,
|
||
|
|
make_alert_group,
|
||
|
|
make_alert,
|
||
|
|
make_mattermost_channel,
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(organization)
|
||
|
|
default_channel = make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
channel_filter = make_channel_filter(
|
||
|
|
alert_receive_channel, notification_backends={"MATTERMOST": {"channel": "invalid_id", "enabled": True}}
|
||
|
|
)
|
||
|
|
|
||
|
|
alert_group = make_alert_group(alert_receive_channel, channel_filter=channel_filter)
|
||
|
|
make_alert(alert_group=alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
|
||
|
|
ch = MattermostChannel.get_channel_for_alert_group(alert_group)
|
||
|
|
assert ch.public_primary_key == default_channel.public_primary_key
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_get_mattermost_channel_channel_filter_not_configured(
|
||
|
|
make_organization, make_alert_receive_channel, make_alert_group, make_alert, make_mattermost_channel
|
||
|
|
):
|
||
|
|
organization = make_organization()
|
||
|
|
alert_receive_channel = make_alert_receive_channel(organization)
|
||
|
|
default_channel = make_mattermost_channel(organization=organization, is_default_channel=True)
|
||
|
|
|
||
|
|
alert_group = make_alert_group(alert_receive_channel)
|
||
|
|
make_alert(alert_group=alert_group, raw_request_data=alert_receive_channel.config.example_payload)
|
||
|
|
|
||
|
|
ch = MattermostChannel.get_channel_for_alert_group(alert_group)
|
||
|
|
assert ch.public_primary_key == default_channel.public_primary_key
|