def test_power_control_off(self):
mock_login = self.patch(virsh.VirshSSH, 'login')
mock_login.return_value = True
mock_state = self.patch(virsh.VirshSSH, 'get_state')
mock_state.return_value = virsh.VirshVMState.ON
mock_poweroff = self.patch(virsh.VirshSSH, 'poweroff')
poweraddr = factory.make_name('poweraddr')
machine = factory.make_name('machine')
virsh.power_control_virsh(poweraddr, machine, 'off')
self.assertThat(mock_login, MockCalledOnceWith(poweraddr, None))
self.assertThat(mock_state, MockCalledOnceWith(machine))
self.assertThat(mock_poweroff, MockCalledOnceWith(machine))
def test_power_control_off(self):
mock_login = self.patch(virsh.VirshSSH, "login")
mock_login.return_value = True
mock_state = self.patch(virsh.VirshSSH, "get_state")
mock_state.return_value = virsh.VirshVMState.ON
mock_poweroff = self.patch(virsh.VirshSSH, "poweroff")
poweraddr = factory.make_name("poweraddr")
machine = factory.make_name("machine")
virsh.power_control_virsh(poweraddr, machine, "off")
self.assertThat(mock_login, MockCalledOnceWith(poweraddr, None))
self.assertThat(mock_state, MockCalledOnceWith(machine))
self.assertThat(mock_poweroff, MockCalledOnceWith(machine))
def power_off(self, system_id, context):
"""Power off Virsh node."""
power_change = 'off'
poweraddr, machine, password = extract_virsh_parameters(context)
power_control_virsh(poweraddr, machine, power_change, password)