From 8544d0afe8847dbfe1b83d66ebe8f754f832372e Mon Sep 17 00:00:00 2001
From: Bertrand Bordage <bordage.bertrand@gmail.com>
Date: Mon, 29 Sep 2014 12:26:28 +0200
Subject: [PATCH] Adds some transaction tests.

---
 cachalot/tests.py | 69 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/cachalot/tests.py b/cachalot/tests.py
index dfbd826..eda5a32 100644
--- a/cachalot/tests.py
+++ b/cachalot/tests.py
@@ -897,3 +897,72 @@ class TransactionTestCase(TestCase):
         with self.assertNumQueries(1):
             data2 = list(Test.objects.all())
         self.assertListEqual(data2, [])
+
+    def test_successful_write_transaction(self):
+        with self.assertNumQueries(1):
+            data1 = list(Test.objects.all())
+        self.assertListEqual(data1, [])
+
+        with self.assertNumQueries(3):
+            with transaction.atomic():
+                t1 = Test.objects.create(name='test1')
+        with self.assertNumQueries(1):
+            data2 = list(Test.objects.all())
+        self.assertListEqual(data2, [t1])
+
+        with self.assertNumQueries(3):
+            with transaction.atomic():
+                t2 = Test.objects.create(name='test2')
+        with self.assertNumQueries(1):
+            data3 = list(Test.objects.all())
+        self.assertListEqual(data3, [t1, t2])
+
+    # TODO: Remove this ``expectedFailure``
+    #       when cachalot will handle transactions
+    @expectedFailure
+    def test_unsuccessful_write_transaction(self):
+        with self.assertNumQueries(1):
+            data1 = list(Test.objects.all())
+        self.assertListEqual(data1, [])
+
+        with self.assertNumQueries(3):
+            try:
+                with transaction.atomic():
+                    Test.objects.create(name='test')
+                    raise ZeroDivisionError
+            except ZeroDivisionError:
+                pass
+        with self.assertNumQueries(0):
+            data2 = list(Test.objects.all())
+        self.assertListEqual(data2, [])
+        with self.assertNumQueries(1):
+            with self.assertRaises(Test.DoesNotExist):
+                Test.objects.get(name='test')
+
+    def test_cache_inside_transaction(self):
+        with self.assertNumQueries(3):
+            with transaction.atomic():
+                data1 = list(Test.objects.all())
+                data2 = list(Test.objects.all())
+        self.assertListEqual(data2, data1)
+        self.assertListEqual(data2, [])
+
+    @skip(NotImplementedError)
+    def test_invalidation_inside_transaction(self):
+        pass
+
+    @skip(NotImplementedError)
+    def test_successful_nested_read_atomic(self):
+        pass
+
+    @skip(NotImplementedError)
+    def test_unsuccessful_nested_read_atomic(self):
+        pass
+
+    @skip(NotImplementedError)
+    def test_successful_nested_write_atomic(self):
+        pass
+
+    @skip(NotImplementedError)
+    def test_unsuccessful_nested_write_atomic(self):
+        pass