forked from goodboy/tractor
1
0
Fork 0

Add trio-cancels-anursery-cancels-aio test

infect_asyncio
Tyler Goodlet 2021-11-18 09:35:59 -05:00
parent 8704664719
commit c19123b588
1 changed files with 25 additions and 4 deletions

View File

@ -96,6 +96,31 @@ def test_tractor_cancels_aio(arb_addr):
trio.run(main) trio.run(main)
def test_trio_cancels_aio(arb_addr):
'''
Much like the above test with ``tractor.Portal.cancel_actor()``
except we just use a standard ``trio`` cancellation api.
'''
async def main():
with trio.move_on_after(1):
# cancel the nursery shortly after boot
async with tractor.open_nursery() as n:
# debug_mode=True
# ) as n:
portal = await n.run_in_actor(
asyncio_actor,
target='sleep_forever',
expect_err='trio.Cancelled',
infect_asyncio=True,
)
trio.run(main)
async def aio_cancel(): async def aio_cancel():
''''Cancel urself boi. ''''Cancel urself boi.
@ -126,10 +151,6 @@ def test_aio_cancelled_from_aio_causes_trio_cancelled(arb_addr):
trio.run(main) trio.run(main)
def test_trio_cancels_aio(arb_addr):
...
def test_trio_error_cancels_aio(arb_addr): def test_trio_error_cancels_aio(arb_addr):
... ...