Merge branch 'fix/testsuite/db-tests' of https://github.com/AlexanderS/bcfg2

4 files changed, 37 insertions, 24 deletions

diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testhelpers.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testhelpers.py
index 5a82100d0..9f6a9f320 100644
--- a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testhelpers.py
+++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testhelpers.py
@@ -51,6 +51,7 @@ class TestFunctions(Bcfg2TestCase):
 
 class TestDatabaseBacked(TestPlugin):
     test_obj = DatabaseBacked
+    synced = False
 
     def setUp(self):
         TestPlugin.setUp(self)
@@ -76,6 +77,15 @@ class TestDatabaseBacked(TestPlugin):
         setattr(Bcfg2.Options.setup, attr, True)
         self.assertRaises(PluginInitError, self.get_obj, core)
 
+    def syncdb(self, modeltest):
+        """ Given an instance of a :class:`DBModelTestCase` object, sync
+        and clean the database """
+        inst = modeltest(methodName='test_syncdb')
+        if not self.__class__.synced:
+            inst.test_syncdb()
+            self.__class__.synced = True
+        inst.test_cleandb()
+
 
 class TestPluginDatabaseModel(Bcfg2TestCase):
     """ placeholder for future tests """
diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestMetadata.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestMetadata.py
index f2721c9ea..5d7ed50b7 100644
--- a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestMetadata.py
+++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestMetadata.py
@@ -123,7 +123,7 @@ class TestClientVersions(TestDatabaseBacked):
     def setUp(self):
         TestDatabaseBacked.setUp(self)
         self.test_obj = ClientVersions
-        syncdb(TestMetadataDB)
+        self.syncdb(TestMetadataDB)
         for client, version in self.test_clients.items():
             MetadataClientModel(hostname=client, version=version).save()
 
@@ -1252,7 +1252,7 @@ class TestMetadataBase(TestMetadata):
         TestClientRunHooks.setUp(self)
         TestDatabaseBacked.setUp(self)
         Bcfg2.Options.setup.metadata_db = True
-        syncdb(TestMetadataDB)
+        self.syncdb(TestMetadataDB)
 
     def load_clients_data(self, metadata=None, xdata=None):
         if metadata is None:
diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py
index 32766b5c1..9729a0449 100644
--- a/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py
+++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestProbes.py
@@ -201,7 +201,7 @@ group-specific"""
                 assert False, "Strange probe found in get_probe_data() return"
 
 
-class TestProbes(TestPlugin):
+class TestProbes(TestDatabaseBacked):
     test_obj = Probes
 
     test_xdata = lxml.etree.Element("test")
@@ -241,7 +241,7 @@ group:      group:with:colons
             self.datastore = None
             Bcfg2.Options.setup.repository = datastore
 
-    def get_obj(self):
+    def get_obj(self, core=None):
         if not Bcfg2.Options.setup.probes_db:
             # actually use a real datastore so we can read and write
             # probed.xml
@@ -251,7 +251,7 @@ group:      group:with:colons
                 datadir = os.path.join(self.datastore, self.test_obj.name)
                 if not os.path.exists(datadir):
                     os.makedirs(datadir)
-        return TestPlugin.get_obj(self)
+        return TestPlugin.get_obj(self, core)
 
     def test__init(self):
         if Bcfg2.Options.setup.probes_db:
@@ -278,7 +278,7 @@ group:      group:with:colons
     def test_probes_db(self):
         """ Set and retrieve probe data with database enabled """
         Bcfg2.Options.setup.probes_db = True
-        syncdb(TestProbesDB)
+        self.syncdb(TestProbesDB)
         self._perform_tests()
 
     def test_allowed_cgroups(self):
diff --git a/testsuite/common.py b/testsuite/common.py
index 9db2cb94a..944471ade 100644
--- a/testsuite/common.py
+++ b/testsuite/common.py
@@ -223,18 +223,29 @@ class DBModelTestCase(Bcfg2TestCase):
             import django.core.management
             from django.core.exceptions import ImproperlyConfigured
 
-            if django.VERSION[0] == 1 and django.VERSION[1] < 7:
-                try:
-                    django.core.management.call_command('syncdb', interactive=False,
-                                                        verbosity=0)
-                except ImproperlyConfigured:
-                    pass
+            dbfile = django.conf.settings.DATABASES['default']['NAME']
 
+            # Close all connections to the old database
+            if django.VERSION[0] == 1 and django.VERSION[1] >= 7:
+                for connection in django.db.connections.all():
+                    connection.close()
+            else:
+                django.db.close_connection()
+            
+            # Remove old database
+            if os.path.exists(dbfile):
+                os.unlink(dbfile)
+            self.assertFalse(os.path.exists(dbfile))
+
+            # Create new
+            if django.VERSION[0] == 1 and django.VERSION[1] < 7:
+                django.core.management.call_command('syncdb', interactive=False,
+                                                    verbosity=1)
             django.core.management.call_command('migrate', interactive=False,
-                                                verbosity=0)
-            self.assertTrue(
-                os.path.exists(
-                    django.conf.settings.DATABASES['default']['NAME']))
+                                                verbosity=1)
+
+            # Check if database exists now
+            self.assertTrue(os.path.exists(dbfile))
 
     @skipUnless(has_django, "Django not found, skipping")
     def test_cleandb(self):
@@ -245,14 +256,6 @@ class DBModelTestCase(Bcfg2TestCase):
             self.assertItemsEqual(list(model.objects.all()), [])
 
 
-def syncdb(modeltest):
-    """ Given an instance of a :class:`DBModelTestCase` object, sync
-    and clean the database """
-    inst = modeltest(methodName='test_syncdb')
-    inst.test_syncdb()
-    inst.test_cleandb()
-
-
 # in order for patchIf() to decorate a function in the same way as
 # patch(), we override the default behavior of __enter__ and __exit__
 # on the _patch() object to basically be noops.