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()