Ensure the mutliple_database tests leave the settings in the same state they found them.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@14313 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Alex Gaynor 2010-10-21 17:37:19 +00:00
parent 9d5d757bda
commit 085e4c9112
1 changed files with 6 additions and 2 deletions

View File

@ -1548,13 +1548,17 @@ class AuthTestCase(TestCase):
command_output = new_io.getvalue().strip() command_output = new_io.getvalue().strip()
self.assertTrue('"email": "alice@example.com",' in command_output) self.assertTrue('"email": "alice@example.com",' in command_output)
_missing = object()
class UserProfileTestCase(TestCase): class UserProfileTestCase(TestCase):
def setUp(self): def setUp(self):
self.old_auth_profile_module = getattr(settings, 'AUTH_PROFILE_MODULE', None) self.old_auth_profile_module = getattr(settings, 'AUTH_PROFILE_MODULE', _missing)
settings.AUTH_PROFILE_MODULE = 'multiple_database.UserProfile' settings.AUTH_PROFILE_MODULE = 'multiple_database.UserProfile'
def tearDown(self): def tearDown(self):
settings.AUTH_PROFILE_MODULE = self.old_auth_profile_module if self.old_auth_profile_module is _missing:
del settings.AUTH_PROFILE_MODULE
else:
settings.AUTH_PROFILE_MODULE = self.old_auth_profile_module
def test_user_profiles(self): def test_user_profiles(self):