[Checkins] SVN: z3c.zalchemy/branches/jukart-SA-0.2/src/z3c/zalchemy/testing/__init__.py Allow registration of tables to be dropped at tearDown

[Jürgen Kartnaller]

Log message for revision 68134:
  Allow registration of tables to be dropped at tearDown

Changed:
  U   z3c.zalchemy/branches/jukart-SA-0.2/src/z3c/zalchemy/testing/__init__.py

-=-
Modified: z3c.zalchemy/branches/jukart-SA-0.2/src/z3c/zalchemy/testing/__init__.py
===================================================================
--- z3c.zalchemy/branches/jukart-SA-0.2/src/z3c/zalchemy/testing/__init__.py	2006-05-15 13:35:07 UTC (rev 68133)
+++ z3c.zalchemy/branches/jukart-SA-0.2/src/z3c/zalchemy/testing/__init__.py	2006-05-15 13:35:46 UTC (rev 68134)
@@ -29,6 +29,12 @@
             transaction.get().commit()
         except:
             pass
+    if _tablesToDrop:
+        session = z3c.zalchemy.getSession()
+        for table, engine in _tablesToDrop:
+            z3c.zalchemy.datamanager.dropTable(table, engine)
+        del _tablesToDrop[:]
+        transaction.get().commit()
     z3c.zalchemy.datamanager._tableToEngine.clear()
     z3c.zalchemy.datamanager._classToEngine.clear()
 
@@ -47,4 +53,12 @@
     tearDown(test)
     setup.placefulTearDown()
     shutil.rmtree(test.tmpDir)
-    
+
+_tablesToDrop = []
+
+def dropTable(name, engine=''):
+    """Drop table at tearDown.
+    """
+    _tablesToDrop.append((name, engine))
+
+