async def test_disconnect(loop):
app = BaseApplication('app', loop=loop, CONFIG='config.tests')
app.platforms = {"ubuntu-14.04-x64": 1}
qc = AsyncQueueConsumer(app)
# mocks
qc.connection = Mock(close=Mock(side_effect=coroutine(lambda: None)))
qc.transport = Mock(close=Mock(side_effect=coroutine(lambda: None)))
await qc.disconnect()
assert qc.connection.close.called
assert qc.transport.close.called
async def test_make_channel(loop):
app = BaseApplication('app', loop=loop, CONFIG='config.tests')
qc = AsyncQueueConsumer(app)
# mocks
qc.connection = Mock()
qc.connection.channel = Mock(side_effect=coroutine(
lambda: Mock(channel_id=1)
))
await qc.make_channel("sessions")
assert qc.connection.channel.called
assert qc.channels[1]["type"] == "sessions"
