commit 922c41753ffbb4b526f2135a23c39f480c58e2cb
author bravof <fbravo@whitestack.com> Tue Nov 24 21:21:43 2020 -0300
committer bravof <fbravo@whitestack.com> Fri Nov 27 12:45:29 2020 -0300
tree 4c2847df8294a6587d8498bd85ebd30399ff2ae4
parent 43d82191273d8771c0da264065f8cfc0426b093d

feat(sol006): LCM migration to SOL006

Change-Id: I48a3ec2fec0b1e0173bb0f3aaa83a774d46b3955
Signed-off-by: bravof <fbravo@whitestack.com>

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
--- 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 @@
     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):