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

resolved recv_window updates,added support for read_EOF

Browse Source
This commit is contained in:
kaneki003
2025-06-21 13:39:03 +05:30
parent 0a7e13f0ed
commit 209deffc8a
2 changed files with 52 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/core/stream_muxer/test_mplex_read_write_lock.py
View File

@ -1,7 +1,7 @@
import pytest
import trio
from libp2p.abc import ISecureConn
from libp2p.abc import IMuxedStream, ISecureConn
from libp2p.crypto.keys import PrivateKey, PublicKey
from libp2p.peer.id import ID
from libp2p.stream_muxer.mplex.constants import (
@ -59,13 +59,15 @@ class DummyMuxedConn(Mplex):
self.event_started = trio.Event()
self.stream_backlog_limit = 256
self.stream_backlog_semaphore = trio.Semaphore(256)
channels = trio.open_memory_channel[MplexStream](0)
# Use IMuxedStream for type consistency with Mplex
channels = trio.open_memory_channel[IMuxedStream](0)
self.new_stream_send_channel, self.new_stream_receive_channel = channels
async def send_message(
self, flag: HeaderTags, data: bytes, stream_id: StreamID
) -> None:
self, flag: HeaderTags, data: bytes | None, stream_id: StreamID
) -> int:
await trio.sleep(0.01)
return 0
@pytest.mark.trio
@ -75,10 +77,11 @@ async def test_concurrent_writes_are_serialized():
class LoggingMuxedConn(DummyMuxedConn):
async def send_message(
self, flag: HeaderTags, data: bytes, stream_id: StreamID
) -> None:
self, flag: HeaderTags, data: bytes | None, stream_id: StreamID
) -> int:
send_log.append(data)
await trio.sleep(0.01)
return 0
memory_send, memory_recv = trio.open_memory_channel(8)
stream = MplexStream(