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))