# -*- coding: utf-8 -*-

from zerver.lib.test_classes import WebhookTestCase

class PagerDutyHookTests(WebhookTestCase):
    STREAM_NAME = 'pagerduty'
    URL_TEMPLATE = u"/api/v1/external/pagerduty?api_key={api_key}&stream={stream}"
    FIXTURE_DIR_NAME = 'pagerduty'

    def test_trigger(self) -> None:
        expected_message = 'Incident [3](https://zulip-test.pagerduty.com/incidents/P140S4Y) triggered by [Test service](https://zulip-test.pagerduty.com/services/PIL5CUQ) (assigned to [armooo](https://zulip-test.pagerduty.com/users/POBCFRJ))\n\n``` quote\nfoo\n```'
        self.send_and_test_stream_message('trigger', u"Incident 3", expected_message)

    def test_trigger_v2(self) -> None:
        expected_message = 'Incident [33](https://webdemo.pagerduty.com/incidents/PRORDTY) triggered by [Production XDB Cluster](https://webdemo.pagerduty.com/services/PN49J75) (assigned to [Laura Haley](https://webdemo.pagerduty.com/users/P553OPV))\n\n``` quote\nMy new incident\n```'
        self.send_and_test_stream_message('trigger_v2', u'Incident 33', expected_message)

    def test_trigger_without_assignee_v2(self) -> None:
        expected_message = 'Incident [33](https://webdemo.pagerduty.com/incidents/PRORDTY) triggered by [Production XDB Cluster](https://webdemo.pagerduty.com/services/PN49J75) (assigned to nobody)\n\n``` quote\nMy new incident\n```'
        self.send_and_test_stream_message('trigger_without_assignee_v2', u'Incident 33', expected_message)

    def test_unacknowledge(self) -> None:
        expected_message = 'Incident [3](https://zulip-test.pagerduty.com/incidents/P140S4Y) unacknowledged by [Test service](https://zulip-test.pagerduty.com/services/PIL5CUQ) (assigned to [armooo](https://zulip-test.pagerduty.com/users/POBCFRJ))\n\n``` quote\nfoo\n```'
        self.send_and_test_stream_message('unacknowledge', u"Incident 3", expected_message)

    def test_resolved(self) -> None:
        expected_message = 'Incident [1](https://zulip-test.pagerduty.com/incidents/PO1XIJ5) resolved by [armooo](https://zulip-test.pagerduty.com/users/POBCFRJ)\n\n``` quote\nIt is on fire\n```'
        self.send_and_test_stream_message('resolved', u"Incident 1", expected_message)

    def test_resolved_v2(self) -> None:
        expected_message = 'Incident [33](https://webdemo.pagerduty.com/incidents/PRORDTY) resolved by [Laura Haley](https://webdemo.pagerduty.com/users/P553OPV)\n\n``` quote\nMy new incident\n```'
        self.send_and_test_stream_message('resolve_v2', 'Incident 33', expected_message)

    def test_auto_resolved(self) -> None:
        expected_message = 'Incident [2](https://zulip-test.pagerduty.com/incidents/PX7K9J2) resolved\n\n``` quote\nnew\n```'
        self.send_and_test_stream_message('auto_resolved', u"Incident 2", expected_message)

    def test_acknowledge(self) -> None:
        expected_message = 'Incident [1](https://zulip-test.pagerduty.com/incidents/PO1XIJ5) acknowledged by [armooo](https://zulip-test.pagerduty.com/users/POBCFRJ)\n\n``` quote\nIt is on fire\n```'
        self.send_and_test_stream_message('acknowledge', u"Incident 1", expected_message)

    def test_acknowledge_v2(self) -> None:
        expected_message = 'Incident [33](https://webdemo.pagerduty.com/incidents/PRORDTY) acknowledged by [Laura Haley](https://webdemo.pagerduty.com/users/P553OPV)\n\n``` quote\nMy new incident\n```'
        self.send_and_test_stream_message('acknowledge_v2', 'Incident 33', expected_message)

    def test_incident_assigned_v2(self) -> None:
        expected_message = 'Incident [33](https://webdemo.pagerduty.com/incidents/PRORDTY) assigned to [Wiley Jacobson](https://webdemo.pagerduty.com/users/PFBSJ2Z)\n\n``` quote\nMy new incident\n```'
        self.send_and_test_stream_message('assign_v2', 'Incident 33', expected_message)

    def test_no_subject(self) -> None:
        expected_message = u'Incident [48219](https://dropbox.pagerduty.com/incidents/PJKGZF9) resolved\n\n``` quote\nmp_error_block_down_critical\u2119\u01b4\n```'
        self.send_and_test_stream_message('mp_fail', u"Incident 48219", expected_message)

    def test_bad_message(self) -> None:
        expected_message = 'Unknown pagerduty message\n```\n{\n  "type":"incident.triggered"\n}\n```'
        self.send_and_test_stream_message('bad_message_type', u"pagerduty", expected_message)

    def test_unknown_message_type(self) -> None:
        expected_message = 'Unknown pagerduty message\n```\n{\n  "type":"foo"\n}\n```'
        self.send_and_test_stream_message('unknown_message_type', u"pagerduty", expected_message)