goodboy
/
tractor
1
0
Fork 1
Code Issues 1 Pull Requests 4 Packages Projects Releases Wiki Activity

Get rid of external teardown trigger

246_facepalm_backup
overclockworked64 2021-10-22 03:10:46 +02:00 committed by Tyler Goodlet
parent 87e3d32992
commit b91adcf38d
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_clustering.py
View File

@ -23,19 +23,15 @@ async def worker(ctx: tractor.Context) -> None:
@tractor_test @tractor_test
async def test_streaming_to_actor_cluster() -> None: async def test_streaming_to_actor_cluster() -> None:
teardown_trigger = trio.Event()
async with ( async with (
open_actor_cluster(modules=[__name__]) as portals, open_actor_cluster(modules=[__name__]) as portals,
async_enter_all( async_enter_all(
mngrs=[p.open_context(worker) for p in portals.values()], mngrs=[p.open_context(worker) for p in portals.values()],
teardown_trigger=teardown_trigger,
) as contexts, ) as contexts,
async_enter_all( async_enter_all(
mngrs=[ctx[0].open_stream() for ctx in contexts], mngrs=[ctx[0].open_stream() for ctx in contexts],
teardown_trigger=teardown_trigger,
) as streams, ) as streams,
): ):
with trio.move_on_after(1): with trio.move_on_after(1):
for stream in itertools.cycle(streams): for stream in itertools.cycle(streams):
await stream.send(MESSAGE) await stream.send(MESSAGE)
teardown_trigger.set()