Skip to content
This repository was archived by the owner on Feb 21, 2023. It is now read-only.

Commit 0724304

Browse files
author
Stephan Meier
committed
add test for async callback
1 parent 56d6b32 commit 0724304

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/test_pubsub.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,9 @@ def setup_method(self, method):
270270
def message_handler(self, message):
271271
self.message = message
272272

273+
async def async_message_handler(self, message):
274+
self.message = message
275+
273276
async def test_published_message_to_channel(self, r):
274277
p = r.pubsub()
275278
await p.subscribe("foo")
@@ -311,6 +314,14 @@ async def test_channel_message_handler(self, r):
311314
assert await wait_for_message(p) is None
312315
assert self.message == make_message("message", "foo", "test message")
313316

317+
async def test_channel_async_message_handler(self, r):
318+
p = r.pubsub(ignore_subscribe_messages=True)
319+
await p.subscribe(foo=self.async_message_handler)
320+
assert await wait_for_message(p) is None
321+
assert await r.publish("foo", "test message") == 1
322+
assert await wait_for_message(p) is None
323+
assert self.message == make_message("message", "foo", "test message")
324+
314325
async def test_pattern_message_handler(self, r):
315326
p = r.pubsub(ignore_subscribe_messages=True)
316327
await p.psubscribe(**{"f*": self.message_handler})

0 commit comments

Comments
 (0)