Ticket #20463: get_or_create_pgsql-test.diff

tests/get_or_create/tests.py

@@ -3 +3 @@
 from datetime import date
 import traceback
 
-from django.db import IntegrityError
+from django.db import IntegrityError, DataError, connection
 from django.test import TestCase, TransactionTestCase
+from django.utils.unittest.case import skipUnless
 
 from .models import Person, ManualPrimaryKeyTest, Profile
 
@@ -64 +65 @@
             formatted_traceback = traceback.format_exc()
             self.assertIn('obj.save', formatted_traceback)
 
+    @skipUnless(connection.vendor == 'postgresql', "This test only applies to PostgreSQL")
+    def test_savepoint_rollback(self):
+        # Provoke a DataError in .get_or_create()
+        try:
+            Person.objects.get_or_create(birthday=date(1970, 1, 1), defaults={
+                'first_name': "\xff",
+                'last_name': "\xff"
+            })
+        except DataError:
+            pass
+        # Create another object to make sure the database is not in a "broken" state
+        Person.objects.create(first_name="Bob", last_name="Ross", birthday=date(1950, 1, 1))
+
 
 class GetOrCreateTransactionTests(TransactionTestCase):