diff options
Diffstat (limited to 'packages/meshbay-node/tests')
| -rw-r--r-- | packages/meshbay-node/tests/test_audio_transcode.py | 6 | ||||
| -rw-r--r-- | packages/meshbay-node/tests/test_disk_io_off_loop.py | 14 | ||||
| -rw-r--r-- | packages/meshbay-node/tests/test_transport_wire_parity.py | 5 |
3 files changed, 13 insertions, 12 deletions
diff --git a/packages/meshbay-node/tests/test_audio_transcode.py b/packages/meshbay-node/tests/test_audio_transcode.py index 941ada2..a494be7 100644 --- a/packages/meshbay-node/tests/test_audio_transcode.py +++ b/packages/meshbay-node/tests/test_audio_transcode.py @@ -19,8 +19,8 @@ from meshbay_common.protocol import MNP, IndexEntry from meshbay_common.webcrypto import chunk_key_aes, decrypt_chunk_aes from meshbay_node.indexer.group_index import GroupIndex from meshbay_node.media_cache import MediaCache -from meshbay_node.transport import webrtc_server as wrs from meshbay_node.transport.webrtc.apps import music +from meshbay_node.transport.webrtc.limits import CHUNK_SIZE from meshbay_node.transport.webrtc_server import WebRTCPeerSession from conftest import needs_subprocess, one_root @@ -180,11 +180,11 @@ async def test_multi_chunk_cached_blob_reassembles_correctly(tmp_path, media_cac session.sent = [] session._send = session.sent.append - blob = os.urandom(int(wrs.CHUNK_SIZE * 2.3)) + blob = os.urandom(int(CHUNK_SIZE * 2.3)) blob_hash = blake3.blake3(blob).hexdigest() await media_cache.put_thumb(blob_hash, "synthetic:test", blob) - total_chunks = -(-len(blob) // wrs.CHUNK_SIZE) + total_chunks = -(-len(blob) // CHUNK_SIZE) for i in range(total_chunks): await session._do_file_request({"file_id": blob_hash, "chunk_index": i}) # One past the end must be a clean miss, not a partial/garbage chunk. 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 b77200a..e404b47 100644 --- a/packages/meshbay-node/tests/test_disk_io_off_loop.py +++ b/packages/meshbay-node/tests/test_disk_io_off_loop.py @@ -35,7 +35,7 @@ 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 media_tools +from meshbay_node.transport.webrtc import files, media_tools from meshbay_node.transport.webrtc_server import WebRTCPeerSession from node_source import webrtc_files @@ -130,8 +130,8 @@ def _slow(fn): async def test_a_slow_chunk_read_does_not_stop_the_loop(tmp_path, monkeypatch): session, _, entry, gek = await _served(tmp_path) - monkeypatch.setattr(webrtc_server, "_read_and_encrypt", - _slow(webrtc_server._read_and_encrypt)) + monkeypatch.setattr(files, "_read_and_encrypt", + _slow(files._read_and_encrypt)) with _Ticker() as ticker: await session._do_file_request( @@ -156,7 +156,7 @@ async def test_a_slow_stat_does_not_stop_the_loop(tmp_path, monkeypatch): would then find the disk already awake — the stall moved, not removed. """ session, _, entry, _ = await _served(tmp_path) - monkeypatch.setattr(webrtc_server, "_locate", _slow(webrtc_server._locate)) + monkeypatch.setattr(files, "_locate", _slow(files._locate)) with _Ticker() as ticker: await session._do_file_request( @@ -180,7 +180,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path): inside = 0 peak = 0 guard = threading.Lock() - real = webrtc_server._read_and_encrypt + real = files._read_and_encrypt def counting(*args, **kwargs): nonlocal inside, peak @@ -194,7 +194,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path): with guard: inside -= 1 - webrtc_server._read_and_encrypt = counting + files._read_and_encrypt = counting try: await asyncio.gather(*[ session._do_file_request( @@ -202,7 +202,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path): for _ in range(6) ]) finally: - webrtc_server._read_and_encrypt = real + files._read_and_encrypt = real assert peak == 1, f"{peak} reads of one root set were inside the disk at once" diff --git a/packages/meshbay-node/tests/test_transport_wire_parity.py b/packages/meshbay-node/tests/test_transport_wire_parity.py index cd31fb4..3250f7f 100644 --- a/packages/meshbay-node/tests/test_transport_wire_parity.py +++ b/packages/meshbay-node/tests/test_transport_wire_parity.py @@ -21,7 +21,8 @@ from meshbay_common.crypto import generate_gek from meshbay_common.groupbox import PURPOSE_INDEX, unseal from meshbay_common.protocol import MNP, file_chunk_plaintext, file_chunk_wire from meshbay_node.indexer import DirectoryIndexer -from meshbay_node.transport import quic_server, webrtc_server +from meshbay_node.transport import quic_server +from meshbay_node.transport.webrtc import disk from meshbay_node.transport.wire import index_sync_message from node_source import webrtc_source @@ -102,7 +103,7 @@ def test_chunk_wire_shape_is_identical_across_transports(gek, shared_dir): path = shared_dir / "film.mkv" file_hash = bytes.fromhex("ab" * 32) - from_webrtc = webrtc_server._read_and_encrypt(gek, path, 0, file_hash, "ab" * 32) + from_webrtc = disk._read_and_encrypt(gek, path, 0, file_hash, "ab" * 32) from_quic = quic_server._read_and_encrypt(gek, path, 0, file_hash, "ab" * 32) assert from_webrtc.keys() == from_quic.keys() |