StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

test_mplex_stream: refactor

Browse Source
This commit is contained in:
mhchia 2020-02-05 17:25:39 +08:00
parent 996b5cf15d
commit 1e7d5c73ee
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/stream_muxer/test_mplex_stream.py
View File

@ -7,26 +7,16 @@ from libp2p.stream_muxer.mplex.exceptions import (
MplexStreamEOF, MplexStreamEOF,
MplexStreamReset, MplexStreamReset,
) )
from libp2p.tools.constants import LISTEN_MADDR, MAX_READ_LEN from libp2p.tools.constants import MAX_READ_LEN
from libp2p.tools.factories import SwarmFactory
from libp2p.tools.utils import connect_swarm
DATA = b"data_123" DATA = b"data_123"
@pytest.mark.trio @pytest.mark.trio
async def test_mplex_stream_read_write(): async def test_mplex_stream_read_write(mplex_stream_pair):
async with SwarmFactory.create_batch_and_listen(False, 2) as swarms: stream_0, stream_1 = mplex_stream_pair
await swarms[0].listen(LISTEN_MADDR) await stream_0.write(DATA)
await swarms[1].listen(LISTEN_MADDR) assert (await stream_1.read(MAX_READ_LEN)) == DATA
await connect_swarm(swarms[0], swarms[1])
conn_0 = swarms[0].connections[swarms[1].get_peer_id()]
conn_1 = swarms[1].connections[swarms[0].get_peer_id()]
stream_0 = await conn_0.muxed_conn.open_stream()
await trio.sleep(1)
stream_1 = tuple(conn_1.muxed_conn.streams.values())[0]
await stream_0.write(DATA)
assert (await stream_1.read(MAX_READ_LEN)) == DATA
@pytest.mark.trio @pytest.mark.trio