From 02aeb0a751dce80fac1bfca31b6f736558953b77 Mon Sep 17 00:00:00 2001 From: Christophe Besson Date: Thu, 24 Sep 2026 10:12:37 +0200 Subject: refactor(node): move uploads out of webrtc_server UploadMixin in transport/webrtc/upload_handlers.py, with the upload cap; _upload_chunk moved unchanged, _append_chunk joins webrtc/disk.py. Co-Authored-By: Claude Opus 5.5 --- packages/meshbay-node/tests/test_disk_io_off_loop.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'packages/meshbay-node/tests/test_disk_io_off_loop.py') diff --git a/packages/meshbay-node/tests/test_disk_io_off_loop.py b/packages/meshbay-node/tests/test_disk_io_off_loop.py index e404b47..a94789a 100644 --- a/packages/meshbay-node/tests/test_disk_io_off_loop.py +++ b/packages/meshbay-node/tests/test_disk_io_off_loop.py @@ -34,8 +34,7 @@ from meshbay_node.indexer.group_index import GroupIndex from meshbay_node.indexer.indexer import DirectoryIndexer from meshbay_node.roots import Root, RootSet from meshbay_node.transfers import LeaselessReads -from meshbay_node.transport import webrtc_server -from meshbay_node.transport.webrtc import files, media_tools +from meshbay_node.transport.webrtc import files, media_tools, upload_handlers from meshbay_node.transport.webrtc_server import WebRTCPeerSession from node_source import webrtc_files @@ -316,8 +315,8 @@ def _upload_session(tmp_path): async def test_a_slow_upload_write_does_not_stop_the_loop(tmp_path, monkeypatch): session, shared = _upload_session(tmp_path) - monkeypatch.setattr(webrtc_server, "_append_chunk", - _slow(webrtc_server._append_chunk)) + monkeypatch.setattr(upload_handlers, "_append_chunk", + _slow(upload_handlers._append_chunk)) with _Ticker() as ticker: await session._do_file_upload(sealed_upload( @@ -344,8 +343,8 @@ async def test_chunks_of_one_upload_keep_their_order_under_a_slow_disk(tmp_path, session, shared = _upload_session(tmp_path) pieces = [b"first-", b"second-", b"third-", b"fourth"] - monkeypatch.setattr(webrtc_server, "_append_chunk", - _slow(webrtc_server._append_chunk)) + monkeypatch.setattr(upload_handlers, "_append_chunk", + _slow(upload_handlers._append_chunk)) # Fired together and in order, which is what the dispatcher does: it creates # one task per message as it arrives. -- cgit v1.2.3