Wire `test_dynamic_pub_sub` to standard fixtures
Pull in the `reg_addr`, `debug_mode`, and `test_log`
fixtures so this test follows the same conventions as
the rest of the suite:
- pass `registry_addrs=[reg_addr]` + `debug_mode` into
`tractor.open_nursery()` (so `--tpdb` etc work).
- after the `pytest.raises` block, add `assert err` +
`test_log.exception('Timed out AS EXPECTED')` so the
expected timeout is logged explicitly instead of
swallowed.
Also,
- drop whitespace-only blank lines around the
`subs` param of `consumer()` and `ctx` param of
`one_task_streams_and_one_handles_reqresp()`.
- promote `test_sigint_both_stream_types`'s one-line
docstring to multi-line form.
(this commit msg was generated in some part by [`claude-code`][claude-code-gh])
[claude-code-gh]: https://github.com/anthropics/claude-code
subint_forkserver_backend
parent
65fcfbf224
commit
9b05f659b3
|
|
@ -76,9 +76,7 @@ async def subscribe(
|
||||||
|
|
||||||
|
|
||||||
async def consumer(
|
async def consumer(
|
||||||
|
|
||||||
subs: list[str],
|
subs: list[str],
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
uid = tractor.current_actor().uid
|
uid = tractor.current_actor().uid
|
||||||
|
|
@ -108,15 +106,21 @@ async def consumer(
|
||||||
print(f'{uid} got: {value}')
|
print(f'{uid} got: {value}')
|
||||||
|
|
||||||
|
|
||||||
def test_dynamic_pub_sub():
|
def test_dynamic_pub_sub(
|
||||||
|
reg_addr: tuple,
|
||||||
|
debug_mode: bool,
|
||||||
|
test_log: tractor.log.StackLevelAdapter,
|
||||||
|
):
|
||||||
global _registry
|
global _registry
|
||||||
|
|
||||||
from multiprocessing import cpu_count
|
from multiprocessing import cpu_count
|
||||||
cpus = cpu_count()
|
cpus = cpu_count()
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
registry_addrs=[reg_addr],
|
||||||
|
debug_mode=debug_mode,
|
||||||
|
) as n:
|
||||||
|
|
||||||
# name of this actor will be same as target func
|
# name of this actor will be same as target func
|
||||||
await n.run_in_actor(publisher)
|
await n.run_in_actor(publisher)
|
||||||
|
|
@ -155,12 +159,13 @@ def test_dynamic_pub_sub():
|
||||||
else:
|
else:
|
||||||
pytest.fail('Never got a `TooSlowError` ?')
|
pytest.fail('Never got a `TooSlowError` ?')
|
||||||
|
|
||||||
|
assert err
|
||||||
|
test_log.exception('Timed out AS EXPECTED')
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def one_task_streams_and_one_handles_reqresp(
|
async def one_task_streams_and_one_handles_reqresp(
|
||||||
|
|
||||||
ctx: tractor.Context,
|
ctx: tractor.Context,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
|
|
@ -257,7 +262,8 @@ async def echo_ctx_stream(
|
||||||
|
|
||||||
|
|
||||||
def test_sigint_both_stream_types():
|
def test_sigint_both_stream_types():
|
||||||
'''Verify that running a bi-directional and recv only stream
|
'''
|
||||||
|
Verify that running a bi-directional and recv only stream
|
||||||
side-by-side will cancel correctly from SIGINT.
|
side-by-side will cancel correctly from SIGINT.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue