devtools/testlib.py
changeset 4913 083b4d454192
parent 4835 13b0b96d7982
child 4917 756ded5bc9eb
--- a/devtools/testlib.py	Wed Mar 10 16:07:24 2010 +0100
+++ b/devtools/testlib.py	Mon Mar 01 11:26:14 2010 +0100
@@ -321,7 +321,10 @@
 
     @nocoverage
     def commit(self):
-        self.cnx.commit()
+        try:
+            return self.cnx.commit()
+        finally:
+            self.session.set_pool() # ensure pool still set after commit
 
     @nocoverage
     def rollback(self):
@@ -329,6 +332,8 @@
             self.cnx.rollback()
         except ProgrammingError:
             pass
+        finally:
+            self.session.set_pool() # ensure pool still set after commit
 
     # # server side db api #######################################################