async def actual():
# Wait for other client to wake up before publishing to it
CLIENT_START_SYNC.wait(5)
# Create a client and subscribe to topics
client = PubSubClient()
client.start_client(uri)
# wait for the client to be ready
await client.wait_until_ready()
# publish event
logger.info("Publishing event")
published = await client.publish([EVENT_TOPIC], data=DATA)
assert published.result == True
async def main():
# Create a client and subscribe to topics
client = PubSubClient(["guns", "germs"], callback=on_events)
async def on_steel(data, topic):
print("running callback steel!")
print("Got data", data)
asyncio.create_task(client.disconnect())
client.subscribe("steel", on_steel)
client.start_client(f"ws://localhost:{PORT}/pubsub")
await client.wait_until_done()
async def main():
# Create a client and subscribe to topics
client = PubSubClient(["ALL_WEBHOOKS"], callback=on_events)
"""
# You can also register it using the commented code below
async def on_data(data, topic):
print(f"{topic}:\n", data)
[client.subscribe(topic, on_data) for topic in topics]
"""
client.start_client(f"ws://{URL}:{3010}/pubsub")
print("Started")
await client.wait_until_done()
async def test_disconnect_callback_without_context(delayed_death_server):
"""
Test reconnecting when a server hangups AFTER connect and that the disconnect callback work
"""
# finish trigger
finish = asyncio.Event()
disconnected = asyncio.Event()
async def on_disconnect(channel):
disconnected.set()
async def on_connect(client, channel):
try:
print("Connected")
# publish events (with sync=False to avoid deadlocks waiting on the publish to ourselves)
published = await client.publish([EVENT_TOPIC],
data=DATA,
sync=False,
notifier_id=gen_uid())
assert published.result == True
except RpcChannelClosedException:
# expected
pass
# Create a client and subscribe to topics
client = PubSubClient(on_disconnect=[on_disconnect],
on_connect=[on_connect])
async def on_event(data, topic):
assert data == DATA
finish.set()
# subscribe for the event
client.subscribe(EVENT_TOPIC, on_event)
# start listentining
client.start_client(uri)
# wait for the client to be ready to receive events
await client.wait_until_ready()
# publish events (with sync=False toa void deadlocks waiting on the publish to ourselves)
published = await client.publish([EVENT_TOPIC],
data=DATA,
sync=False,
notifier_id=gen_uid())
assert published.result == True
# wait for finish trigger
await asyncio.wait_for(finish.wait(), 5)
await client.disconnect()
await asyncio.wait_for(disconnected.wait(), 1)
assert disconnected.is_set()
