← Back to branch summary

~ubuntu-branches/ubuntu/maverick/python-django/maverick

« back to all changes in this revision

Viewing changes to django/contrib/auth/tests/models.py

  • Committer: Bazaar Package Importer
  • Author(s): Chris Lamb
  • Date: 2010-05-21 07:52:55 UTC
  • mfrom: (1.1.10 upstream) (4.4.7 sid)
  • Revision ID: james.westby@ubuntu.com-20100521075255-i1zpeyc0k8512pd7
Tags: 1.2-1
New upstream stable release.

Show diffs side-by-side

added added

removed removed

Lines of Context:
django/contrib/auth/tests/models.py
 
1
from django.conf import settings
 
2
from django.test import TestCase
 
3
from django.contrib.auth.models import User, SiteProfileNotAvailable
 
4
 
 
5
class ProfileTestCase(TestCase):
 
6
    fixtures = ['authtestdata.json']
 
7
    def setUp(self):
 
8
        """Backs up the AUTH_PROFILE_MODULE"""
 
9
        self.old_AUTH_PROFILE_MODULE = getattr(settings,
 
10
                                               'AUTH_PROFILE_MODULE', None)
 
11
 
 
12
    def tearDown(self):
 
13
        """Restores the AUTH_PROFILE_MODULE -- if it was not set it is deleted,
 
14
        otherwise the old value is restored"""
 
15
        if self.old_AUTH_PROFILE_MODULE is None and \
 
16
                hasattr(settings, 'AUTH_PROFILE_MODULE'):
 
17
            del settings.AUTH_PROFILE_MODULE
 
18
 
 
19
        if self.old_AUTH_PROFILE_MODULE is not None:
 
20
            settings.AUTH_PROFILE_MODULE = self.old_AUTH_PROFILE_MODULE
 
21
 
 
22
    def test_site_profile_not_available(self):
 
23
        # calling get_profile without AUTH_PROFILE_MODULE set
 
24
        if hasattr(settings, 'AUTH_PROFILE_MODULE'):
 
25
            del settings.AUTH_PROFILE_MODULE
 
26
        user = User.objects.get(username='testclient')
 
27
        self.assertRaises(SiteProfileNotAvailable, user.get_profile)
 
28
 
 
29
        # Bad syntax in AUTH_PROFILE_MODULE: 
 
30
        settings.AUTH_PROFILE_MODULE = 'foobar'
 
31
        self.assertRaises(SiteProfileNotAvailable, user.get_profile)
 
32
 
 
33
        # module that doesn't exist
 
34
        settings.AUTH_PROFILE_MODULE = 'foo.bar'
 
35
        self.assertRaises(SiteProfileNotAvailable, user.get_profile)