2010-10-11 20:55:17 +08:00
|
|
|
from django.utils import unittest
|
2010-01-05 11:56:19 +08:00
|
|
|
|
|
|
|
from django.core.exceptions import ValidationError
|
|
|
|
|
|
|
|
class ValidationTestCase(unittest.TestCase):
|
|
|
|
def assertFailsValidation(self, clean, failed_fields):
|
|
|
|
self.assertRaises(ValidationError, clean)
|
|
|
|
try:
|
|
|
|
clean()
|
|
|
|
except ValidationError, e:
|
2011-03-03 23:04:39 +08:00
|
|
|
self.assertEqual(sorted(failed_fields), sorted(e.message_dict.keys()))
|
2010-01-05 11:56:19 +08:00
|
|
|
|
|
|
|
def assertFieldFailsValidationWithMessage(self, clean, field_name, message):
|
|
|
|
self.assertRaises(ValidationError, clean)
|
|
|
|
try:
|
|
|
|
clean()
|
|
|
|
except ValidationError, e:
|
|
|
|
self.assertTrue(field_name in e.message_dict)
|
2011-03-03 23:04:39 +08:00
|
|
|
self.assertEqual(message, e.message_dict[field_name])
|
2010-01-05 11:56:19 +08:00
|
|
|
|
|
|
|
|