Modified validation to work with unittest2 discovery.

1 files changed, 19 insertions, 0 deletions

diff --git a/tests/validation/test_validators.py b/tests/validation/test_validators.py
new file mode 100644
index 0000000000..e58d9fd4a6
--- /dev/null
+++ b/tests/validation/test_validators.py
@@ -0,0 +1,19 @@
+from __future__ import absolute_import, unicode_literals
+
+from . import ValidationTestCase
+from .models import ModelToValidate
+
+
+class TestModelsWithValidators(ValidationTestCase):
+    def test_custom_validator_passes_for_correct_value(self):
+        mtv = ModelToValidate(number=10, name='Some Name', f_with_custom_validator=42)
+        self.assertEqual(None, mtv.full_clean())
+
+    def test_custom_validator_raises_error_for_incorrect_value(self):
+        mtv = ModelToValidate(number=10, name='Some Name', f_with_custom_validator=12)
+        self.assertFailsValidation(mtv.full_clean, ['f_with_custom_validator'])
+        self.assertFieldFailsValidationWithMessage(
+            mtv.full_clean,
+            'f_with_custom_validator',
+            ['This is not the answer to life, universe and everything!']
+        )