healthchecks/hc/accounts/tests/test_signup.py

from django.contrib.auth.models import User
from django.core import mail
from django.test import TestCase
from django.test.utils import override_settings
from hc.api.models import Check
from django.conf import settings


class SignupTestCase(TestCase):

    def test_it_sends_link(self):
        form = {"identity": "[email protected]"}

        r = self.client.post("/accounts/signup/", form)
        self.assertContains(r, "Account created")

        # An user should have been created
        self.assertEqual(User.objects.count(), 1)

        # And email sent
        self.assertEqual(len(mail.outbox), 1)
        subject = "Log in to %s" % settings.SITE_NAME
        self.assertEqual(mail.outbox[0].subject, subject)

        # And check should be associated with the new user
        check = Check.objects.get()
        self.assertEqual(check.name, "My First Check")

    @override_settings(REGISTRATION_OPEN=False)
    def test_it_obeys_registration_open(self):
        form = {"identity": "[email protected]"}

        r = self.client.post("/accounts/signup/", form)
        self.assertEqual(r.status_code, 403)

    def test_it_ignores_case(self):
        form = {"identity": "[email protected]"}
        self.client.post("/accounts/signup/", form)

        # There should be exactly one user:
        q = User.objects.filter(email="[email protected]")
        self.assertTrue(q.exists)

    def test_it_checks_for_existing_users(self):
        alice = User(username="alice", email="[email protected]")
        alice.save()

        form = {"identity": "[email protected]"}
        r = self.client.post("/accounts/signup/", form)
        self.assertContains(r, "already exists")

    def test_it_checks_syntax(self):
        form = {"identity": "alice at example org"}
        r = self.client.post("/accounts/signup/", form)
        self.assertContains(r, "Enter a valid email address")