Replaces direct use of aiokafka with osm_common message bus in agent and

[osm/POL.git] / osm_policy_module / tests / integration / test_kafka_messages.py

diff --git a/osm_policy_module/tests/integration/test_kafka_messages.py b/osm_policy_module/tests/integration/test_kafka_messages.py
index 29d88e2..28b2e0f 100644 (file)
--- a/osm_policy_module/tests/integration/test_kafka_messages.py
+++ b/osm_policy_module/tests/integration/test_kafka_messages.py
@@ -42,11 +42,10 @@ log.addHandler(stream_handler)
 class KafkaMessagesTest(unittest.TestCase):
     def setUp(self):
         super()
-        cfg = Config.instance()
-        self.kafka_server = '{}:{}'.format(cfg.OSMPOL_MESSAGE_HOST,
-                                           cfg.OSMPOL_MESSAGE_PORT)
+        cfg = Config()
+        self.kafka_server = '{}:{}'.format(cfg.get('message', 'host'),
+                                           cfg.get('message', 'port'))
         self.loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(None)
 
     def tearDown(self):
         super()