class TestConsumer(unittest.TestCase):
def setUp(self):
broker_address = cfg.BROKER_ADDRESS
broker_exchange_name = cfg.EXCHANGE_NAME
broker_queue_name = cfg.QUEUE_NAME
broker_client_obj = BrokerClient(broker_address, broker_exchange_name, broker_queue_name)
self._consumer = Consumer(broker_client_obj)
def test_connect_to_broker(self):
try:
loop = asyncio.get_event_loop()
loop.run_until_complete(self._consumer.connect_to_broker())
self.assertTrue(True)
except ConnectionError:
self.assertTrue(False)
def test_consume(self):
try:
loop = asyncio.get_event_loop()
loop.run_until_complete(self._consumer.connect_to_broker())
loop.run_until_complete(self._consumer.consume(pv_simulator.process_message))
self.assertTrue(True)
except ConnectionError:
self.assertTrue(False)
def test_disconnect_from_broker(self):
try:
loop = asyncio.get_event_loop()
loop.run_until_complete(self._consumer.connect_to_broker())
loop.run_until_complete(self._consumer.disconnect_from_broker())
self.assertTrue(True)
except ConnectionError:
self.assertTrue(False)
