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