def test_configcluagent_call_before_execute(self): cluagent = ConfigurableCluAgent({}) before_execute=Mock() cluagent.before_execute=before_execute with self.assertRaises(CluException): cluagent.run() before_execute.assert_called_once_with()
def test_configcluagent_call_before_rexecute_after(self): cluagent = ConfigurableCluAgent({}) before_execute=Mock() after_execute=Mock() execute=Mock() cluagent.before_execute=before_execute cluagent.execute=execute cluagent.after_execute=after_execute cluagent.run() before_execute.assert_called_once_with() execute.assert_called_once_with() after_execute.assert_called_once_with()
def test_cluagent_call_ensure_afterexecute_on_run_exception(self): """ Ensure that ensure_after_execute is called when execute() raise an Exception """ cluagent = ConfigurableCluAgent({}) before_execute=Mock() after_execute=Mock() ensure_after_execute=Mock() execute=Mock(side_effect=Exception("In your face")) cluagent.before_execute=before_execute cluagent.execute=execute cluagent.after_execute=after_execute cluagent.ensure_after_execute=ensure_after_execute with(self.assertRaises(CluAgentException)) as ex: cluagent.run() before_execute.assert_called_once_with() execute.assert_called_once_with() ensure_after_execute.assert_called_once_with()
def test_cluagent_call_ensure_afterexecute_on_after_execute_exception_is_first(self): """ Ensure that when execute() and after_execute() raises an Exception, the first is raised""" ex_ensure_after=Exception("In your face") ensure_after_execute=Mock(side_effect=ex_ensure_after) cluagent = ConfigurableCluAgent({}) before_execute=Mock() execute=Mock() after_execute=Mock() cluagent.before_execute=before_execute cluagent.execute=execute cluagent.after_execute=after_execute cluagent.ensure_after_execute=ensure_after_execute with(self.assertRaises(CluAgentException)) as e: cluagent.run() before_execute.assert_called_once_with() execute.assert_called_once_with() after_execute.assert_called_once_with() ensure_after_execute.assert_called_once_with()
def test_configcluagent_execute_raises_cluagentexception(self): cluagent = ConfigurableCluAgent({}) with self.assertRaises(CluException): cluagent.run()