git.ozlabs.org Git - patchwork/blob - apps/patchwork/tests/test_registration.py

   1 # Patchwork - automated patch tracking system
   2 # Copyright (C) 2010 Jeremy Kerr <jk@ozlabs.org>
   3 #
   4 # This file is part of the Patchwork package.
   5 #
   6 # Patchwork is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # Patchwork is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with Patchwork; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 import unittest
  21 from django.test import TestCase
  22 from django.test.client import Client
  23 from django.core import mail
  24 from django.core.urlresolvers import reverse
  25 from django.contrib.auth.models import User
  26 from patchwork.models import EmailConfirmation, Person
  27 from patchwork.tests.utils import create_user
  28
  29 def _confirmation_url(conf):
  30     return reverse('patchwork.views.confirm', kwargs = {'key': conf.key})
  31
  32 class TestUser(object):
  33     firstname = 'Test'
  34     lastname = 'User'
  35     username = 'testuser'
  36     email = 'test@example.com'
  37     password = 'foobar'
  38
  39 class RegistrationTest(TestCase):
  40     def setUp(self):
  41         self.user = TestUser()
  42         self.client = Client()
  43         self.default_data = {'username': self.user.username,
  44                              'first_name': self.user.firstname,
  45                              'last_name': self.user.lastname,
  46                              'email': self.user.email,
  47                              'password': self.user.password}
  48         self.required_error = 'This field is required.'
  49         self.invalid_error = 'Enter a valid value.'
  50
  51     def testRegistrationForm(self):
  52         response = self.client.get('/register/')
  53         self.assertEquals(response.status_code, 200)
  54         self.assertTemplateUsed(response, 'patchwork/registration_form.html')
  55
  56     def testBlankFields(self):
  57         for field in ['username', 'email', 'password']:
  58             data = self.default_data.copy()
  59             del data[field]
  60             response = self.client.post('/register/', data)
  61             self.assertEquals(response.status_code, 200)
  62             self.assertFormError(response, 'form', field, self.required_error)
  63
  64     def testInvalidUsername(self):
  65         data = self.default_data.copy()
  66         data['username'] = 'invalid user'
  67         response = self.client.post('/register/', data)
  68         self.assertEquals(response.status_code, 200)
  69         self.assertFormError(response, 'form', 'username', self.invalid_error)
  70
  71     def testExistingUsername(self):
  72         user = create_user()
  73         data = self.default_data.copy()
  74         data['username'] = user.username
  75         response = self.client.post('/register/', data)
  76         self.assertEquals(response.status_code, 200)
  77         self.assertFormError(response, 'form', 'username',
  78                 'This username is already taken. Please choose another.')
  79
  80     def testExistingEmail(self):
  81         user = create_user()
  82         data = self.default_data.copy()
  83         data['email'] = user.email
  84         response = self.client.post('/register/', data)
  85         self.assertEquals(response.status_code, 200)
  86         self.assertFormError(response, 'form', 'email',
  87                 'This email address is already in use ' + \
  88                 'for the account "%s".\n' % user.username)
  89
  90     def testValidRegistration(self):
  91         response = self.client.post('/register/', self.default_data)
  92         self.assertEquals(response.status_code, 200)
  93         self.assertContains(response, 'confirmation email has been sent')
  94
  95         # check for presence of an inactive user object
  96         users = User.objects.filter(username = self.user.username)
  97         self.assertEquals(users.count(), 1)
  98         user = users[0]
  99         self.assertEquals(user.username, self.user.username)
 100         self.assertEquals(user.email, self.user.email)
 101         self.assertEquals(user.is_active, False)
 102
 103         # check for confirmation object
 104         confs = EmailConfirmation.objects.filter(user = user,
 105                                                  type = 'registration')
 106         self.assertEquals(len(confs), 1)
 107         conf = confs[0]
 108         self.assertEquals(conf.email, self.user.email)
 109
 110         # check for a sent mail
 111         self.assertEquals(len(mail.outbox), 1)
 112         msg = mail.outbox[0]
 113         self.assertEquals(msg.subject, 'Patchwork account confirmation')
 114         self.assertTrue(self.user.email in msg.to)
 115         self.assertTrue(_confirmation_url(conf) in msg.body)
 116
 117         # ...and that the URL is valid
 118         response = self.client.get(_confirmation_url(conf))
 119         self.assertEquals(response.status_code, 200)
 120
 121 class RegistrationConfirmationTest(TestCase):
 122
 123     def setUp(self):
 124         self.user = TestUser()
 125         self.default_data = {'username': self.user.username,
 126                              'first_name': self.user.firstname,
 127                              'last_name': self.user.lastname,
 128                              'email': self.user.email,
 129                              'password': self.user.password}
 130
 131     def testRegistrationConfirmation(self):
 132         self.assertEqual(EmailConfirmation.objects.count(), 0)
 133         response = self.client.post('/register/', self.default_data)
 134         self.assertEquals(response.status_code, 200)
 135         self.assertContains(response, 'confirmation email has been sent')
 136
 137         self.assertEqual(EmailConfirmation.objects.count(), 1)
 138         conf = EmailConfirmation.objects.filter()[0]
 139         self.assertFalse(conf.user.is_active)
 140         self.assertTrue(conf.active)
 141
 142         response = self.client.get(_confirmation_url(conf))
 143         self.assertEquals(response.status_code, 200)
 144         self.assertTemplateUsed(response, 'patchwork/registration-confirm.html')
 145
 146         conf = EmailConfirmation.objects.get(pk = conf.pk)
 147         self.assertTrue(conf.user.is_active)
 148         self.assertFalse(conf.active)
 149
 150     def testRegistrationNewPersonSetup(self):
 151         """ Check that the person object created after registration has the
 152             correct details """
 153
 154         # register
 155         self.assertEqual(EmailConfirmation.objects.count(), 0)
 156         response = self.client.post('/register/', self.default_data)
 157         self.assertEquals(response.status_code, 200)
 158         self.assertFalse(Person.objects.exists())
 159
 160         # confirm
 161         conf = EmailConfirmation.objects.filter()[0]
 162         response = self.client.get(_confirmation_url(conf))
 163         self.assertEquals(response.status_code, 200)
 164
 165         qs = Person.objects.filter(email = self.user.email)
 166         self.assertTrue(qs.exists())
 167         person = Person.objects.get(email = self.user.email)
 168
 169         self.assertEquals(person.name,
 170                     self.user.firstname + ' ' + self.user.lastname)
 171
 172     def testRegistrationExistingPersonSetup(self):
 173         """ Check that the person object created after registration has the
 174             correct details """
 175
 176         fullname = self.user.firstname + ' '  + self.user.lastname
 177         person = Person(name = fullname, email = self.user.email)
 178         person.save()
 179
 180         # register
 181         self.assertEqual(EmailConfirmation.objects.count(), 0)
 182         response = self.client.post('/register/', self.default_data)
 183         self.assertEquals(response.status_code, 200)
 184
 185         # confirm
 186         conf = EmailConfirmation.objects.filter()[0]
 187         response = self.client.get(_confirmation_url(conf))
 188         self.assertEquals(response.status_code, 200)
 189
 190         person = Person.objects.get(email = self.user.email)
 191
 192         self.assertEquals(person.name, fullname)
 193
 194     def testRegistrationExistingPersonUnmodified(self):
 195         """ Check that an unconfirmed registration can't modify an existing
 196             Person object"""
 197
 198         fullname = self.user.firstname + ' '  + self.user.lastname
 199         person = Person(name = fullname, email = self.user.email)
 200         person.save()
 201
 202         # register
 203         data = self.default_data.copy()
 204         data['first_name'] = 'invalid'
 205         data['last_name'] = 'invalid'
 206         self.assertEquals(data['email'], person.email)
 207         response = self.client.post('/register/', data)
 208         self.assertEquals(response.status_code, 200)
 209
 210         self.assertEquals(Person.objects.get(pk = person.pk).name, fullname)