From 340effae110d6857134ce016703e278d3b4091c2 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Thu, 10 Dec 2020 13:49:11 -0500 Subject: [PATCH] Add initial infected asyncio error propagation test --- tests/test_infected_asyncio.py | 24 ++++++++++++++++++++++++ tractor/to_asyncio.py | 3 +++ 2 files changed, 27 insertions(+) create mode 100644 tests/test_infected_asyncio.py diff --git a/tests/test_infected_asyncio.py b/tests/test_infected_asyncio.py new file mode 100644 index 0000000..def1776 --- /dev/null +++ b/tests/test_infected_asyncio.py @@ -0,0 +1,24 @@ +import asyncio + +import pytest +import tractor + +async def sleep_and_err(): + await asyncio.sleep(0.1) + assert 0 + + +async def asyncio_actor(): + assert tractor.current_actor().is_infected_aio() + + await tractor.to_asyncio.run_task(sleep_and_err) + + +def test_infected_simple_error(arb_addr): + + async def main(): + async with tractor.open_nursery() as n: + await n.run_in_actor(asyncio_actor, infected_asyncio=True) + + with pytest.raises(tractor.RemoteActorError) as excinfo: + tractor.run(main, arbiter_addr=arb_addr) diff --git a/tractor/to_asyncio.py b/tractor/to_asyncio.py index 1fa923d..6e46903 100644 --- a/tractor/to_asyncio.py +++ b/tractor/to_asyncio.py @@ -164,16 +164,19 @@ def run_as_asyncio_guest( :) """ async def aio_main(trio_main): + loop = asyncio.get_running_loop() trio_done_fut = asyncio.Future() def trio_done_callback(main_outcome): + log.info(f"trio_main finished: {main_outcome!r}") trio_done_fut.set_result(main_outcome) # start the infection: run trio on the asyncio loop in "guest mode" log.info(f"Infecting asyncio process with {trio_main}") + trio.lowlevel.start_guest_run( trio_main, run_sync_soon_threadsafe=loop.call_soon_threadsafe,