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]})