Fixed model_fields tests to use django TestCase.

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

View File

@ -245,7 +245,7 @@ class ManyToManyFieldTests(test.TestCase):
) )
class DateTimeFieldTests(unittest.TestCase): class DateTimeFieldTests(test.TestCase):
def test_datetimefield_to_python_usecs(self): def test_datetimefield_to_python_usecs(self):
"""DateTimeField.to_python should support usecs""" """DateTimeField.to_python should support usecs"""
f = models.DateTimeField() f = models.DateTimeField()
@ -275,7 +275,7 @@ class DateTimeFieldTests(unittest.TestCase):
self.assertEqual(obj.t, tim) self.assertEqual(obj.t, tim)
class BooleanFieldTests(unittest.TestCase): class BooleanFieldTests(test.TestCase):
def _test_get_db_prep_lookup(self, f): 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', True, connection=connection), [True])
self.assertEqual(f.get_db_prep_lookup('exact', '1', connection=connection), [True]) self.assertEqual(f.get_db_prep_lookup('exact', '1', connection=connection), [True])
@ -414,8 +414,9 @@ class BooleanFieldTests(unittest.TestCase):
self.assertFalse(boolean_field.has_default()) self.assertFalse(boolean_field.has_default())
b = BooleanModel() b = BooleanModel()
self.assertIsNone(b.bfield) self.assertIsNone(b.bfield)
with self.assertRaises(IntegrityError): with transaction.atomic():
b.save() with self.assertRaises(IntegrityError):
b.save()
finally: finally:
boolean_field.default = old_default boolean_field.default = old_default