Ticket #16682: 16682-atomic.diff

tests/transactions/tests.py

@@ -1 +1 @@
 from __future__ import unicode_literals
 
+import os
 import sys
 import threading
 import time
@@ -218 +219 @@
             transaction.savepoint_rollback(sid)
         self.assertQuerysetEqual(Reporter.objects.all(), ['<Reporter: Tintin>'])
 
+    def test_rollback_on_keyboardinterrupt(self):
+        try:
+            with transaction.atomic():
+                Reporter.objects.create(first_name='Haddock')
+                # Send SIGINT (simulate Ctrl-C). One call isn't enough.
+                os.kill(os.getpid(), 2)
+                os.kill(os.getpid(), 2)
+        except KeyboardInterrupt:
+            pass
+        self.assertEqual(Reporter.objects.all().count(), 0)
+
 
 class AtomicInsideTransactionTests(AtomicTests):
     """All basic tests for atomic should also pass within an existing transaction."""