[1.8.x] Fixed model_fields tests to use django TestCase.

Backport of 97bb48d9ba from master
This commit is contained in:
Jon Dufresne 2015-03-26 09:49:13 -07:00 committed by Tim Graham
parent a988cc8f18
commit c38d8f0f87
1 changed files with 5 additions and 4 deletions

View File

@ -202,7 +202,7 @@ class ForeignKeyTests(test.TestCase):
self.assertIsInstance(rel_name, six.text_type)
class DateTimeFieldTests(unittest.TestCase):
class DateTimeFieldTests(test.TestCase):
def test_datetimefield_to_python_usecs(self):
"""DateTimeField.to_python should support usecs"""
f = models.DateTimeField()
@ -232,7 +232,7 @@ class DateTimeFieldTests(unittest.TestCase):
self.assertEqual(obj.t, tim)
class BooleanFieldTests(unittest.TestCase):
class BooleanFieldTests(test.TestCase):
def _test_get_db_prep_lookup(self, f):
self.assertEqual(f.get_db_prep_lookup('exact', True, connection=connection), [True])
self.assertEqual(f.get_db_prep_lookup('exact', '1', connection=connection), [True])
@ -371,8 +371,9 @@ class BooleanFieldTests(unittest.TestCase):
self.assertFalse(boolean_field.has_default())
b = BooleanModel()
self.assertIsNone(b.bfield)
with self.assertRaises(IntegrityError):
b.save()
with transaction.atomic():
with self.assertRaises(IntegrityError):
b.save()
finally:
boolean_field.default = old_default