Ticket #1764: TurboGears-1.5-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 turbogears.config.server.get("sqlalchemy.dburi"):
+            self._sa_setUp()
+        elif turbogears.config.server.get("sqlobject.dburi"):
+            self._so_setUp()
+        else:
+            raise Exception("Unable to find sqlalchemy or sqlobject dburi")
+
+    def tearDown(self):
+        if turbogears.config.server.get("sqlalchemy.dburi"):
+            self._sa_tearDown()
+        elif turbogears.config.server.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()
 
-    def setUp(self):
+    def _so_setUp(self):
         if not self.model:
             self.model = get_model()
             if not self.model:
-                raise "Unable to run database tests without a model"
-
+                raise Exception("Unable to run database tests without a 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)
 
-    def tearDown(self):
+    def _so_tearDown(self):
         database.rollback_all()
         for item in reversed(self._get_soClasses()):
             if isinstance(item, types.TypeType) and issubclass(item,