e67d3519fe
* remove email verification related code * remove email verification related code * remove sendgrid callback * remove sendgrid related code * remove sendgrid related code * rename sendgrid app to email * remove email from built-in channels * remove email from built-in channels * remove email from built-in channels * add email backend: https://github.com/grafana/oncall/pull/50 * add email templater * add email templater * convert md to html * add email settings to live settings * use task to send email, handle some exceptions to create logs * remove ERROR_NOTIFICATION_MAIL_DELIVERY_FAILED usage * add email limit logic * fix tests * add docs * remove old email templates * remove old email templates * add template_fields to messaging backend * add messaging backends templates to public api * add comment for deprecated fields * fix test * fix tests * disable email by default * don't retry on SMTPException and TimeoutError * add tests * bring email back to public api docs * return ERROR_NOTIFICATION_MAIL_LIMIT_EXCEEDED * make template_fields tuple * build_subject_and_title -> build_subject_and_message * add one more comment about template deprecation * use 8 as backend id * add comment about gaierror and BadHeaderError * add comment on importing in notify_user_async * edit oss docs
572 lines
22 KiB
Python
572 lines
22 KiB
Python
import pytest
|
|
from django.urls import reverse
|
|
from rest_framework import status
|
|
from rest_framework.test import APIClient
|
|
|
|
from apps.base.tests.messaging_backend import TestOnlyBackend
|
|
|
|
TEST_MESSAGING_BACKEND_FIELD = TestOnlyBackend.backend_id.lower()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_get_list_integrations(
|
|
make_organization_and_user_with_token,
|
|
make_alert_receive_channel,
|
|
make_channel_filter,
|
|
make_integration_heartbeat,
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
expected_response = {
|
|
"count": 1,
|
|
"next": None,
|
|
"previous": None,
|
|
"results": [
|
|
{
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
],
|
|
}
|
|
url = reverse("api-public:integrations-list")
|
|
response = client.get(url, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.json() == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_integration(
|
|
make_organization_and_user_with_token,
|
|
make_escalation_chain,
|
|
):
|
|
organization, _, token = make_organization_and_user_with_token()
|
|
make_escalation_chain(organization)
|
|
|
|
client = APIClient()
|
|
data_for_create = {
|
|
"type": "grafana",
|
|
"name": "grafana_created",
|
|
"team_id": None,
|
|
}
|
|
url = reverse("api-public:integrations-list")
|
|
response = client.post(url, data=data_for_create, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_201_CREATED
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_integrations_with_none_templates(
|
|
make_organization_and_user_with_token,
|
|
make_escalation_chain,
|
|
):
|
|
organization, _, token = make_organization_and_user_with_token()
|
|
make_escalation_chain(organization)
|
|
|
|
client = APIClient()
|
|
data_for_create = {
|
|
"type": "grafana",
|
|
"team_id": None,
|
|
"name": "grafana_created",
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": None,
|
|
"web": None,
|
|
"sms": None,
|
|
"phone_call": None,
|
|
"telegram": None,
|
|
},
|
|
}
|
|
|
|
url = reverse("api-public:integrations-list")
|
|
|
|
response = client.post(url, data=data_for_create, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_201_CREATED
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_integration_with_invalid_type(
|
|
make_organization_and_user_with_token,
|
|
):
|
|
_, _, token = make_organization_and_user_with_token()
|
|
|
|
client = APIClient()
|
|
data_for_create = {
|
|
"type": "this_is_invalid_integration_type",
|
|
"name": "grafana_created",
|
|
"team_id": None,
|
|
}
|
|
url = reverse("api-public:integrations-list")
|
|
response = client.post(url, data=data_for_create, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_integration_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"grouping_key": "ip_addr", "slack": {"title": "Incident"}}}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": "ip_addr",
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": "Incident", "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_integration_template_messaging_backend(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"grouping_key": "ip_addr", TEST_MESSAGING_BACKEND_FIELD: {"title": "Incident"}}}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": "ip_addr",
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": "Incident",
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_invalid_integration_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"slack": {"title": "{% invalid jinja template }}"}}}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_resolve_signal_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"resolve_signal": "resig"}}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": "resig",
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_invalid_resolve_signal_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"resolve_signal": "{% invalid jinja template }}"}}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_empty_grouping_key_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"grouping_key": {}}}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_invalid_flat_web_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"web": "invalid_web_template"}}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_400_BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_sms_template_with_empty_dict(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"sms": {}}}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_update_integration_name(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"name": "grafana_updated"}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana_updated",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_set_default_template(
|
|
make_organization_and_user_with_token, make_alert_receive_channel, make_channel_filter, make_integration_heartbeat
|
|
):
|
|
organization, user, token = make_organization_and_user_with_token()
|
|
integration = make_alert_receive_channel(organization, verbal_name="grafana")
|
|
integration.slack_title_template = "updated_template"
|
|
integration.grouping_id_template = "updated_template"
|
|
integration.save()
|
|
default_channel_filter = make_channel_filter(integration, is_default=True)
|
|
make_integration_heartbeat(integration)
|
|
|
|
client = APIClient()
|
|
data_for_update = {"templates": {"grouping_key": None, "slack": {"title": None}}}
|
|
expected_response = {
|
|
"id": integration.public_primary_key,
|
|
"team_id": None,
|
|
"name": "grafana",
|
|
"link": integration.integration_url,
|
|
"type": "grafana",
|
|
"default_route": {
|
|
"escalation_chain_id": None,
|
|
"id": default_channel_filter.public_primary_key,
|
|
"slack": {"channel_id": None, "enabled": True},
|
|
"telegram": {"id": None, "enabled": False},
|
|
TEST_MESSAGING_BACKEND_FIELD: {"id": None, "enabled": False},
|
|
},
|
|
"heartbeat": {
|
|
"link": f"{integration.integration_url}heartbeat/",
|
|
},
|
|
"templates": {
|
|
"grouping_key": None,
|
|
"resolve_signal": None,
|
|
"acknowledge_signal": None,
|
|
"slack": {"title": None, "message": None, "image_url": None},
|
|
"web": {"title": None, "message": None, "image_url": None},
|
|
"sms": {
|
|
"title": None,
|
|
},
|
|
"phone_call": {
|
|
"title": None,
|
|
},
|
|
"telegram": {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
TEST_MESSAGING_BACKEND_FIELD: {
|
|
"title": None,
|
|
"message": None,
|
|
"image_url": None,
|
|
},
|
|
},
|
|
"maintenance_mode": None,
|
|
"maintenance_started_at": None,
|
|
"maintenance_end_at": None,
|
|
}
|
|
url = reverse("api-public:integrations-detail", args=[integration.public_primary_key])
|
|
response = client.put(url, data=data_for_update, format="json", HTTP_AUTHORIZATION=f"{token}")
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert response.data == expected_response
|