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