From 469df73ce741e7559abb96ce2f897a6999370698 Mon Sep 17 00:00:00 2001 From: rdb Date: Thu, 28 Aug 2025 12:31:51 +0200 Subject: [PATCH] tests: Add unit tests for send() and throw() on generator wrapper --- tests/event/test_futures.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/event/test_futures.py b/tests/event/test_futures.py index ee6b0f2894..4aa11715e4 100644 --- a/tests/event/test_futures.py +++ b/tests/event/test_futures.py @@ -46,6 +46,31 @@ def check_result(fut, expected): return True +def test_future_await_send(): + fut = core.AsyncFuture() + + i = fut.__await__() + assert i.send(None) == fut + assert i.send(None) == fut + assert i.send(None) == fut + assert i.send(None) == fut + + fut.set_result(123) + + with pytest.raises(StopIteration) as e: + i.send(None) + + assert e.value.value == 123 + + +def test_future_await_throw(): + fut = core.AsyncFuture() + + i = fut.__await__() + with pytest.raises(RuntimeError): + i.throw(RuntimeError) + + def test_future_cancelled(): fut = core.AsyncFuture()