18 # with CubicWeb. If not, see <http://www.gnu.org/licenses/>. |
18 # with CubicWeb. If not, see <http://www.gnu.org/licenses/>. |
19 |
19 |
20 from cubicweb import ValidationError |
20 from cubicweb import ValidationError |
21 from cubicweb.devtools.testlib import CubicWebTC |
21 from cubicweb.devtools.testlib import CubicWebTC |
22 import cubicweb.server.session |
22 import cubicweb.server.session |
23 from cubicweb.server.session import Transaction as OldTransaction |
23 from cubicweb.server.session import Connection as OldConnection |
24 from cubicweb.transaction import * |
24 from cubicweb.transaction import * |
25 |
25 |
26 from cubicweb.server.sources.native import UndoTransactionException, _UndoException |
26 from cubicweb.server.sources.native import UndoTransactionException, _UndoException |
27 |
27 |
28 |
28 |
33 self.toto = self.create_user(req, 'toto', password='toto', groups=('users',), |
33 self.toto = self.create_user(req, 'toto', password='toto', groups=('users',), |
34 commit=False) |
34 commit=False) |
35 self.txuuid = self.commit() |
35 self.txuuid = self.commit() |
36 |
36 |
37 def setUp(self): |
37 def setUp(self): |
38 class Transaction(OldTransaction): |
38 class Connection(OldConnection): |
39 """Force undo feature to be turned on in all case""" |
39 """Force undo feature to be turned on in all case""" |
40 undo_actions = property(lambda tx: True, lambda x, y:None) |
40 undo_actions = property(lambda tx: True, lambda x, y:None) |
41 cubicweb.server.session.Transaction = Transaction |
41 cubicweb.server.session.Connection = Connection |
42 super(UndoableTransactionTC, self).setUp() |
42 super(UndoableTransactionTC, self).setUp() |
43 |
43 |
44 def tearDown(self): |
44 def tearDown(self): |
45 cubicweb.server.session.Transaction = OldTransaction |
45 cubicweb.server.session.Connection = OldConnection |
46 self.restore_connection() |
46 self.restore_connection() |
47 self.session.undo_support = set() |
47 self.session.undo_support = set() |
48 super(UndoableTransactionTC, self).tearDown() |
48 super(UndoableTransactionTC, self).tearDown() |
49 |
49 |
50 def check_transaction_deleted(self, txuuid): |
50 def check_transaction_deleted(self, txuuid): |