X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_lcm%2Ftests%2Ftest_prometheus.py;h=72785dfc5dab02f267f018eca587a11eb20e7708;hb=021e70d2eb5caa25a7e731ed70f2a008cf8057d7;hp=064ede830a2929ad3329a61a375178aeb10d9dcd;hpb=b996d94c0f74b1922d7f93a51ec328c6f370ff86;p=osm%2FLCM.git

diff --git a/osm_lcm/tests/test_prometheus.py b/osm_lcm/tests/test_prometheus.py
index 064ede8..72785df 100644
--- a/osm_lcm/tests/test_prometheus.py
+++ b/osm_lcm/tests/test_prometheus.py
@@ -18,7 +18,7 @@
 import asynctest
 from osm_lcm.prometheus import Prometheus, initial_prometheus_data
 from asynctest.mock import Mock
-from osm_common.dbmemory import DbMemory
+from osm_lcm.data_utils.database.database import Database
 
 __author__ = 'Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>'
 
@@ -28,8 +28,16 @@ class TestPrometheus(asynctest.TestCase):
     async def setUp(self):
         config = {'uri': 'http:prometheus:9090',
                   'path': '/etc/prometheus'}
-        self.db = Mock(DbMemory())
-        self.p = Prometheus(config, worker_id='1', db=self.db, loop=self.loop)
+        # Cleanup singleton Database instance
+        Database.instance = None
+
+        self.db = Mock(Database({
+            "database": {
+                "driver": "memory"
+            }
+        }).instance.db)
+        Database().instance.db = self.db
+        self.p = Prometheus(config, worker_id='1', loop=self.loop)
 
     @asynctest.fail_on(active_handles=True)
     async def test_start(self):