Fix bug 1442: set SDN target in vim_info for NS VLD

[osm/LCM.git] / osm_lcm / tests / test_prometheus.py

diff --git a/osm_lcm/tests/test_prometheus.py b/osm_lcm/tests/test_prometheus.py
index 064ede8..72785df 100644 (file)
--- 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):