class TestFuncKeyValidator(unittest.TestCase):
def setUp(self):
self.first_dest_validator = Mock(Validator)
self.second_dest_validator = Mock(Validator)
self.validator = FuncKeyValidator({'foobar': [self.first_dest_validator,
self.second_dest_validator]})
def test_given_no_validator_for_destination_when_validating_then_raises_error(self):
destination = Mock(type='spam')
model = FuncKey(destination=destination)
assert_that(calling(self.validator.validate).with_args(model),
raises(InputError))
def test_given_multiple_validators_for_destination_when_validating_then_calls_each_validator(self):
destination = Mock(type='foobar')
model = FuncKey(destination=destination)
self.validator.validate(model)
self.first_dest_validator.validate.assert_called_once_with(destination)
self.second_dest_validator.validate.assert_called_once_with(destination)
def setUp(self):
self.first_dest_validator = Mock(Validator)
self.second_dest_validator = Mock(Validator)
self.validator = FuncKeyValidator({'foobar': [self.first_dest_validator,
self.second_dest_validator]})
