summaryrefslogtreecommitdiffstats
path: root/packages/meshbay-node/tests/test_task_lifetime.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-node/tests/test_task_lifetime.py')
-rw-r--r--packages/meshbay-node/tests/test_task_lifetime.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/packages/meshbay-node/tests/test_task_lifetime.py b/packages/meshbay-node/tests/test_task_lifetime.py
index 3a5b8a5..9dffedb 100644
--- a/packages/meshbay-node/tests/test_task_lifetime.py
+++ b/packages/meshbay-node/tests/test_task_lifetime.py
@@ -245,8 +245,13 @@ def test_chunks_wait_for_room_on_the_channel(session):
work through the rest — which is what "stuck at 1 MB" looks like, one chunk
being exactly one megabyte.
"""
- fn = session[session.index("async def _do_file_request"):]
- fn = fn[:fn.index("\n def _do_stream_segment")]
+ start = session.index("async def _do_file_request")
+ # Up to whatever the next member is. This used to end at
+ # "\n def _do_stream_segment" — a neighbour removed in MNP 2.0 — and an
+ # `index()` on a name that no longer exists fails the test for a reason
+ # that has nothing to do with what it is about.
+ nxt = re.search(r"\n (?:@|(?:async )?def )", session[start:])
+ fn = session[start:start + nxt.start()] if nxt else session[start:]
assert "DOWNLOAD_BUFFER_HIGH" in fn, "the send buffer has to be watched"
assert "await asyncio.sleep" in fn, "waiting for room is the point"
assert 'readyState != "open"' in fn, (