class TerminateMachineTest(CommandBase):
def setUp(self):
super(TerminateMachineTest, self).setUp()
self.cmd = TerminateMachine(self.config, self.provider, self.env)
def test_terminate_machine(self):
self.setup_env()
self.env.status.return_value = {
'machines': {
'1': {
'dns-name': '10.0.1.23',
'instance-id': 'manual:ip_address'}
}}
self.provider.get_instances.return_value = [
Instance(dict(
id=221, hostname="slayer-123123",
primaryIpAddress="10.0.1.23")),
Instance(dict(
id=258, hostname="slayer-209123",
primaryIpAddress="10.0.1.103"))]
self.config.options.machines = ["1"]
self.cmd.run()
self.provider.terminate_instance.assert_called_once_with(221)
def setUp(self):
super(TerminateMachineTest, self).setUp()
self.cmd = TerminateMachine(self.config, self.provider, self.env)