varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

intgrated n==-1 case in read()

Browse Source
This commit is contained in:
kaneki003
2025-06-21 17:51:27 +05:30
parent df17788ec3
commit d7cdae8a0f
2 changed files with 70 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/core/stream_muxer/test_yamux_read_write_lock.py
View File

@ -16,6 +16,7 @@ from libp2p.peer.id import (
from libp2p.security.insecure.transport import (
InsecureTransport,
)
from libp2p.stream_muxer.exceptions import MuxedStreamEOF
from libp2p.stream_muxer.yamux.yamux import (
Yamux,
YamuxStream,
@ -139,8 +140,8 @@ async def test_yamux_race_condition_without_locks(yamux_pair):
client_yamux, server_yamux = yamux_pair
client_stream: YamuxStream = await client_yamux.open_stream()
server_stream: YamuxStream = await server_yamux.accept_stream()
MSG_COUNT = 10
MSG_SIZE = 256 * 1024
MSG_COUNT = 1
MSG_SIZE = 512 * 1024
client_msgs = [
f"CLIENT-MSG-{i:03d}-".encode().ljust(MSG_SIZE, b"C") for i in range(MSG_COUNT)
]
@ -160,11 +161,17 @@ async def test_yamux_race_condition_without_locks(yamux_pair):
async def reader(stream, received, name):
"""Read messages and store them for verification."""
for i in range(MSG_COUNT):
data = await stream.read(MSG_SIZE)
received.append(data)
if i % 3 == 0:
await trio.sleep(0.001)
try:
data = await stream.read()
if data:
received.append(data)
except MuxedStreamEOF:
pass
# for i in range(MSG_COUNT):
# data = await stream.read()
# received.append(data)
# if i % 3 == 0:
# await trio.sleep(0.001)
# Running all operations concurrently
async with trio.open_nursery() as nursery:
@ -173,12 +180,12 @@ async def test_yamux_race_condition_without_locks(yamux_pair):
nursery.start_soon(reader, client_stream, client_received, "client")
nursery.start_soon(reader, server_stream, server_received, "server")
assert len(client_received) == MSG_COUNT, (
f"Client received {len(client_received)} messages, expected {MSG_COUNT}"
)
assert len(server_received) == MSG_COUNT, (
f"Server received {len(server_received)} messages, expected {MSG_COUNT}"
)
# assert len(client_received) == MSG_COUNT, (
# f"Client received {len(client_received)} messages, expected {MSG_COUNT}"
# )
# assert len(server_received) == MSG_COUNT, (
# f"Server received {len(server_received)} messages, expected {MSG_COUNT}"
# )
assert client_received == server_msgs, (
"Client did not receive server messages in order or intact!"
)