added unit tests for [ticket:1024]

1 files changed, 14 insertions, 0 deletions

diff --git a/test/orm/transaction.py b/test/orm/transaction.py
index 24a2768c3..273a35222 100644
--- a/test/orm/transaction.py
+++ b/test/orm/transaction.py
@@ -153,6 +153,20 @@ class AutoExpireTest(TransactionTest):
 
         assert u1.name == 'will'
 
+class TwoPhaseTest(TransactionTest):
+    only_tables = True
+
+    @testing.requires.two_phase_transactions
+    def test_rollback_on_prepare(self):
+        s = self.session(twophase=True)
+    
+        u = User(name='ed')
+        s.add(u)
+        s.prepare()
+        s.rollback()
+        
+        assert u not in s
+        
 class RollbackRecoverTest(TransactionTest):
     only_tables = True