async def test_start_consuming_for_commands(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') qc = AsyncQueueConsumer(app) # mocks service_channel = Mock(channel_id=1) qc.make_channel = Mock(side_effect=coroutine(lambda _type: service_channel)) qc.channels[service_channel.channel_id] = { "mute": False, "type": "services", "channel": service_channel } qc.create_queue_and_consume = Mock(side_effect=coroutine( lambda channel, on_message_method, queue_name: ("queue_name", "consumer_tag") )) queue_name = await qc.start_consuming_for_commands() assert qc.create_queue_and_consume.called assert queue_name == "queue_name" assert qc.channels[service_channel.channel_id]["consumer_tag"] == "consumer_tag"
async def test_start_consuming_for_platforms(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') app.platforms = {"ubuntu-14.04-x64": 1} qc = AsyncQueueConsumer(app) # mocks platform_channel = Mock(channel_id=1) qc.make_channel = Mock(side_effect=coroutine(lambda _type: platform_channel)) qc.channels[platform_channel.channel_id] = { "mute": False, "type": "platforms", "channel": platform_channel } qc.create_queue_and_consume = Mock(side_effect=coroutine( lambda channel, on_message_method, queue_name: ("queue_name", "consumer_tag") )) await qc.start_consuming_for_platforms() assert qc.create_queue_and_consume.called assert qc.channels[platform_channel.channel_id]["consumer_tag"] == "consumer_tag" assert qc.channels[platform_channel.channel_id]["platform"] == "ubuntu-14.04-x64"