def test_power_off_powers_off_with_wsman(self): amt_power_driver = AMTPowerDriver() context = make_context() _get_amt_command_mock = self.patch( amt_power_driver, "_get_amt_command" ) _get_amt_command_mock.return_value = "wsman" wsman_query_state_mock = self.patch( amt_power_driver, "wsman_query_state" ) wsman_query_state_mock.return_value = "on" wsman_power_off_mock = self.patch(amt_power_driver, "wsman_power_off") amt_power_driver.power_off(context["system_id"], context) self.expectThat( _get_amt_command_mock, MockCalledOnceWith(context["ip_address"], context["power_pass"]), ) self.expectThat( wsman_query_state_mock, MockCalledOnceWith(context["ip_address"], context["power_pass"]), ) self.expectThat( wsman_power_off_mock, MockCalledOnceWith(context["ip_address"], context["power_pass"]), )
def test_power_off_powers_off_with_wsman(self): amt_power_driver = AMTPowerDriver() context = make_context() _get_amt_command_mock = self.patch(amt_power_driver, '_get_amt_command') _get_amt_command_mock.return_value = 'wsman' wsman_query_state_mock = self.patch(amt_power_driver, 'wsman_query_state') wsman_query_state_mock.return_value = 'on' wsman_power_off_mock = self.patch(amt_power_driver, 'wsman_power_off') amt_power_driver.power_off(context['system_id'], context) self.expectThat( _get_amt_command_mock, MockCalledOnceWith(context['ip_address'], context['power_pass'])) self.expectThat( wsman_query_state_mock, MockCalledOnceWith(context['ip_address'], context['power_pass'])) self.expectThat( wsman_power_off_mock, MockCalledOnceWith(context['ip_address'], context['power_pass']))