Changeset 5277

Timestamp:

08/27/08 23:00:40 (3 months ago)

Author:

kskuhlman

Message:

Adding DBTest support for SQLAlchemy. Closes #1764.

Files:

• branches/1.1/turbogears/testutil.py (modified) (3 diffs)

branches/1.1/turbogears/testutil.py

@@ -289 +289 @@
-
-
-
+Abstract
-    """A database enabled unit testing class.
-
@@ -298 +298 @@
-    model = None
-
+    def setUp(self):
+        raise NotImplementedError()
+
+    def tearDown(self):
+        raise NotImplementedError()
+
+class DBTestSO(AbstractDBTest):
-    def _get_soClasses(self):
-        try:
@@ -323 +330 @@
-                item.dropTable(ifExists=True, cascade=True)
-
+class DBTestSA(AbstractDBTest):
+    def setUp(self):
+        database.get_engine()
+        database.metadata.create_all()
+
+    def tearDown(self):
+        database.metadata.drop_all()
+
+
+#Determine which class to use for "DBTest".  Setup & teardown should behave 
+#simularly regardless of which ORM you choose.
+if config.get("sqlobject.dburi"):
+    DBTest = DBTestSO
+elif config.get("sqlalchemy.dburi"):
+    DBTest = DBTestSA
+else:
+    raise Exception("Unable to find sqlalchemy or sqlobject dburi")
+
+
-def unmount():
-    """Remove an application from the object traversal tree."""