class BootstrapTest(CommandBase):
def setUp(self):
super(BootstrapTest, self).setUp()
self.cmd = Bootstrap(self.config, self.provider, self.env)
@mock.patch('juju_slayer.ops.ssh')
def test_bootstrap(self, mock_ssh):
self.setup_env()
self.env.is_running.return_value = False
self.config.series = "precise"
mock_ssh.check_ssh.return_value = True
mock_ssh.update_instance.return_value = True
self.provider.get_instance.return_value = Instance(dict(
id=2121,
hostname='slayer-13290123j13',
primaryIpAddress="10.0.2.1"))
self.cmd.run()
mock_ssh.check_ssh.assert_called_once_with('10.0.2.1')
mock_ssh.update_instance.assert_called_once_with('10.0.2.1')
def setUp(self):
super(BootstrapTest, self).setUp()
self.cmd = Bootstrap(self.config, self.provider, self.env)
