class TestAgentActionValidator(unittest.TestCase): def setUp(self): self.dao = Mock() self.validator = AgentActionValidator(self.dao) def test_given_agent_action_does_not_exist_when_validating_then_raises_error(self): self.dao.find_all_agent_action_extensions.return_value = [] destination = AgentDestination(action='login') assert_that(calling(self.validator.validate).with_args(destination), raises(InputError)) def test_given_agent_action_exists_when_validating_then_validation_passes(self): agent_action_extensions = [AgentActionExtension(id=sentinel.extension_id, exten='*31', action='login'), AgentActionExtension(id=sentinel.extension_id, exten='*32', action='logoff')] self.dao.find_all_agent_action_extensions.return_value = agent_action_extensions destination = AgentDestination(action='login') self.validator.validate(destination) self.dao.find_all_agent_action_extensions.assert_called_once_with()
def setUp(self): self.dao = Mock() self.validator = AgentActionValidator(self.dao)