def test__logs_other_errors(self):
node = yield deferToDatabase(transactional(factory.make_Node))
mock_power_query = self.patch(Node, "power_query")
mock_power_query.side_effect = factory.make_exception('Error')
mock_log_err = self.patch(power.log, "err")
yield power.update_power_state_of_node(node.system_id)
self.assertThat(
mock_log_err,
MockCalledOnceWith(
ANY, "Failed to update power state of machine after state "
"transition."))
def test__traps_failure_for_UnknownPowerType(self):
node = yield deferToDatabase(transactional(factory.make_Node))
mock_power_query = self.patch(Node, "power_query")
mock_power_query.side_effect = UnknownPowerType()
power_state = yield power.update_power_state_of_node(node.system_id)
self.assertIsNone(power_state)
def test__traps_failure_for_Node_DoesNotExist(self):
self.assertIsNone(
power.update_power_state_of_node(factory.make_name("system_id"))
)
def test__retrieves_power_state(self):
node = yield deferToDatabase(transactional(factory.make_Node))
mock_power_query = self.patch(Node, "power_query")
mock_power_query.return_value = POWER_STATE.ON
power_state = yield power.update_power_state_of_node(node.system_id)
self.assertEqual(power_state, POWER_STATE.ON)
