tests: Add unit tests for send() and throw() on generator wrapper

This commit is contained in:
rdb 2025-08-28 12:31:51 +02:00
parent 2625c19ccf
commit 469df73ce7

View File

@ -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()