From 3398153c5224991c43f28702dd09bfadb6ef23f6 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Tue, 11 Oct 2022 13:35:55 -0400 Subject: [PATCH] Add timeout around `trio`-callee-task --- tests/test_infected_asyncio.py | 31 ++++++++++++++++--------------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/tests/test_infected_asyncio.py b/tests/test_infected_asyncio.py index 976741d..30ddfb6 100644 --- a/tests/test_infected_asyncio.py +++ b/tests/test_infected_asyncio.py @@ -203,24 +203,25 @@ def test_context_spawns_aio_task_that_errors( ''' async def main(): - async with tractor.open_nursery() as n: - p = await n.start_actor( - 'aio_daemon', - enable_modules=[__name__], - infect_asyncio=True, - # debug_mode=True, - loglevel='cancel', - ) - async with p.open_context( - trio_ctx, - ) as (ctx, first): + with trio.fail_after(2): + async with tractor.open_nursery() as n: + p = await n.start_actor( + 'aio_daemon', + enable_modules=[__name__], + infect_asyncio=True, + # debug_mode=True, + loglevel='cancel', + ) + async with p.open_context( + trio_ctx, + ) as (ctx, first): - assert first == 'start' + assert first == 'start' - if parent_cancels: - await p.cancel_actor() + if parent_cancels: + await p.cancel_actor() - await trio.sleep_forever() + await trio.sleep_forever() with pytest.raises(RemoteActorError) as excinfo: trio.run(main)