healthchecks/hc/front/tests/test_add_matrix.py

from unittest.mock import patch

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


@override_settings(MATRIX_ACCESS_TOKEN="foo")
@override_settings(MATRIX_HOMESERVER="fake-homeserver")
class AddMatrixTestCase(BaseTestCase):
    def setUp(self):
        super().setUp()
        self.url = "/projects/%s/add_matrix/" % self.project.code

    @override_settings(MATRIX_ACCESS_TOKEN="foo")
    def test_instructions_work(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self.url)
        self.assertContains(r, "Integration Settings", status_code=200)

    @patch("hc.front.forms.requests.post")
    def test_it_works(self, mock_post):
        mock_post.return_value.status_code = 200
        mock_post.return_value.json.return_value = {"room_id": "fake-room-id"}

        form = {"alias": "!foo:example.org"}
        self.client.login(username="[email protected]", password="password")
        r = self.client.post(self.url, form)
        self.assertRedirects(r, self.channels_url)

        c = Channel.objects.get()
        self.assertEqual(c.kind, "matrix")
        self.assertEqual(c.value, "fake-room-id")
        self.assertEqual(c.project, self.project)

    @override_settings(MATRIX_ACCESS_TOKEN=None)
    def test_it_requires_access_token(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 404)

    @patch("hc.front.forms.requests.post")
    def test_it_handles_429(self, mock_post):
        mock_post.return_value.status_code = 429

        form = {"alias": "!foo:example.org"}
        self.client.login(username="[email protected]", password="password")
        r = self.client.post(self.url, form)

        self.assertContains(r, "Matrix server returned status code 429")
        self.assertFalse(Channel.objects.exists())

    @patch("hc.front.forms.requests.post")
    def test_it_handles_502(self, mock_post):
        mock_post.return_value.status_code = 502

        form = {"alias": "!foo:example.org"}
        self.client.login(username="[email protected]", password="password")
        r = self.client.post(self.url, form)

        self.assertContains(r, "Matrix server returned status code 502")
        self.assertFalse(Channel.objects.exists())

    def test_it_requires_rw_access(self):
        self.bobs_membership.role = "r"
        self.bobs_membership.save()

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