upstream-zulip/zerver/webhooks/librato/tests.py

from urllib.parse import urlencode

from typing_extensions import override

from zerver.lib.test_classes import WebhookTestCase


class LibratoHookTests(WebhookTestCase):
    IS_ATTACHMENT = False

    @override
    def get_body(self, fixture_name: str) -> str:
        if self.IS_ATTACHMENT:
            return self.webhook_fixture_data("librato", fixture_name, file_type="json")
        return urlencode(
            {"payload": self.webhook_fixture_data("librato", fixture_name, file_type="json")}
        )

    def test_alert_message_with_default_topic(self) -> None:
        expected_topic_name = "Alert alert.name"
        expected_message = "Alert [alert_name](https://metrics.librato.com/alerts#/6294535) has triggered! [Reaction steps](http://www.google.pl):\n * Metric `librato.cpu.percent.idle`, sum was below 44 by 300s, recorded at <time:2016-03-31T09:11:42+00:00>.\n * Metric `librato.swap.swap.cached`, average was absent  by 300s, recorded at <time:2016-03-31T09:11:42+00:00>.\n * Metric `librato.swap.swap.cached`, derivative was above 9 by 300s, recorded at <time:2016-03-31T09:11:42+00:00>."
        self.check_webhook(
            "alert",
            expected_topic_name,
            expected_message,
            content_type="application/x-www-form-urlencoded",
        )

    def test_alert_message_with_custom_topic(self) -> None:
        custom_topic_name = "custom_name"
        self.url = self.build_webhook_url(topic=custom_topic_name)
        expected_message = "Alert [alert_name](https://metrics.librato.com/alerts#/6294535) has triggered! [Reaction steps](http://www.google.pl):\n * Metric `librato.cpu.percent.idle`, sum was below 44 by 300s, recorded at <time:2016-03-31T09:11:42+00:00>.\n * Metric `librato.swap.swap.cached`, average was absent  by 300s, recorded at <time:2016-03-31T09:11:42+00:00>.\n * Metric `librato.swap.swap.cached`, derivative was above 9 by 300s, recorded at <time:2016-03-31T09:11:42+00:00>."
        self.check_webhook(
            "alert",
            custom_topic_name,
            expected_message,
            content_type="application/x-www-form-urlencoded",
        )

    def test_three_conditions_alert_message(self) -> None:
        expected_message = "Alert [alert_name](https://metrics.librato.com/alerts#/6294535) has triggered! [Reaction steps](http://www.use.water.pl):\n * Metric `collectd.interface.eth0.if_octets.tx`, absolute_value was above 4 by 300s, recorded at <time:2016-04-11T20:40:14+00:00>.\n * Metric `collectd.load.load.longterm`, max was above 99, recorded at <time:2016-04-11T20:40:14+00:00>.\n * Metric `librato.swap.swap.cached`, average was absent  by 60s, recorded at <time:2016-04-11T20:40:14+00:00>."
        expected_topic_name = "Alert TooHighTemperature"
        self.check_webhook(
            "three_conditions_alert",
            expected_topic_name,
            expected_message,
            content_type="application/x-www-form-urlencoded",
        )

    def test_alert_clear(self) -> None:
        expected_topic_name = "Alert Alert_name"
        expected_message = "Alert [alert_name](https://metrics.librato.com/alerts#/6309313) has cleared at <time:2016-04-12T13:11:44+00:00>!"
        self.check_webhook(
            "alert_cleared",
            expected_topic_name,
            expected_message,
            content_type="application/x-www-form-urlencoded",
        )

    def test_snapshot(self) -> None:
        self.IS_ATTACHMENT = True
        expected_topic_name = "Snapshots"
        expected_message = "**Hamlet** sent a [snapshot](http://snapshots.librato.com/chart/nr5l3n0c-82162.png) of [metric](https://metrics.librato.com/s/spaces/167315/explore/1731491?duration=72039&end_time=1460569409)."
        self.check_webhook(
            "snapshot",
            expected_topic_name,
            expected_message,
            content_type="application/x-www-form-urlencoded",
        )
        self.IS_ATTACHMENT = False

    def test_bad_request(self) -> None:
        with self.assertRaises(AssertionError) as e:
            self.check_webhook(
                "bad",
                "",
                "",
                content_type="application/json",
            )
        self.assertIn("Malformed JSON input", e.exception.args[0])

    def test_bad_msg_type(self) -> None:
        with self.assertRaises(AssertionError) as e:
            self.check_webhook(
                "bad_msg_type",
                "",
                "",
                content_type="application/x-www-form-urlencoded",
            )
        self.assertIn("Unexpected message type", e.exception.args[0])