Refactors codebase

[osm/MON.git] / osm_mon / test / core / test_common_consumer.py

diff --git a/osm_mon/test/core/test_common_consumer.py b/osm_mon/test/core/test_common_consumer.py
index 0306893..eaf06c1 100644 (file)
--- a/osm_mon/test/core/test_common_consumer.py
+++ b/osm_mon/test/core/test_common_consumer.py
@@ -5,8 +5,8 @@ from kafka import KafkaProducer
 from kafka.errors import KafkaError
 from osm_common import dbmongo
 
-from osm_mon.core.database import VimCredentials
-from osm_mon.core.message_bus.common_consumer import *
+from osm_mon.core.database import VimCredentials, DatabaseManager
+from osm_mon.core.message_bus.common_consumer import CommonConsumer
 
 
 @mock.patch.object(dbmongo.DbMongo, "db_connect", mock.Mock())