summaryrefslogtreecommitdiff
path: root/test/testbase.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/testbase.py')
-rw-r--r--test/testbase.py11
1 files changed, 7 insertions, 4 deletions
diff --git a/test/testbase.py b/test/testbase.py
index 7d981509d..bc5153af0 100644
--- a/test/testbase.py
+++ b/test/testbase.py
@@ -205,6 +205,8 @@ class AssertMixin(PersistTest):
self.assert_(db.sql_count == count, "desired statement count %d does not match %d" % (count, db.sql_count))
class ORMTest(AssertMixin):
+ keep_mappers = False
+ keep_data = False
def setUpAll(self):
global metadata
metadata = BoundMetaData(db)
@@ -217,10 +219,11 @@ class ORMTest(AssertMixin):
def tearDownAll(self):
metadata.drop_all()
def tearDown(self):
- clear_mappers()
- for t in metadata.table_iterator(reverse=True):
- t.delete().execute().close()
-
+ if not self.keep_mappers:
+ clear_mappers()
+ if not self.keep_data:
+ for t in metadata.table_iterator(reverse=True):
+ t.delete().execute().close()
class EngineAssert(proxy.BaseProxyEngine):
"""decorates a SQLEngine object to match the incoming queries against a set of assertions."""