def test_power_query_crashes_for_uknown_power_state(self):
driver = eaton_module.EatonPowerDriver()
system_id = factory.make_name("system_id")
context = self.make_context()
mock_run_process = self.patch(driver, "run_process")
mock_run_process.return_value = "Error"
self.assertRaises(PowerActionError, driver.power_query, system_id,
context)
def test_power_off_calls_run_process(self):
driver = eaton_module.EatonPowerDriver()
system_id = factory.make_name("system_id")
context = self.make_context()
mock_run_process = self.patch(driver, "run_process")
driver.power_off(system_id, context)
command = (["snmpset"] + COMMON_ARGS.format(
context["power_address"],
eaton_module.EatonFunction.OFF,
context["node_outlet"],
).split() + ["i", "0"])
mock_run_process.assert_called_once_with(*command)
def test_power_query_returns_power_state_off(self):
driver = eaton_module.EatonPowerDriver()
system_id = factory.make_name("system_id")
context = self.make_context()
mock_run_process = self.patch(driver, "run_process")
mock_run_process.return_value = eaton_module.EatonState.OFF
result = driver.power_query(system_id, context)
command = ["snmpget"] + COMMON_ARGS.format(
context["power_address"],
eaton_module.EatonFunction.QUERY,
context["node_outlet"],
).split()
mock_run_process.assert_called_once_with(*command)
self.expectThat(result, Equals("off"))
def test_run_process_calls_command_and_returns_output(self):
driver = eaton_module.EatonPowerDriver()
context = self.make_context()
command = ["snmpget"] + COMMON_ARGS.format(
context["power_address"],
eaton_module.EatonFunction.QUERY,
context["node_outlet"],
).split()
mock_run_command = self.patch_run_command(
stdout=COMMON_OUTPUT %
(eaton_module.EatonFunction.QUERY, context["node_outlet"]),
stderr="error_output",
)
output = driver.run_process(*command)
mock_run_command.assert_called_once_with(*command)
self.expectThat(output, Equals(eaton_module.EatonState.OFF))
def test_power_on_calls_run_process(self):
driver = eaton_module.EatonPowerDriver()
system_id = factory.make_name("system_id")
context = self.make_context()
mock_power_query = self.patch(driver, "power_query")
mock_power_query.return_value = "on"
self.patch(driver, "power_off")
mock_sleep = self.patch(eaton_module, "sleep")
mock_run_process = self.patch(driver, "run_process")
driver.power_on(system_id, context)
self.expectThat(mock_power_query,
MockCalledOnceWith(system_id, context))
self.expectThat(mock_sleep,
MockCalledOnceWith(float(context["power_on_delay"])))
command = (["snmpset"] + COMMON_ARGS.format(
context["power_address"],
eaton_module.EatonFunction.ON,
context["node_outlet"],
).split() + ["i", "0"])
mock_run_process.assert_called_once_with(*command)
def test_run_process_crashes_on_no_power_state_match_found(self):
driver = eaton_module.EatonPowerDriver()
self.patch_run_command(stdout="Error")
self.assertRaises(PowerActionError, driver.run_process,
factory.make_name("command"))
def test_run_process_crashes_on_external_process_error(self):
driver = eaton_module.EatonPowerDriver()
self.patch_run_command(returncode=1)
self.assertRaises(PowerActionError, driver.run_process,
factory.make_name("command"))
def test_no_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = True
driver = eaton_module.EatonPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual([], missing)