def test_get_ssh_keys(self):
self.provider.get_ssh_keys.return_value = [
SSHKey.from_dict({'id': 1, 'name': 'abc'}),
SSHKey.from_dict({'id': 32, 'name': 'bcd'})]
self.assertEqual(
self.cmd.get_linode_ssh_keys(),
[1, 32])
def test_get_ssh_keys(self):
self.provider.get_ssh_keys.return_value = [
SSHKey.from_dict({
'id': 1,
'name': 'abc'
}),
SSHKey.from_dict({
'id': 32,
'name': 'bcd'
})
]
self.assertEqual(self.cmd.get_linode_ssh_keys(), [1, 32])
def setup_env(self, conf=None):
self.provider.get_ssh_keys.return_value = [
SSHKey.from_dict({'id': 1, 'name': 'abc'})]
self.config.series = "precise"
with tempfile.NamedTemporaryFile(delete=False) as f:
self.config.get_env_conf.return_value = f.name
self.config.get_env_name.return_value = 'linode'
if conf is None:
conf = {
'environments': {
'linode': {
'type': 'null',
'bootstrap-host': None}}}
f.write(yaml.safe_dump(conf))
f.flush()
self.addCleanup(lambda: os.remove(f.name))
def setup_env(self, conf=None):
self.provider.get_ssh_keys.return_value = [
SSHKey.from_dict({
'id': 1,
'name': 'abc'
})
]
self.config.series = "precise"
with tempfile.NamedTemporaryFile(delete=False) as f:
self.config.get_env_conf.return_value = f.name
self.config.get_env_name.return_value = 'linode'
if conf is None:
conf = {
'environments': {
'linode': {
'type': 'null',
'bootstrap-host': None
}
}
}
f.write(yaml.safe_dump(conf))
f.flush()
self.addCleanup(lambda: os.remove(f.name))