zulip/zerver/webhooks/librato/tests.py

# -*- coding: utf-8 -*-
from six.moves import urllib
from typing import Optional, Text
from zerver.lib.test_classes import WebhookTestCase

class LibratoHookTests(WebhookTestCase):
    STREAM_NAME = 'librato'
    URL_TEMPLATE = u"/api/v1/external/librato?api_key={api_key}&stream=librato"
    FIXTURE_DIR_NAME = 'librato'
    IS_ATTACHMENT = False

    def get_body(self, fixture_name):
        # type: (Text) -> Text
        if self.IS_ATTACHMENT:
            return self.fixture_data("librato", fixture_name, file_type='json')
        return urllib.parse.urlencode({'payload': self.fixture_data("librato", fixture_name, file_type='json')})

    def build_webhook_url(self, topic=None):
        # type: (Optional[Text]) -> Text
        api_key = self.get_api_key(self.TEST_USER_EMAIL)
        url = self.URL_TEMPLATE.format(stream=self.STREAM_NAME, api_key=api_key)
        if topic:
            url = u"{}&topic={}".format(url, topic)
        return url

    def test_alert_message_with_default_topic(self):
        # type: () -> None
        expected_subject = '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 2016-03-31 05:11:42\n>Metric `librato.swap.swap.cached`, average was absent  by 300s, recorded at 2016-03-31 05:11:42\n>Metric `librato.swap.swap.cached`, derivative was above 9 by 300s, recorded at 2016-03-31 05:11:42"
        self.send_and_test_stream_message('alert', expected_subject, expected_message, content_type="application/x-www-form-urlencoded")

    def test_alert_message_with_custom_topic(self):
        # type: () -> None
        custom_topic = 'custom_name'
        self.url = self.build_webhook_url(custom_topic)
        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 2016-03-31 05:11:42\n>Metric `librato.swap.swap.cached`, average was absent  by 300s, recorded at 2016-03-31 05:11:42\n>Metric `librato.swap.swap.cached`, derivative was above 9 by 300s, recorded at 2016-03-31 05:11:42"
        self.send_and_test_stream_message('alert', custom_topic, expected_message, content_type="application/x-www-form-urlencoded")

    def test_three_conditions_alert_message(self):
        # type: () -> 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 2016-04-11 16:40:14\n>Metric `collectd.load.load.longterm`, max was above 99, recorded at 2016-04-11 16:40:14\n>Metric `librato.swap.swap.cached`, average was absent  by 60s, recorded at 2016-04-11 16:40:14"
        expected_subject = 'Alert ToHighTemeprature'
        self.send_and_test_stream_message('three_conditions_alert', expected_subject, expected_message, content_type="application/x-www-form-urlencoded")

    def test_alert_clear(self):
        # type: () -> None
        expected_subject = 'Alert Alert_name'
        expected_message = "Alert [alert_name](https://metrics.librato.com/alerts#/6309313) has cleared at 2016-04-12 09:11:44!"
        self.send_and_test_stream_message('alert_cleared', expected_subject, expected_message, content_type="application/x-www-form-urlencoded")

    def test_snapshot(self):
        # type: () -> None
        self.IS_ATTACHMENT = True
        expected_subject = '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.send_and_test_stream_message('snapshot', expected_subject, expected_message, content_type="application/x-www-form-urlencoded")
        self.IS_ATTACHMENT = False