class TestServiceValidator(unittest.TestCase): def setUp(self): self.dao = Mock() self.validator = ServiceValidator(self.dao) def test_given_service_does_not_exist_when_validating_then_raises_error(self): self.dao.find_all_service_extensions.return_value = [] destination = ServiceDestination(service='enablevm') assert_that(calling(self.validator.validate).with_args(destination), raises(InputError)) def test_given_service_exists_when_validating_then_validation_passes(self): service_extensions = [ServiceExtension(id=sentinel.extension_id, exten='*25', service='enablednd'), ServiceExtension(id=sentinel.extension_id, exten='*90', service='enablevm')] self.dao.find_all_service_extensions.return_value = service_extensions destination = ServiceDestination(service='enablevm') self.validator.validate(destination) self.dao.find_all_service_extensions.assert_called_once_with()
def setUp(self): self.dao = Mock() self.validator = ServiceValidator(self.dao)