def _create_roles_and_users(self):
# Create master role
if not DatabaseRole.exists(MASTER_ROLE_NAME):
self.master_role = DatabaseRole(MASTER_ROLE_NAME, create=True)
else:
self.master_role = DatabaseRole(MASTER_ROLE_NAME, create=False)
# Create standalone role (test1)
if not DatabaseRole.exists(EMPTY_ROLE_NAME):
self.empty_role = DatabaseRole(EMPTY_ROLE_NAME, create=True)
else:
self.empty_role = DatabaseRole(EMPTY_ROLE_NAME, create=False)
# Create users based on master tole (test2 and test3)
if not DatabaseUser.exists(TEST_USER_NAME1):
self.test1 = DatabaseUser(TEST_USER_NAME1, 'testpass', create=True)
else:
self.test1 = DatabaseUser(TEST_USER_NAME1, 'testpass', create=False)
self.master_role.grant(self.test1)
if not DatabaseUser.exists(TEST_USER_NAME2):
self.test2 = DatabaseUser(TEST_USER_NAME2, create=True)
else:
self.test2 = DatabaseUser(TEST_USER_NAME2, create=False)
self.master_role.grant(self.test2)
def setUp(self):
test_settings = {
'DATABASE_USER': '******',
'DATABASE_PASSWORD': '',
}
if hasattr(settings, 'DATABASES'):
test_settings['DATABASES'] = settings.DATABASES.copy()
test_settings['DATABASES']['default'].update({
'USER': '******',
'PASSWORD': '',
})
self.old_settings = switch_settings(**test_settings)
self._refresh_connection()
self._capture_stdout_stderr()
self._create_roles_and_users()
# Make sure that dummy role doesn't exist before testing starts.
if DatabaseRole.exists('dummy'):
DatabaseRole('dummy', create=False).delete()
def _delete_roles_and_users(self):
# clean up everything
if DatabaseUser.exists(TEST_USER_NAME1):
if DatabaseRole.exists(MASTER_ROLE_NAME):
self.master_role.revoke(self.test1)
self.test1.delete()
if DatabaseUser.exists(TEST_USER_NAME2):
if DatabaseRole.exists(MASTER_ROLE_NAME):
self.master_role.revoke(self.test2)
self.test2.delete()
if DatabaseRole.exists(MASTER_ROLE_NAME):
self.master_role.delete()
if DatabaseRole.exists(EMPTY_ROLE_NAME):
self.empty_role.delete()
# make sure everything has been cleaned up
self.assertFalse(DatabaseUser.exists(TEST_USER_NAME1))
self.assertFalse(DatabaseUser.exists(TEST_USER_NAME2))
self.assertFalse(DatabaseRole.exists(MASTER_ROLE_NAME))
self.assertFalse(DatabaseRole.exists(EMPTY_ROLE_NAME))