healthchecks/hc/api/tests/test_admin.py

from hc.api.models import Channel, Check
from hc.test import BaseTestCase


class ApiAdminTestCase(BaseTestCase):

    def setUp(self):
        super(ApiAdminTestCase, self).setUp()
        self.check = Check.objects.create(user=self.alice, tags="foo bar")

        self.alice.is_staff = True
        self.alice.is_superuser = True
        self.alice.save()

    def test_it_shows_channel_list_with_pushbullet(self):
        self.client.login(username="[email protected]", password="password")

        Channel.objects.create(user=self.alice, kind="pushbullet",
                               value="test-token")

        r = self.client.get("/admin/api/channel/")
        self.assertContains(r, "Pushbullet")

    def test_it_shows_channel_list_with_unverified_email(self):
        self.client.login(username="[email protected]", password="password")

        Channel.objects.create(user=self.alice, kind="email",
                               value="[email protected]")

        r = self.client.get("/admin/api/channel/")
        self.assertContains(r, "Email <i>(unverified)</i>")