Compare commits
3 Commits
67f673bf36
...
f2ce4a3469
Author | SHA1 | Date |
---|---|---|
Tyler Goodlet | f2ce4a3469 | |
Tyler Goodlet | 3aa964315a | |
Tyler Goodlet | f3ca8608d5 |
|
@ -6,6 +6,7 @@ sync-opening a ``tractor.Context`` beforehand.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from itertools import count
|
from itertools import count
|
||||||
|
import math
|
||||||
import platform
|
import platform
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
from typing import (
|
from typing import (
|
||||||
|
@ -872,7 +873,7 @@ def test_one_end_stream_not_opened(
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
with trio.fail_after(1):
|
with trio.fail_after(0.8):
|
||||||
async with portal.open_context(
|
async with portal.open_context(
|
||||||
entrypoint,
|
entrypoint,
|
||||||
) as (ctx, sent):
|
) as (ctx, sent):
|
||||||
|
@ -1059,7 +1060,17 @@ def test_maybe_allow_overruns_stream(
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
)
|
)
|
||||||
seq = list(range(10))
|
|
||||||
|
# stream-sequence batch info with send delay to determine
|
||||||
|
# approx timeout determining whether test has hung.
|
||||||
|
total_batches: int = 2
|
||||||
|
num_items: int = 10
|
||||||
|
seq = list(range(num_items))
|
||||||
|
parent_send_delay: float = 0.16
|
||||||
|
timeout: float = math.ceil(
|
||||||
|
total_batches * num_items * parent_send_delay
|
||||||
|
)
|
||||||
|
with trio.fail_after(timeout):
|
||||||
async with portal.open_context(
|
async with portal.open_context(
|
||||||
echo_back_sequence,
|
echo_back_sequence,
|
||||||
seq=seq,
|
seq=seq,
|
||||||
|
@ -1075,7 +1086,6 @@ def test_maybe_allow_overruns_stream(
|
||||||
allow_overruns=(allow_overruns_side in {'parent', 'both'}),
|
allow_overruns=(allow_overruns_side in {'parent', 'both'}),
|
||||||
) as stream:
|
) as stream:
|
||||||
|
|
||||||
total_batches: int = 2
|
|
||||||
for _ in range(total_batches):
|
for _ in range(total_batches):
|
||||||
for msg in seq:
|
for msg in seq:
|
||||||
# print(f'root tx {msg}')
|
# print(f'root tx {msg}')
|
||||||
|
@ -1084,7 +1094,7 @@ def test_maybe_allow_overruns_stream(
|
||||||
# NOTE: we make the parent slightly
|
# NOTE: we make the parent slightly
|
||||||
# slower, when it is slow, to make sure
|
# slower, when it is slow, to make sure
|
||||||
# that in the overruns everywhere case
|
# that in the overruns everywhere case
|
||||||
await trio.sleep(0.16)
|
await trio.sleep(parent_send_delay)
|
||||||
|
|
||||||
batch = []
|
batch = []
|
||||||
async for msg in stream:
|
async for msg in stream:
|
||||||
|
|
|
@ -438,8 +438,8 @@ _ctxvar_MsgCodec: MsgCodec = RunVar(
|
||||||
'msgspec_codec',
|
'msgspec_codec',
|
||||||
|
|
||||||
# TODO: move this to our new `Msg`-spec!
|
# TODO: move this to our new `Msg`-spec!
|
||||||
# default=_def_msgspec_codec,
|
default=_def_msgspec_codec,
|
||||||
default=_def_tractor_codec,
|
# default=_def_tractor_codec,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue