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'))