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)