def test_add_user_and_user_setup(self): """testing add a user to the vault and setup passphrase""" def givepass(): return 'passphrase' ures1 = self.vault.user_add('testuser', False) self.assertTrue(ures1 is not None) tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) ures2 = tmp_vault.user_setup('testuser', 'http://localhost:6555/vault/rpc', 'passphrase') self.assertTrue(ures2 is not None)
def test_group_del_user(self): """testing delete a user from a group from the vault""" #Add test user ures1 = self.vault.user_add('testuser2') self.assertTrue(ures1 is not None) tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) ures2 = tmp_vault.user_setup('testuser2', 'http://localhost:6555/vault/rpc', 'passphrase') self.assertTrue(ures2 is not None) gcres = self.vault.group_add(u'Test group del user') uares = self.vault.group_add_user(gcres['group_id'], 'testuser2') udres = self.vault.group_del_user(gcres['group_id'], 'testuser2') self.assertTrue("Removed user from group successfully" in udres['message'])
def test_user_setup_already_has_public_key(self): """ A user cannot do his setup twice """ ures1 = self.vault.user_add('testuser', False) self.assertTrue(ures1 is not None) tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) ures2 = tmp_vault.user_setup('testuser', 'http://localhost:6555/vault/rpc', 'passphrase') self.assertTrue(ures2 is not None) tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) self.assertRaises(VaultError, tmp_vault.user_setup, 'testuser', 'http://localhost:6555/vault/rpc', 'passphrase')
def test_user_setup_no_user(self): tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) self.assertRaises(VaultError, tmp_vault.user_setup, 'invalid user', 'http://localhost:6555/vault/rpc', 'passphrase')
def test_user_setup_expired(self): import time tmp_vault = SFLvaultClient(self.getConfFileUser(), shell=True) self.vault.user_add('testuser', False) time.sleep(5) self.assertRaises(VaultError, tmp_vault.user_setup, 'testuser', 'http://localhost:6555/vault/rpc', 'passphrase')
def test_group_add_user(self): """testing add a user to a group to the vault""" ures1 = self.vault.user_add('test_add_user_group_vault') self.assertTrue(ures1 is not None) tmp_vault = SFLvaultClient(self.getConfFileUser()) ures2 = tmp_vault.user_setup('test_add_user_group_vault', 'http://localhost:6555/vault/rpc', 'passphrase') self.assertTrue(ures2 is not None) log.warn(ures1) log.warn(ures2) gres1 = self.vault.group_add("test_group1_user") gres2 = self.vault.group_add("test_group2_user") gares1 = self.vault.group_add_user(gres1['group_id'], ures1['user_id']) gares2 = self.vault.group_add_user(gres2['group_id'], ures1['user_id'], True) self.assertTrue("Added user to group successfully" in gares1['message']) self.assertFalse("Error adding user to group" in gares2['message'])
def __init__(self, config=None, vault=None, parser=None): """Create a SFLvaultCommand object :param config: config filename to use, required if no vault specified :param vault: an existing SFLvaultClient object, otherwise it will be created, using specified config :param parser: an option parser, otherwise it will be created (recommended) """ self.parser = (parser or optparse.OptionParser(usage=optparse.SUPPRESS_USAGE)) if not config and not vault: raise ValueError("`config` required if `vault` not specified") # Use the specified, or create a new one. self.vault = (vault or SFLvaultClient(config))
def __init__(self, config, vault=None): self.config = config self.vault = (vault or SFLvaultClient(config, shell=True))