summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/orm/test_transaction.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/test/orm/test_transaction.py b/test/orm/test_transaction.py
index 396ed4e54..b8b8d8049 100644
--- a/test/orm/test_transaction.py
+++ b/test/orm/test_transaction.py
@@ -280,6 +280,7 @@ class SavepointTest(TransactionTest):
@testing.requires.savepoints
def test_savepoint_rollback(self):
+ User = self.classes.User
s = self.session()
u1 = User(name='ed')
u2 = User(name='jack')
@@ -303,6 +304,7 @@ class SavepointTest(TransactionTest):
@testing.requires.savepoints
def test_savepoint_delete(self):
+ User = self.classes.User
s = self.session()
u1 = User(name='ed')
s.add(u1)
@@ -316,6 +318,7 @@ class SavepointTest(TransactionTest):
@testing.requires.savepoints
def test_savepoint_commit(self):
+ User = self.classes.User
s = self.session()
u1 = User(name='ed')
u2 = User(name='jack')
@@ -406,6 +409,7 @@ class SavepointTest(TransactionTest):
@testing.requires.savepoints
def test_expunge_pending_on_rollback(self):
+ User = self.classes.User
sess = self.session()
sess.begin_nested()
@@ -417,6 +421,7 @@ class SavepointTest(TransactionTest):
@testing.requires.savepoints
def test_update_deleted_on_rollback(self):
+ User = self.classes.User
s = self.session()
u1 = User(name='ed')
s.add(u1)