Ticket #1764: TurboGears-1.1-DBTest-SQLAlchemy.patch

turbogears/testutil.py

@@ -165 +165 @@
-
-
-class DBTest(unittest.TestCase):
+    """A database enabled unit testing class.
-
+    Creates and destroys your database before and after each unit test,
+    based on the configuration defined in test.cfg.  This class supports
+    both SQLAlchemy and SQLObject configurations.
+
+    """
-    model = None
-
+    def setUp(self):
+        if config.get("sqlalchemy.dburi"):
+            self._sa_setUp()
+        elif config.get("sqlobject.dburi"):
+            self._so_setUp()
+        else:
+            raise Exception("Unable to find sqlalchemy or sqlobject dburi")
+
+    def tearDown(self):
+        if config.get("sqlalchemy.dburi"):
+            self._sa_tearDown()
+        elif config.get("sqlobject.dburi"):
+            self._so_tearDown()
+
+    ## SQLAlchemy specific methods
+
+    def _sa_setUp(self):
+        database.get_engine()
+        database.metadata.create_all()
+
+    def _sa_tearDown(self):
+        database.metadata.drop_all()
+
+    ## SQLObject specific methods
+
-    def _get_soClasses(self):
-        try:
-            return [self.model.__dict__[x] for x in self.model.soClasses]
-        except AttributeError:
-            return self.model.__dict__.values()
-
-
+_so_
-        if not self.model:
-            self.model = get_model()
-            if not self.model:
-
-
+
-        for item in self._get_soClasses():
-            if isinstance(item, types.TypeType) and issubclass(item,
-                sqlobject.SQLObject) and item != sqlobject.SQLObject \
-                and item != InheritableSQLObject:
-                item.createTable(ifNotExists=True)
-
-
+_so_
-        database.rollback_all()
-        for item in reversed(self._get_soClasses()):
-            if isinstance(item, types.TypeType) and issubclass(item,