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({'id': 1, 'label': 'abc'}),
SSHKey({'id': 32, 'label': 'bcd'})]
self.assertEqual(
self.cmd.get_slayer_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': {
'softlayer': {
'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)
