healthchecks/hc/front/tests/test_add_sms.py

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


@override_settings(TWILIO_ACCOUNT="foo", TWILIO_AUTH="foo", TWILIO_FROM="123")
class AddSmsTestCase(BaseTestCase):
    url = "/integrations/add_sms/"

    def test_instructions_work(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self.url)
        self.assertContains(r, "Get a SMS message")

    @override_settings(USE_PAYMENTS=True)
    def test_it_warns_about_limits(self):
        self.profile.sms_limit = 0
        self.profile.save()

        self.client.login(username="[email protected]", password="password")
        r = self.client.get(self.url)
        self.assertContains(r, "upgrade to a")

    def test_it_creates_channel(self):
        form = {"label": "My Phone", "value": "+1234567890"}

        self.client.login(username="[email protected]", password="password")
        r = self.client.post(self.url, form)
        self.assertRedirects(r, "/integrations/")

        c = Channel.objects.get()
        self.assertEqual(c.kind, "sms")
        self.assertEqual(c.sms_number, "+1234567890")
        self.assertEqual(c.sms_label, "My Phone")

    def test_it_rejects_bad_number(self):
        form = {"value": "not a phone number address"}

        self.client.login(username="[email protected]", password="password")
        r = self.client.post(self.url, form)
        self.assertContains(r, "Invalid phone number format.")

    def test_it_trims_whitespace(self):
        form = {"value": "   +1234567890   "}

        self.client.login(username="[email protected]", password="password")
        self.client.post(self.url, form)

        c = Channel.objects.get()
        self.assertEqual(c.sms_number, "+1234567890")

    @override_settings(TWILIO_AUTH=None)
    def test_it_requires_credentials(self):
        self.client.login(username="[email protected]", password="password")
        r = self.client.get("/integrations/add_sms/")
        self.assertEqual(r.status_code, 404)