from unittest import TestCase from django.core.exceptions import ValidationError from django.core.validators import validate_integer, validate_email, RequiredIfOtherFieldBlank class TestSimpleValidators(TestCase): pass SIMPLE_VALIDATORS_VALUES = ( # (validator, value, expected), (validate_integer, '42', None), (validate_integer, '-42', None), (validate_integer, -42, None), (validate_integer, -42.5, None), (validate_integer, None, ValidationError), (validate_integer, 'a', ValidationError), (validate_email, 'email@here.com', None), (validate_email, 'weirder-email@here.and.there.com', None), (validate_email, None, ValidationError), (validate_email, '', ValidationError), (validate_email, 'abc', ValidationError), (validate_email, 'a @x.cz', ValidationError), (validate_email, 'something@@somewhere.com', ValidationError), ) def get_simple_test_func(validator, expected, value, num): if isinstance(expected, type) and issubclass(expected, Exception): test_mask = 'test_%s_raises_error_%d' def test_func(self): self.assertRaises(expected, validator, value) else: test_mask = 'test_%s_%d' def test_func(self): self.assertEqual(expected, validator(value)) test_name = test_mask % (validator.__name__, num) return test_name, test_func test_counter = {} for validator, value, expected in SIMPLE_VALIDATORS_VALUES: num = test_counter[validator] = test_counter.setdefault(validator, 0) + 1 setattr(TestSimpleValidators, *get_simple_test_func(validator, expected, value, num)) class TestComplexValidators(TestCase): pass COMPLEX_VALIDATORS_VALUES = ( #(validator, value, all_values, obj, expected), (RequiredIfOtherFieldBlank('other'), 'given', {'other': 'given'}, None, None), (RequiredIfOtherFieldBlank('other'), '', {'other': 'given'}, None, None), (RequiredIfOtherFieldBlank('other'), 'given', {}, None, AssertionError), (RequiredIfOtherFieldBlank('other'), '', {}, None, AssertionError), (RequiredIfOtherFieldBlank('other'), '', {'other': ''}, None, ValidationError), ) def get_complex_test_func(validator, expected, value, all_values, obj, num): if isinstance(expected, type) and issubclass(expected, Exception): test_mask = 'test_%s_raises_error_%d' def test_func(self): self.assertRaises(expected, validator, value, all_values=all_values, obj=obj) else: test_mask = 'test_%s_%d' def test_func(self): self.assertEqual(expected, validator(value, all_values=all_values, obj=obj)) test_name = test_mask % (validator.__class__.__name__, num) return test_name, test_func test_counter = {} for validator, value, all_values, obj, expected in COMPLEX_VALIDATORS_VALUES: num = test_counter[validator.__class__.__name__] = test_counter.setdefault(validator.__class__.__name__, 0) + 1 setattr(TestComplexValidators, *get_complex_test_func(validator, expected, value, all_values, obj, num))