diff options
Diffstat (limited to 'tests/get_or_create/tests.py')
| -rw-r--r-- | tests/get_or_create/tests.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/get_or_create/tests.py b/tests/get_or_create/tests.py index d4aca54c91..1ddb2447fe 100644 --- a/tests/get_or_create/tests.py +++ b/tests/get_or_create/tests.py @@ -45,14 +45,16 @@ class GetOrCreateTests(TestCase): # If you don't specify a value or default value for all required # fields, you will get an error. - self.assertRaises(IntegrityError, + self.assertRaises( + IntegrityError, Person.objects.get_or_create, first_name="Tom", last_name="Smith" ) # If you specify an existing primary key, but different other fields, # then you will get an error and data will not be updated. ManualPrimaryKeyTest.objects.create(id=1, data="Original") - self.assertRaises(IntegrityError, + self.assertRaises( + IntegrityError, ManualPrimaryKeyTest.objects.get_or_create, id=1, data="Different" ) self.assertEqual(ManualPrimaryKeyTest.objects.get(id=1).data, "Original") @@ -181,7 +183,8 @@ class UpdateOrCreateTests(TestCase): # If you specify an existing primary key, but different other fields, # then you will get an error and data will not be updated. ManualPrimaryKeyTest.objects.create(id=1, data="Original") - self.assertRaises(IntegrityError, + self.assertRaises( + IntegrityError, ManualPrimaryKeyTest.objects.update_or_create, id=1, data="Different" ) self.assertEqual(ManualPrimaryKeyTest.objects.get(id=1).data, "Original") |