Add test_suite function.

diff --git a/buildfarm/tests/__init__.py b/buildfarm/tests/__init__.py
index 7d9e4041178e8117a8a833003012c3e09eb30346..964e750ba0e680575ec534c66508711c54b78f9e 100644 (file)
--- a/buildfarm/tests/__init__.py
+++ b/buildfarm/tests/__init__.py
@@ -23,6 +23,7 @@ from storm.store import Store
 from testtools import TestCase
 import shutil
 import tempfile
+import unittest
 
 
 class BuildFarmTestCase(TestCase):
@@ -95,11 +96,29 @@ class BuildFarmTestCase(TestCase):
         for subdir in ["data", "data/upload", "data/oldrevs", "db", "web", "lcov", "lcov/data"]:
             os.mkdir(os.path.join(self.path, subdir))
 
-        db = database.create_database("sqlite:"+os.path.join(self.path, "db", "hostdb.sqlite"))
+        self.db_url = "sqlite:"+os.path.join(self.path, "db", "hostdb.sqlite")
+        db = database.create_database(self.db_url)
         store = Store(db)
         setup_schema(store)
-        store.flush()
+        store.commit()
 
     def tearDown(self):
         shutil.rmtree(self.path)
         super(BuildFarmTestCase, self).tearDown()
+
+
+def test_suite():
+    names = [
+        '__init__',
+        'test_build',
+        'test_history',
+        'test_hostdb',
+        'test_sqldb',
+        'test_util',
+        ]
+    module_names = ['buildfarm.tests.' + name for name in names]
+    result = unittest.TestSuite()
+    loader = unittest.TestLoader()
+    suite = loader.loadTestsFromNames(module_names)
+    result.addTests(suite)
+    return result