diff options
Diffstat (limited to 'packages/meshbay-node/tests/test_disk_io_off_loop.py')
| -rw-r--r-- | packages/meshbay-node/tests/test_disk_io_off_loop.py | 11 |
1 files changed, 5 insertions, 6 deletions
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. |