async def test_pubsub():
pubsub = MemoryPubSub()
subscriber = await pubsub.subscribe("a_chan")
await pubsub.publish("a_chan", "hello world!")
await pubsub.publish("a_chan", "hello universe!")
assert subscriber.messages.qsize() == 2, subscriber.messages.qsize()
assert await subscriber.__anext__() == "hello world!"
assert await subscriber.__anext__() == "hello universe!"
async def test_broadcast():
pubsub = MemoryPubSub()
subscriber_1 = await pubsub.subscribe("a_chan")
subscriber_2 = await pubsub.subscribe("a_chan")
await pubsub.publish("a_chan", "hello world!")
subscriber_1 = subscriber_1.__aiter__()
subscriber_2 = subscriber_2.__aiter__()
assert await subscriber_1.__anext__() == "hello world!"
assert await subscriber_2.__anext__() == "hello world!"
async def test_not_subscribed_chan():
pubsub = MemoryPubSub()
subscriber_a_chan = await pubsub.subscribe("a_chan")
subscriber_c_chan = await pubsub.subscribe("c_chan")
await pubsub.publish("a_chan", "hello world!")
await pubsub.publish("b_chan", "junk message")
await pubsub.publish("c_chan", "hello universe!")
assert await subscriber_a_chan.__anext__() == "hello world!"
assert await subscriber_c_chan.__anext__() == "hello universe!"
async def ttest_pubsub():
pubsub = MemoryPubSub()
# Create subscriber
# Push message
await pubsub.publish("a_chan", "hello world!")
subscriber = await pubsub.subscribe("a_chan")
await pubsub.publish("a_chan", "hello universe!")
# And listening channel
try:
async for message in subscriber:
print(message, flush=True)
except KeyboardInterrupt:
print("Finish listening")
async def test_iteration_protocol():
pubsub = MemoryPubSub()
subscriber = await pubsub.subscribe("a_chan")
await pubsub.publish("a_chan", "hello world!")
subscriber = subscriber.__aiter__()
assert await subscriber.__anext__() == "hello world!"
async def test_dispose_subscriber():
pubsub = MemoryPubSub()
subscriber = await pubsub.subscribe("a_chan")
assert await pubsub.publish("a_chan", "hello world!") == 1
del subscriber
assert await pubsub.publish("a_chan", "hello world!") == 0