Ticket #19442: transaction_behaviour.diff

tests/modeltests/transactions/tests.py

@@ -2 +2 @@
 
 from django.db import connection, transaction, IntegrityError
 from django.test import TransactionTestCase, skipUnlessDBFeature
+from unittest import skipUnless
 
 from .models import Reporter
 
@@ -177 +178 @@
         """
         execute_bad_sql = transaction.commit_on_success(self.execute_bad_sql)
         self.assertRaises(IntegrityError, execute_bad_sql)
+
+    @skipUnless(connection.vendor=='postgres', 'Needs Postgres...')
+    def test_tx(self):
+        from psycopg2 import extensions
+        IN_TX = extensions.TRANSACTION_STATUS_IN_TX
+        with transaction.commit_on_success():
+            cursor = connection.cursor()
+            cursor.execute("INSERT INTO transactions_reporter (first_name, last_name, email) VALUES ('Douglas', 'Adams', 'da@example.com');")
+        # commit_on_success commits raw SQL
+        transaction.rollback()
+        self.assertTrue(Reporter.objects.filter(last_name='Adams').exists())
+        Reporter.objects.filter(last_name='Adams').delete()
+
+        with transaction.autocommit():
+            cursor = connection.cursor()
+            cursor.execute("INSERT INTO transactions_reporter (first_name, last_name, email) VALUES ('Douglas', 'Adams', 'da@example.com');")
+        # autocommit doesn't
+        transaction.rollback()
+        self.assertFalse(Reporter.objects.filter(last_name='Adams').exists())
+        Reporter.objects.filter(last_name='Adams').delete()
+
+        with transaction.autocommit():
+            Reporter.objects.create(first_name='Douglas', last_name='Adams',
+                                    email='da@example.com')
+        # But it does commit when writing using Django's ORM
         transaction.rollback()
+        self.assertTrue(Reporter.objects.filter(last_name='Adams').exists())
+
+        with transaction.commit_on_success():
+            Reporter.objects.filter(last_name='Adams').exists()
+        # commit_on_success also commits when only reading
+        self.assertNotEqual(connection.connection.get_transaction_status(), IN_TX)
+        self.assertFalse(transaction.is_dirty())
+        with transaction.autocommit():
+            Reporter.objects.filter(last_name='Adams').exists()
+        # Using autocommit we are in transaction...
+        self.assertEqual(connection.connection.get_transaction_status(), IN_TX)
+        # But the transaction isn't dirty
+        self.assertFalse(transaction.is_dirty())
+
 
 class TransactionContextManagerTests(TransactionTestCase):
     def create_reporter_and_fail(self):