healthchecks/hc/front/tests/test_add_pagerduty_complete.py

import json

from django.test.utils import override_settings
from hc.api.models import Channel
from hc.test import BaseTestCase
from urllib.parse import urlencode


@override_settings(PD_APP_ID="FOOBAR")
class AddPagerDutyCompleteTestCase(BaseTestCase):
    def setUp(self):
        super().setUp()

        session = self.client.session
        session["pagerduty"] = ("ABC", str(self.project.code))
        session.save()

    def _url(self, state="ABC"):
        config = {
            "account": {"name": "Foo"},
            "integration_keys": [{"integration_key": "foo", "name": "bar"}],
        }

        url = "/integrations/add_pagerduty/?"
        url += urlencode({"state": state, "config": json.dumps(config)})
        return url

    def test_it_adds_channel(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self._url())
        self.assertRedirects(r, self.channels_url)

        channel = Channel.objects.get()
        self.assertEqual(channel.kind, "pd")
        self.assertEqual(channel.pd_service_key, "foo")
        self.assertEqual(channel.pd_account, "Foo")

    def test_it_validates_state(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self._url(state="XYZ"))
        self.assertEqual(r.status_code, 403)

    @override_settings(PD_APP_ID=None)
    def test_it_requires_app_id(self):
        self.client.login(username="[email protected]", password="password")

        r = self.client.get(self._url())
        self.assertEqual(r.status_code, 404)

    @override_settings(PD_ENABLED=False)
    def test_it_requires_pd_enabled(self):
        self.client.login(username="[email protected]", password="password")

        r = self.client.get(self._url())
        self.assertEqual(r.status_code, 404)

    def test_it_requires_rw_access(self):
        self.bobs_membership.rw = False
        self.bobs_membership.save()

        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self._url())
        self.assertEqual(r.status_code, 403)