def test_get_amt_command_raises_power_error(self): amt_power_driver = AMTPowerDriver() for error, error_info in AMT_ERRORS.items(): self.patch_run_command(stderr=error.encode("utf-8"), decode=True) self.assertRaises( error_info.get("exception"), amt_power_driver._get_amt_command, factory.make_ipv4_address(), factory.make_name("power_pass"), )
def test__get_amt_command_raises_power_error(self): amt_power_driver = AMTPowerDriver() for error, error_info in AMT_ERRORS.items(): popen_mock = self.patch(amt_module, 'Popen') process = popen_mock.return_value process.communicate.return_value = (b'', error.encode('utf-8')) self.assertRaises(error_info.get('exception'), amt_power_driver._get_amt_command, factory.make_ipv4_address(), factory.make_name('power_pass'))