- added a standardized test harness for ORM tests

- added three-level mapping test.  needed some massaging for postgres

1 files changed, 20 insertions, 1 deletions

diff --git a/test/testbase.py b/test/testbase.py
index e96b30dcb..7d981509d 100644
--- a/test/testbase.py
+++ b/test/testbase.py
@@ -10,7 +10,8 @@ import sqlalchemy.pool as pool
 import re
 import sqlalchemy
 import optparse
-
+from sqlalchemy.schema import BoundMetaData
+from sqlalchemy.orm import clear_mappers
 
 db = None
 metadata = None
@@ -203,6 +204,24 @@ class AssertMixin(PersistTest):
         finally:
             self.assert_(db.sql_count == count, "desired statement count %d does not match %d" % (count, db.sql_count))
 
+class ORMTest(AssertMixin):
+    def setUpAll(self):
+        global metadata
+        metadata = BoundMetaData(db)
+        self.define_tables(metadata)
+        metadata.create_all()
+    def define_tables(self, metadata):
+        raise NotImplementedError()
+    def get_metadata(self):
+        return metadata
+    def tearDownAll(self):
+        metadata.drop_all()
+    def tearDown(self):
+        clear_mappers()
+        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."""
     def __init__(self, engine):