def test_stopped(create_connection, server, loop):
sub = yield from create_connection(server.tcp_address, loop=loop)
pub = yield from create_connection(server.tcp_address, loop=loop)
mpsc = Receiver(loop=loop)
yield from sub.execute_pubsub('subscribe', mpsc.channel('channel:1'))
assert mpsc.is_active
mpsc.stop()
with pytest.logs('aioredis', 'DEBUG') as cm:
yield from pub.execute('publish', 'channel:1', b'Hello')
yield from asyncio.sleep(0, loop=loop)
assert len(cm.output) == 1
# Receiver must have 1 EndOfStream message
warn_messaege = (
"WARNING:aioredis:Pub/Sub listener message after stop: "
"sender: <_Sender name:b'channel:1', is_pattern:False, receiver:"
"<Receiver is_active:True, senders:1, qsize:1>>, data: b'Hello'")
assert cm.output == [warn_messaege]
assert (yield from mpsc.get()) is None
with pytest.raises(ChannelClosedError):
yield from mpsc.get()
res = yield from mpsc.wait_message()
assert res is False
def test_unsubscribe(create_connection, server, loop):
sub = yield from create_connection(server.tcp_address, loop=loop)
pub = yield from create_connection(server.tcp_address, loop=loop)
mpsc = Receiver(loop=loop)
yield from sub.execute_pubsub('subscribe',
mpsc.channel('channel:1'),
mpsc.channel('channel:3'))
res = yield from pub.execute("publish", "channel:3", "Hello world")
assert res == 1
res = yield from pub.execute("publish", "channel:1", "Hello world")
assert res == 1
assert mpsc.is_active
assert (yield from mpsc.wait_message()) is True
ch, msg = yield from mpsc.get()
assert ch.name == b'channel:3'
assert not ch.is_pattern
assert msg == b"Hello world"
assert (yield from mpsc.wait_message()) is True
ch, msg = yield from mpsc.get()
assert ch.name == b'channel:1'
assert not ch.is_pattern
assert msg == b"Hello world"
yield from sub.execute_pubsub('unsubscribe', 'channel:1')
assert mpsc.is_active
res = yield from pub.execute("publish", "channel:3", "message")
assert res == 1
assert (yield from mpsc.wait_message()) is True
ch, msg = yield from mpsc.get()
assert ch.name == b'channel:3'
assert not ch.is_pattern
assert msg == b"message"
yield from sub.execute_pubsub('unsubscribe', 'channel:3')
assert not mpsc.is_active
res = yield from mpsc.get()
assert res is None
async def test_wait_message(create_connection, server, loop):
sub = await create_connection(server.tcp_address, loop=loop)
pub = await create_connection(server.tcp_address, loop=loop)
mpsc = Receiver(loop=loop)
await sub.execute_pubsub('subscribe', mpsc.channel('channel:1'))
fut = asyncio.ensure_future(mpsc.wait_message(), loop=loop)
assert not fut.done()
await asyncio.sleep(0, loop=loop)
assert not fut.done()
await pub.execute('publish', 'channel:1', 'hello')
await asyncio.sleep(0, loop=loop) # read in connection
await asyncio.sleep(0, loop=loop) # call Future.set_result
assert fut.done()
res = await fut
assert res is True
async def test_wait_message(create_connection, server):
sub = await create_connection(server.tcp_address)
pub = await create_connection(server.tcp_address)
mpsc = Receiver()
await sub.execute_pubsub("subscribe", mpsc.channel("channel:1"))
fut = asyncio.ensure_future(mpsc.wait_message())
assert not fut.done()
await asyncio.sleep(0)
assert not fut.done()
await pub.execute("publish", "channel:1", "hello")
await asyncio.sleep(0) # read in connection
await asyncio.sleep(0) # call Future.set_result
assert fut.done()
res = await fut
assert res is True
def test_wait_message(create_connection, server, loop):
sub = yield from create_connection(server.tcp_address, loop=loop)
pub = yield from create_connection(server.tcp_address, loop=loop)
mpsc = Receiver(loop=loop)
yield from sub.execute_pubsub('subscribe', mpsc.channel('channel:1'))
fut = async_task(mpsc.wait_message(), loop=loop)
assert not fut.done()
yield from asyncio.sleep(0, loop=loop)
assert not fut.done()
yield from pub.execute('publish', 'channel:1', 'hello')
yield from asyncio.sleep(0, loop=loop) # read in connection
yield from asyncio.sleep(0, loop=loop) # call Future.set_result
assert fut.done()
res = yield from fut
assert res is True
async def test_wait_message(create_connection, server, loop):
sub = await create_connection(server.tcp_address, loop=loop)
pub = await create_connection(server.tcp_address, loop=loop)
mpsc = Receiver(loop=loop)
await sub.execute_pubsub('subscribe', mpsc.channel('channel:1'))
fut = asyncio.ensure_future(mpsc.wait_message(), loop=loop)
assert not fut.done()
await asyncio.sleep(0, loop=loop)
assert not fut.done()
await pub.execute('publish', 'channel:1', 'hello')
await asyncio.sleep(0, loop=loop) # read in connection
await asyncio.sleep(0, loop=loop) # call Future.set_result
assert fut.done()
res = await fut
assert res is True
