def execute(self, args):
mgr = SshKeyManager(self.client)
key_id = resolve_id(mgr.resolve_ids,
args.get('<identifier>'), 'SshKey')
if not mgr.edit_key(key_id,
label=args['--label'],
notes=args['--notes']):
raise CLIAbort('Failed to edit SSH key')
def execute(self, args):
mgr = SshKeyManager(self.client)
key_id = resolve_id(mgr.resolve_ids, args.get("<identifier>"), "SshKey")
data = {}
if args.get("--label"):
data["label"] = args["--label"]
if args.get("--notes"):
data["notes"] = args["--notes"]
if not mgr.edit_key(key_id, **data):
raise CLIAbort("Failed to edit SSH key")
def execute(self, args):
mgr = SshKeyManager(self.client)
key_id = resolve_id(mgr.resolve_ids,
args.get('<identifier>'), 'SshKey')
data = {}
if args.get('--label'):
data['label'] = args['--label']
if args.get('--notes'):
data['notes'] = args['--notes']
if not mgr.edit_key(key_id, **data):
raise CLIAbort('Failed to edit SSH key')
class SshKeyTests(unittest.TestCase):
def setUp(self):
self.client = FixtureClient()
self.sshkey = SshKeyManager(self.client)
def test_add_key(self):
key = 'pretend this is a public SSH key'
label = 'Test label'
notes = 'My notes'
data = {
'key': key,
'label': label,
'notes': notes,
}
mcall = call(data)
service = self.client['Security_Ssh_Key']
self.sshkey.add_key(key=key, label=label, notes=notes)
service.createObject.assert_has_calls(mcall)
def test_delete_key(self):
id = 1234
mcall = call(id=id)
service = self.client['Security_Ssh_Key']
self.sshkey.delete_key(id)
service.deleteObject.assert_has_calls(mcall)
def test_edit_key(self):
id = 1234
label = 'Test label'
notes = 'My notes'
data = {
'label': label,
'notes': notes,
}
mcall = call(data, id=id)
service = self.client['Security_Ssh_Key']
self.sshkey.edit_key(id, label=label, notes=notes)
service.editObject.assert_has_calls(mcall)
def test_get_key(self):
id = 1234
mcall = call(id=id)
service = self.client['Security_Ssh_Key']
self.sshkey.get_key(id)
service.getObject.assert_has_calls(mcall)
def test_list_keys(self):
service = self.client['Account']
self.sshkey.list_keys(label='some label')
service.getSshKeys.assert_called_with(
filter={'sshKeys': {'label': {'operation': '_= some label'}}})
def test_resolve_ids_label(self):
_id = self.sshkey._get_ids_from_label('Test 1')
self.assertEqual(_id, ['100'])
_id = self.sshkey._get_ids_from_label('nope')
self.assertEqual(_id, [])
