1 import unittest

     2 

     3 from django.core.exceptions import ValidationError

     4 

     5 class ValidationTestCase(unittest.TestCase):

     6     def assertFailsValidation(self, clean, failed_fields):

     7         self.assertRaises(ValidationError, clean)

     8         try:

     9             clean()

    10         except ValidationError, e:

    11             self.assertEquals(sorted(failed_fields), sorted(e.message_dict.keys()))

    12     

    13     def assertFieldFailsValidationWithMessage(self, clean, field_name, message):

    14         self.assertRaises(ValidationError, clean)

    15         try:

    16             clean()

    17         except ValidationError, e:

    18             self.assertTrue(field_name in e.message_dict)

    19             self.assertEquals(message, e.message_dict[field_name])

    20 

    21