healthchecks/hc/accounts/tests/test_project_model.py

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


class ProjectModelTestCase(BaseTestCase):
    def test_num_checks_available_handles_multiple_projects(self):
        # One check in Alice's primary project:
        Check.objects.create(project=self.project)

        # One check in Alice's secondary project:
        p2 = Project.objects.create(owner=self.alice)
        Check.objects.create(project=p2)

        self.assertEqual(self.project.num_checks_available(), 18)

    def test_it_handles_zero_broken_channels(self):
        self.assertFalse(self.project.have_broken_channels())

    def test_it_handles_one_broken_channel(self):
        Channel.objects.create(kind="webhook", last_error="x", project=self.project)

        self.assertTrue(self.project.have_broken_channels())