Ticket #16682: 16682-test.diff

tests/regressiontests/transactions_regress/tests.py

@@ -1 +1 @@
 from __future__ import absolute_import
+import os
 
 from django.core.exceptions import ImproperlyConfigured
 from django.db import connection, transaction
@@ -174 +175 @@
         """
         self.test_failing_query_transaction_closed()
 
+    def test_rollback_on_keyboardinterrupt(self):
+        try:
+            with transaction.commit_on_success():
+                Mod.objects.create(fld=17624)
+                # Send SIGINT (simulate Ctrl-C) to self, one call seems not enough??
+                os.kill(os.getpid(), 2)
+                os.kill(os.getpid(), 2)
+        except KeyboardInterrupt:
+            pass
+        self.assertEqual(Mod.objects.all().count(), 0)
+
 
 class TestManyToManyAddTransaction(TransactionTestCase):
     def test_manyrelated_add_commit(self):