def test_deploys_to_host_with_environment(self):
self.command.execute(node_name="something", image="imageId",
flavor="fileId", public_key_file=StringIO("whatever"),
environment='staging')
expected_host = Host()
expected_host.environment = 'staging'
self.deployer.deploy.assert_any_call(host=expected_host)
def test_deploys_to_host_with_environment(self):
self.command.execute(name="something",
image="imageId",
public_key_file=StringIO("whatever"),
progress=StringIO(),
environment='staging')
expected_host = Host()
expected_host.environment = 'staging'
self.deployer.deploy.assert_any_call(host=expected_host)
def setUp(self):
self.host = Host(name="test.example.com",
ip_address="50.56.57.58")
self.ohai_data = {
'cloud': {
'inet1_address': '1.2.3.4'
},
'network': {
'eth0': 'eth 0 data',
'eth1': 'eth 1 data'
},
'hostname': 'test',
'keys': {
'ssh': {
'public_keys': 'yep'
}
}
}
def test_deploys_to_host_with_kwargs(self):
kwargs = {
'runlist': ['role[web]', 'recipe[test'],
'plugins': 'bootstrap',
'post_plugins': 'all_done'
}
self.command.execute(name="something",
image="imageId",
public_key_file=StringIO("whatever"),
progress=StringIO(),
**kwargs)
expected_args = {'host': Host()}
expected_args.update(kwargs)
self.deployer.deploy.assert_any_call(**expected_args)
def setUp(self):
self.host = Host(host_string="test.example.com",
ip_address="50.56.57.58")
self.ohai_data = {
'cloud': {
'inet1_address': '1.2.3.4'
},
'network': {
'eth0': 'eth 0 data',
'eth1': 'eth 1 data'
},
'hostname': 'test',
'keys': {
'ssh': {
'public_keys': 'yep'
}
}
}
def test_returns_host_information(self):
conn = mock.Mock()
api = self._get_api_with_mocked_conn(conn)
public_ipv4_address = "50.51.52.53"
self.active_node.public_ips = ["2222::2222:2", public_ipv4_address]
self.active_node.extra['password'] = '******'
self.active_node.name = "some name"
conn.create_node.return_value = self.active_node
result = api.create_node(name="some name",
image="5cebb13a-f783-4f8c-8058-c4182c724ccd",
flavor="2",
public_key_file=StringIO("some public key"))
self.assertEquals(
result, Host(name="some name", ip_address=public_ipv4_address))
class ChefDeployerTest(unittest.TestCase):
def setUp(self):
self.host = Host(host_string="test.example.com",
ip_address="50.56.57.58")
self.ohai_data = {
'cloud': {
'inet1_address': '1.2.3.4'
},
'network': {
'eth0': 'eth 0 data',
'eth1': 'eth 1 data'
},
'hostname': 'test',
'keys': {
'ssh': {
'public_keys': 'yep'
}
}
}
def _get_deployer(self, key_filename):
deployer = ChefDeployer(key_filename=key_filename,)
deployer._create_bootstrap_ssh_config = mock.Mock()
return deployer
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_sets_fabric_settings(self, littlechef, lc):
key_filename = "~/.ssh/bootstrap_rsa"
deployer = self._get_deployer(key_filename=key_filename)
deployer.deploy(self.host)
self.assertEquals("root", lc.env.user)
self.assertEquals(self.host.get_host_string(), lc.env.host_string)
self.assertEquals(self.host.get_host_string(), lc.env.host)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_calls_deploys_chef(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
lc.deploy_chef.assert_any_call(ask="no")
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_does_not_call_deploys_chef_if_options_is_set(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, use_opscode_chef=False)
self.assertEquals(0, len(lc.deploy_chef.call_args_list))
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_no_runlist_calls_save_config(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
node_data = {
'predefined': 'data'
}
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host)
littlechef.lib.get_node.assert_any_call(self.host.host_string)
node_data.update(self.ohai_data)
littlechef.chef.save_config.assert_any_call(node_data, force=True)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_without_hostname_gets_node_with_ip_address(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
node_data = {}
self.host.host_string = None
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host)
littlechef.lib.get_node.assert_any_call(self.host.ip_address)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_sets_node_runlist(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
predefined_data = {'run_list': []}
littlechef.lib.get_node.return_value = predefined_data
runlist = ['role[web]', 'recipe[apache2]']
deployer.deploy(self.host, runlist=runlist)
predefined_data.update(self.ohai_data)
predefined_data['run_list'] = runlist
littlechef.chef.save_config.assert_any_call(predefined_data, force=True)
littlechef.lib.get_node.assert_any_call(self.host.host_string)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_environment_sets_environment(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
self.host.environment = 'staging'
expected_data = { 'chef_environment' : 'staging' }
expected_data.update(self.ohai_data)
littlechef.lib.get_node.return_value = expected_data
deployer.deploy(self.host)
littlechef.chef.save_config.assert_any_call(expected_data, force=True)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_creates_temporary_ssh_config_file(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
expected_ssh_config = """User root
IdentityFile ~/.ssh/id_rsa
StrictHostKeyChecking no
Host test.example.com
HostName 50.56.57.58
"""
deployer._create_bootstrap_ssh_config.assert_any_call(
"./.bootstrap-config_{0}".format(self.host.get_host_string()),
expected_ssh_config
)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_uses_temporary_ssh_config_values(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
self.assertTrue(lc.env.use_ssh_config)
self.assertEquals(lc.env.ssh_config_path,
"./.bootstrap-config_{0}".format(self.host.get_host_string()))
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_runs_node_on_ip_address(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
lc.node.assert_any_call(self.host.host_string)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_plugins_executes_plugins_on_node(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
plugin = mock.Mock()
littlechef.lib.import_plugin.return_value = plugin
node_data = {'predefined': [ 'values' ]}
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host, plugins=['awesome_plugin'])
littlechef.lib.import_plugin.assert_any_call('awesome_plugin')
plugin.execute.assert_any_call(node_data)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_plugin_prints_header(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, plugins=['plugin1'])
littlechef.lib.print_header.assert_any_call(
"Executing plugin '{0}' on {1}".format('plugin1', self.host.get_host_string())
)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_multiple_plugins(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, plugins=['plugin1', 'plugin2', 'plugin3'])
littlechef.lib.import_plugin.assert_any_call('plugin1')
littlechef.lib.import_plugin.assert_any_call('plugin2')
littlechef.lib.import_plugin.assert_any_call('plugin3')
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_post_plugins(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, post_plugins=['postplugin'])
littlechef.lib.import_plugin.assert_any_call('postplugin')
def setUp(self):
self.api = mock.Mock(spec=RackspaceApi)
self.deployer = mock.Mock(spec=ChefDeployer)
self.command = RackspaceRebuild(rackspace_api=self.api,
chef_deployer=self.deployer)
self.api.rebuild_node.return_value = Host()
class ChefDeployerTest(unittest.TestCase):
def setUp(self):
self.host = Host(name="test.example.com",
ip_address="50.56.57.58")
self.ohai_data = {
'cloud': {
'inet1_address': '1.2.3.4'
},
'network': {
'eth0': 'eth 0 data',
'eth1': 'eth 1 data'
},
'hostname': 'test',
'keys': {
'ssh': {
'public_keys': 'yep'
}
}
}
def _get_deployer(self, key_filename):
deployer = ChefDeployer(key_filename=key_filename,)
deployer._create_bootstrap_ssh_config = mock.Mock()
return deployer
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_sets_fabric_settings(self, littlechef, lc):
key_filename = "~/.ssh/bootstrap_rsa"
deployer = self._get_deployer(key_filename=key_filename)
deployer.deploy(self.host)
self.assertEquals("root", lc.env.user)
self.assertEquals(self.host.get_host_string(), lc.env.host_string)
self.assertEquals(self.host.get_host_string(), lc.env.host)
self.assertTrue(lc.env.disable_known_hosts)
self.assertEquals(10, lc.env.connection_attempts)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_calls_deploys_chef(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
lc.deploy_chef.assert_any_call(ask="no")
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_does_not_call_deploys_chef_if_option_is_set(
self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, use_opscode_chef=False)
self.assertEquals(0, len(lc.deploy_chef.call_args_list))
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_no_runlist_calls_save_config(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
node_data = {
'predefined': 'data'
}
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host)
littlechef.lib.get_node.assert_any_call(self.host.name)
node_data.update(self.ohai_data)
littlechef.chef.save_config.assert_any_call(node_data, force=True)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_without_hostname_gets_node_with_ip_address(
self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
node_data = {}
self.host.name = None
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host)
littlechef.lib.get_node.assert_any_call(self.host.ip_address)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_sets_node_runlist(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
predefined_data = {'run_list': []}
littlechef.lib.get_node.return_value = predefined_data
runlist = ['role[web]', 'recipe[apache2]']
deployer.deploy(self.host, runlist=runlist)
predefined_data.update(self.ohai_data)
predefined_data['run_list'] = runlist
littlechef.chef.save_config.assert_any_call(
predefined_data,
force=True)
littlechef.lib.get_node.assert_any_call(self.host.name)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_environment_sets_environment(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
self.host.environment = 'staging'
expected_data = {'chef_environment': 'staging'}
expected_data.update(self.ohai_data)
littlechef.lib.get_node.return_value = expected_data
deployer.deploy(self.host)
littlechef.chef.save_config.assert_any_call(expected_data, force=True)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_creates_temporary_ssh_config_file(
self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
expected_ssh_config = """User root
IdentityFile ~/.ssh/id_rsa
StrictHostKeyChecking no
Host test.example.com
HostName 50.56.57.58
"""
deployer._create_bootstrap_ssh_config.assert_any_call(
"./.bootstrap-config_{0}".format(self.host.get_host_string()),
expected_ssh_config
)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_runlist_uses_temporary_ssh_config_values(
self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host)
self.assertTrue(lc.env.use_ssh_config)
self.assertEquals(lc.env.ssh_config_path,
"./.bootstrap-config_{0}".format(
self.host.get_host_string()))
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_plugins_executes_plugins_on_node(
self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
plugin = mock.Mock()
littlechef.lib.import_plugin.return_value = plugin
node_data = {'predefined': ['values']}
littlechef.lib.get_node.return_value = node_data
deployer.deploy(self.host, plugins=['awesome_plugin'])
littlechef.lib.import_plugin.assert_any_call('awesome_plugin')
plugin.execute.assert_any_call(node_data)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_plugin_prints_header(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, plugins=['plugin1'])
littlechef.lib.print_header.assert_any_call(
"Executing plugin '{0}' on {1}".format(
'plugin1',
self.host.get_host_string())
)
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_multiple_plugins(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, plugins=['plugin1', 'plugin2', 'plugin3'])
littlechef.lib.import_plugin.assert_any_call('plugin1')
littlechef.lib.import_plugin.assert_any_call('plugin2')
littlechef.lib.import_plugin.assert_any_call('plugin3')
@mock.patch('littlechef_rackspace.deploy.lc')
@mock.patch('littlechef_rackspace.deploy.littlechef')
def test_deploy_with_post_plugins(self, littlechef, lc):
deployer = self._get_deployer(key_filename="~/.ssh/id_rsa")
deployer.deploy(self.host, post_plugins=['postplugin'])
littlechef.lib.import_plugin.assert_any_call('postplugin')
