class BaseCommandTest(CommandBase):
def setUp(self):
super(BaseCommandTest, self).setUp()
self.cmd = BaseCommand(self.config, self.provider, self.env)
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_check_preconditions_okay(self):
self.setup_env()
self.assertEqual(self.cmd.check_preconditions(), [1])
def test_check_preconditions_host_exist(self):
self.setup_env({
'environments': {
'linode': {
'type': 'null',
'bootstrap-host': '1.1.1.1'}}})
try:
self.cmd.check_preconditions()
except ConfigError, e:
self.assertIn('already has a bootstrap-host', str(e))
else:
class BaseCommandTest(CommandBase):
def setUp(self):
super(BaseCommandTest, self).setUp()
self.cmd = BaseCommand(self.config, self.provider, self.env)
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_check_preconditions_okay(self):
self.setup_env()
self.assertEqual(self.cmd.check_preconditions(), [1])
def test_check_preconditions_host_exist(self):
self.setup_env({
'environments': {
'linode': {
'type': 'null',
'bootstrap-host': '1.1.1.1'
}
}
})
try:
self.cmd.check_preconditions()
except ConfigError, e:
self.assertIn('already has a bootstrap-host', str(e))
else:
def setUp(self):
super(BaseCommandTest, self).setUp()
self.cmd = BaseCommand(self.config, self.provider, self.env)
def setUp(self):
super(BaseCommandTest, self).setUp()
self.cmd = BaseCommand(self.config, self.provider, self.env)