def test_change_machine_status_calls_change_lxc_machine_status_with_machines_from_config(
self):
change_machine_status(settings.CONFIG)
calls = [
call(m, status="stop") for m in settings.CONFIG["machines"].keys()
]
self.change_lxc_machine_status.assert_has_calls(calls)
def test_change_machine_status_calls_change_lxc_machine_status_with_different_status(
self):
change_machine_status(settings.CONFIG, status="start")
calls = [
call(m, status="start")
for m in settings.CONFIG["machines"].keys()
]
self.change_lxc_machine_status.assert_has_calls(calls)
def preform_stop_action(self):
change_machine_status(self.config,
machines=self._machines,
status="stop")
# If specific machines are specified, we don't want to mess with the interfaces
if self._machines:
logger.warning(
"Not bringing down VNet interfaces as we are only stopping specific machines, this may leave lingering sniffers"
)
else:
bring_down_vnet_interfaces(self.config)
def preform_start_action(self):
bring_up_vnet_interfaces(self.config, sniffer=self.sniffer)
change_machine_status(self.config,
machines=self._machines,
status="start")
def test_change_machine_status_skips_non_existent_machines(self):
machines = ["banaan1"]
change_machine_status(settings.CONFIG, machines=machines)
self.assertFalse(self.change_lxc_machine_status.called)
def test_change_machine_status_calls_change_lxc_machine_status_only_with_given_machine_list(
self):
machines = ["router100"]
change_machine_status(settings.CONFIG, machines=machines)
self.change_lxc_machine_status.assert_called_once_with(machines[0],
status="stop")
def test_change_machine_status_raises_not_implemented_error_when_invalid_status(
self):
with self.assertRaises(NotImplementedError):
change_machine_status(settings.CONFIG, status="blaap")