Add a "show me the code" test from the readme

asyncgen_closing_fix
Tyler Goodlet 2018-07-07 22:22:05 -04:00
parent 77e34049b8
commit d94be22ef2
1 changed files with 128 additions and 15 deletions

View File

@ -9,11 +9,6 @@ import trio
import tractor
@pytest.fixture
def us_symbols():
return ['TSLA', 'AAPL', 'CGC', 'CRON']
@pytest.mark.trio
async def test_no_arbitter():
"""An arbitter must be established before any nurseries
@ -72,7 +67,7 @@ async def spawn(is_arbiter):
)
assert len(nursery._children) == 1
assert portal.channel.uid in tractor.current_actor()._peers
# be sure we can still get the result
# be sure we can still get the result
result = await portal.result()
assert result == 10
return result
@ -116,31 +111,31 @@ async def stream_from_single_subactor():
seq = range(10)
gen = await portal.run(
agen = await portal.run(
__name__,
'stream_seq', # the func above
sequence=list(seq), # has to be msgpack serializable
)
# it'd sure be nice to have an asyncitertools here...
iseq = iter(seq)
async for val in gen:
async for val in agen:
assert val == next(iseq)
break
# TODO: test breaking the loop (should it kill the
# far end?)
# break
# terminate far-end async-gen
# await gen.asend(None)
# break
# stop all spawned subactors
await nursery.cancel()
await portal.cancel_actor()
# await nursery.cancel()
def test_stream_from_single_subactor(us_symbols):
def test_stream_from_single_subactor():
"""Verify streaming from a spawned async generator.
"""
tractor.run(
stream_from_single_subactor,
name='client',
)
tractor.run(stream_from_single_subactor)
async def assert_err():
@ -159,10 +154,128 @@ def test_remote_error():
try:
return await portal.result()
except tractor.RemoteActorError:
print("Look Maa that actor failed hard, hehh")
raise
except Exception:
pass
assert 0, "Remote error was not raised?"
with pytest.raises(tractor.RemoteActorError):
# also raises
tractor.run(main)
def do_nothing():
pass
def test_cancel_single_subactor():
async def main():
async with tractor.open_nursery() as nursery:
portal = await nursery.start_actor(
'nothin', rpc_module_paths=[__name__],
)
assert not await portal.run(__name__, 'do_nothing')
# would hang otherwise
await nursery.cancel()
tractor.run(main)
async def stream_data(seed):
for i in range(seed):
yield i
# await trio.sleep(1/10000.) # trigger scheduler
await trio.sleep(0) # trigger scheduler
async def aggregate(seed):
"""Ensure that the two streams we receive match but only stream
a single set of values to the parent.
"""
async with tractor.open_nursery() as nursery:
portals = []
for i in range(1, 3):
# fork point
portal = await nursery.start_actor(
name=f'streamer_{i}',
rpc_module_paths=[__name__],
outlive_main=True, # daemonize these actors
)
portals.append(portal)
q = trio.Queue(int(1e3))
async def push_to_q(portal):
async for value in await portal.run(
__name__, 'stream_data', seed=seed
):
await q.put(value)
await q.put(None)
print(f"FINISHED ITERATING {portal.channel.uid}")
# spawn 2 trio tasks to collect streams and push to a local queue
async with trio.open_nursery() as n:
for portal in portals:
n.start_soon(push_to_q, portal)
unique_vals = set()
async for value in q:
if value not in unique_vals:
unique_vals.add(value)
# yield upwards to the spawning parent actor
yield value
continue
assert value in unique_vals
if value is None:
break
print("FINISHED ITERATING in aggregator")
await nursery.cancel()
print("WAITING on `ActorNursery` to finish")
print("AGGREGATOR COMPLETE!")
async def main():
# a nursery which spawns "actors"
async with tractor.open_nursery() as nursery:
seed = int(10)
import time
pre_start = time.time()
portal = await nursery.start_actor(
name='aggregator',
# executed in the actor's "main task" immediately
main=partial(aggregate, seed),
)
start = time.time()
# the portal call returns exactly what you'd expect
# as if the remote "main" function was called locally
result_stream = []
async for value in await portal.result():
result_stream.append(value)
print(f"STREAM TIME = {time.time() - start}")
print(f"STREAM + SPAWN TIME = {time.time() - pre_start}")
assert result_stream == list(range(seed)) + [None]
def test_show_me_the_code():
"""Verify the *show me the code* readme example works.
"""
tractor.run(main, arbiter_addr=('127.0.0.1', 1616))
def test_cancel_smtc():
"""Verify we can cancel midway through the smtc example gracefully.
"""
pass