diff options
Diffstat (limited to 'tests/test_async.py')
-rw-r--r-- | tests/test_async.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/tests/test_async.py b/tests/test_async.py new file mode 100644 index 0000000..fc01838 --- /dev/null +++ b/tests/test_async.py @@ -0,0 +1,44 @@ +import asyncio + +import blinker +from blinker._async import send_async + + +def test_send_async(): + calls = [] + + @asyncio.coroutine + def receiver_a(sender): + calls.append(receiver_a) + return 'value a' + + @asyncio.coroutine + def receiver_b(sender): + calls.append(receiver_b) + return 'value b' + + def receiver_c(sender): + calls.append(receiver_c) + return 'value c' + + sig = blinker.Signal(None, send_async()) + sig.connect(receiver_a) + sig.connect(receiver_b) + sig.connect(receiver_c) + + @asyncio.coroutine + def collect(): + return sig.send() + + loop = asyncio.get_event_loop() + results = loop.run_until_complete(collect()) + + expected = { + receiver_a: 'value a', + receiver_b: 'value b', + receiver_c: 'value c', + } + + assert set(calls) == set(expected.keys()) + collected_results = {v.result() for r, v in results} + assert collected_results == set(expected.values()) |