def test_amttool_restart_power_cycles(self):
amt_power_driver = AMTPowerDriver()
ip_address = factory.make_ipv4_address()
power_pass = factory.make_name('power_pass')
amttool_boot_mode = factory.make_name('amttool_boot_mode')
_issue_amttool_command_mock = self.patch(
amt_power_driver, '_issue_amttool_command')
amt_power_driver.amttool_restart(
ip_address, power_pass, amttool_boot_mode)
self.assertThat(
_issue_amttool_command_mock, MockCalledOnceWith(
'power_cycle', ip_address, power_pass,
amttool_boot_mode=amttool_boot_mode, stdin=b'yes'))
def test_amttool_restart_power_cycles(self):
amt_power_driver = AMTPowerDriver()
ip_address = factory.make_ipv4_address()
power_pass = factory.make_name("power_pass")
amttool_boot_mode = factory.make_name("amttool_boot_mode")
_issue_amttool_command_mock = self.patch(amt_power_driver,
"_issue_amttool_command")
amt_power_driver.amttool_restart(ip_address, power_pass,
amttool_boot_mode)
self.assertThat(
_issue_amttool_command_mock,
MockCalledOnceWith(
"power_cycle",
ip_address,
power_pass,
amttool_boot_mode=amttool_boot_mode,
stdin=b"yes",
),
)
