36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
|
from django.conf import settings
|
||
|
|
from django.test import TestCase
|
||
|
|
from django.contrib.auth.models import User, SiteProfileNotAvailable
|
||
|
|
|
||
|
|
class ProfileTestCase(TestCase):
|
||
|
|
fixtures = ['authtestdata.json']
|
||
|
|
def setUp(self):
|
||
|
|
"""Backs up the AUTH_PROFILE_MODULE"""
|
||
|
|
self.old_AUTH_PROFILE_MODULE = getattr(settings,
|
||
|
|
'AUTH_PROFILE_MODULE', None)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
"""Restores the AUTH_PROFILE_MODULE -- if it was not set it is deleted,
|
||
|
|
otherwise the old value is restored"""
|
||
|
|
if self.old_AUTH_PROFILE_MODULE is None and \
|
||
|
|
hasattr(settings, 'AUTH_PROFILE_MODULE'):
|
||
|
|
del settings.AUTH_PROFILE_MODULE
|
||
|
|
|
||
|
|
if self.old_AUTH_PROFILE_MODULE is not None:
|
||
|
|
settings.AUTH_PROFILE_MODULE = self.old_AUTH_PROFILE_MODULE
|
||
|
|
|
||
|
|
def test_site_profile_not_available(self):
|
||
|
|
# calling get_profile without AUTH_PROFILE_MODULE set
|
||
|
|
if hasattr(settings, 'AUTH_PROFILE_MODULE'):
|
||
|
|
del settings.AUTH_PROFILE_MODULE
|
||
|
|
user = User.objects.get(username='testclient')
|
||
|
|
self.assertRaises(SiteProfileNotAvailable, user.get_profile)
|
||
|
|
|
||
|
|
# Bad syntax in AUTH_PROFILE_MODULE:
|
||
|
|
settings.AUTH_PROFILE_MODULE = 'foobar'
|
||
|
|
self.assertRaises(SiteProfileNotAvailable, user.get_profile)
|
||
|
|
|
||
|
|
# module that doesn't exist
|
||
|
|
settings.AUTH_PROFILE_MODULE = 'foo.bar'
|
||
|
|
self.assertRaises(SiteProfileNotAvailable, user.get_profile)
|