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)
