oncall-engine/engine/apps/alerts/tests/test_custom_button.py
Michael Derynck 6b40f95033 World, meet OnCall!
Co-authored-by: Eve832 <eve.meelan@grafana.com>
    Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
    Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
    Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
    Co-authored-by: Julia <ferril.darkdiver@gmail.com>
    Co-authored-by: maskin25 <kengurek@gmail.com>
    Co-authored-by: Matias Bordese <mbordese@gmail.com>
    Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
    Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
    Co-authored-by: Richard Hartmann <richih@richih.org>
    Co-authored-by: Robby Milo <robbymilo@fastmail.com>
    Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
    Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
    Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
2022-06-03 08:09:47 -06:00

63 lines
1.7 KiB
Python

import pytest
@pytest.mark.django_db
def test_escaping_payload_with_double_quotes(
make_organization,
make_custom_action,
make_alert_receive_channel,
make_alert_group,
make_alert,
):
organization = make_organization()
alert_receive_channel = make_alert_receive_channel(organization)
alert_group = make_alert_group(alert_receive_channel)
alert_payload = {
"text": '"Hello world"',
}
alert = make_alert(alert_group=alert_group, raw_request_data=alert_payload)
custom_button = make_custom_action(
name="github_button",
webhook="https://github.com/",
user="Chris Vanstras",
password="qwerty",
data='{\n "text" : "{{ alert_payload.text }}"\n}',
authorization_header="auth_token",
organization=organization,
)
custom_button.build_post_kwargs(alert)
@pytest.mark.django_db
def test_escaping_payload_with_single_quote_in_string(
make_organization,
make_custom_action,
make_alert_receive_channel,
make_alert_group,
make_alert,
):
organization = make_organization()
alert_receive_channel = make_alert_receive_channel(organization)
alert_group = make_alert_group(alert_receive_channel)
alert_payload = {
"text": "Hi, it's alert",
}
alert = make_alert(alert_group=alert_group, raw_request_data=alert_payload)
custom_button = make_custom_action(
name="github_button",
webhook="https://github.com/",
user="Chris Vanstras",
password="qwerty",
data='{"data" : "{{ alert_payload }}"}',
authorization_header="auth_token",
organization=organization,
)
custom_button.build_post_kwargs(alert)